BatchRestApi

Author: turboFei

kyuubi-rest-client/src/main/java/org/apache/kyuubi/client/BatchRestApi.java

@@ -174,6 +174,13 @@ public CloseBatchResponse deleteBatch(String batchId, Map<String, String> header
         .delete(path, null, CloseBatchResponse.class, client.getAuthHeader(), headers);
   }
 
+  public ReassignBatchResponse reassignBatch(ReassignBatchRequest request) {
+    String path = String.format("%s/reassign", API_BASE_PATH);
+    String requestBody = JsonUtils.toJson(request);
+    return this.getClient()
+        .post(path, requestBody, ReassignBatchResponse.class, client.getAuthHeader());
+  }
+
   private IRestClient getClient() {
     return this.client.getHttpClient();
   }

kyuubi-rest-client/src/test/java/org/apache/kyuubi/client/BatchRestClientTest.java

@@ -26,6 +26,8 @@
 import org.apache.kyuubi.client.api.v1.dto.CloseBatchResponse;
 import org.apache.kyuubi.client.api.v1.dto.GetBatchesResponse;
 import org.apache.kyuubi.client.api.v1.dto.OperationLog;
+import org.apache.kyuubi.client.api.v1.dto.ReassignBatchRequest;
+import org.apache.kyuubi.client.api.v1.dto.ReassignBatchResponse;
 import org.apache.kyuubi.client.exception.KyuubiRestException;
 import org.junit.After;
 import org.junit.Before;
@@ -276,4 +278,19 @@ public void deleteBatchTest() {
     response = basicBatchRestApi.deleteBatch("71535");
     assertTrue(response.isSuccess());
   }
+
+  @Test
+  public void reassignBatchTest() {
+    // test spnego auth
+    BatchTestServlet.setAuthSchema(NEGOTIATE_AUTH);
+    ReassignBatchRequest request = new ReassignBatchRequest("http://127.0.0.1:10012");
+    ReassignBatchResponse response = spnegoBatchRestApi.reassignBatch(request);
+    assertTrue(response.getBatchIds().isEmpty());
+
+    // test basic auth
+    BatchTestServlet.setAuthSchema(BASIC_AUTH);
+    BatchTestServlet.allowAnonymous(false);
+    response = basicBatchRestApi.reassignBatch(request);
+    assertTrue(response.getBatchIds().isEmpty());
+  }
 }

kyuubi-rest-client/src/test/java/org/apache/kyuubi/client/BatchTestServlet.java

@@ -28,6 +28,7 @@
 import org.apache.kyuubi.client.api.v1.dto.CloseBatchResponse;
 import org.apache.kyuubi.client.api.v1.dto.GetBatchesResponse;
 import org.apache.kyuubi.client.api.v1.dto.OperationLog;
+import org.apache.kyuubi.client.api.v1.dto.ReassignBatchResponse;
 import org.apache.kyuubi.client.util.JsonUtils;
 
 public class BatchTestServlet extends HttpServlet {
@@ -81,6 +82,11 @@ protected void doPost(HttpServletRequest req, HttpServletResponse resp) throws I
       Batch batch = generateTestBatch();
       resp.getWriter().write(JsonUtils.toJson(batch));
       resp.getWriter().flush();
+    } else if (req.getPathInfo().equalsIgnoreCase("/api/v1/batches/reassign")) {
+      resp.setStatus(HttpServletResponse.SC_OK);
+
+      resp.getWriter().write(JsonUtils.toJson(new ReassignBatchResponse()));
+      resp.getWriter().flush();
     } else {
       resp.setStatus(HttpServletResponse.SC_NOT_FOUND);
     }