Browse Source

Rename import() to bulkload() (#285)

Signed-off-by: groot <yihua.mo@zilliz.com>
groot 3 years ago
parent
commit
fedc4e2ef3

+ 3 - 3
src/main/java/io/milvus/client/AbstractMilvusGrpcClient.java

@@ -1187,7 +1187,7 @@ public abstract class AbstractMilvusGrpcClient implements MilvusClient {
     }
     }
 
 
     @Override
     @Override
-    public R<ImportResponse> importData(@NonNull ImportParam requestParam) {
+    public R<ImportResponse> bulkload(@NonNull BulkloadParam requestParam) {
         if (!clientIsReady()) {
         if (!clientIsReady()) {
             return R.failed(new ClientNotConnectedException("Client rpc channel is not ready"));
             return R.failed(new ClientNotConnectedException("Client rpc channel is not ready"));
         }
         }
@@ -1224,7 +1224,7 @@ public abstract class AbstractMilvusGrpcClient implements MilvusClient {
     }
     }
 
 
     @Override
     @Override
-    public R<GetImportStateResponse> getImportState(GetImportStateParam requestParam) {
+    public R<GetImportStateResponse> getBulkloadState(GetBulkloadStateParam requestParam) {
         if (!clientIsReady()) {
         if (!clientIsReady()) {
             return R.failed(new ClientNotConnectedException("Client rpc channel is not ready"));
             return R.failed(new ClientNotConnectedException("Client rpc channel is not ready"));
         }
         }
@@ -1253,7 +1253,7 @@ public abstract class AbstractMilvusGrpcClient implements MilvusClient {
     }
     }
 
 
     @Override
     @Override
-    public R<ListImportTasksResponse> listImportTasks(@NonNull ListImportTasksParam requestParam) {
+    public R<ListImportTasksResponse> listBulkloadTasks(@NonNull ListBulkloadTasksParam requestParam) {
         if (!clientIsReady()) {
         if (!clientIsReady()) {
             return R.failed(new ClientNotConnectedException("Client rpc channel is not ready"));
             return R.failed(new ClientNotConnectedException("Client rpc channel is not ready"));
         }
         }

+ 8 - 8
src/main/java/io/milvus/client/MilvusClient.java

@@ -288,26 +288,26 @@ public interface MilvusClient {
     /**
     /**
      * Import data from external files, currently support JSON/Numpy format
      * Import data from external files, currently support JSON/Numpy format
      *
      *
-     * @param requestParam {@link ImportParam}
+     * @param requestParam {@link BulkloadParam}
      * @return {status:result code, data:RpcStatus{msg: result message}}
      * @return {status:result code, data:RpcStatus{msg: result message}}
      */
      */
-    R<ImportResponse> importData(ImportParam requestParam);
+    R<ImportResponse> bulkload(BulkloadParam requestParam);
 
 
     /**
     /**
-     * Import data from external files, currently support JSON/Numpy format
+     * Get state of bulk a load task
      *
      *
-     * @param requestParam {@link GetImportStateParam}
+     * @param requestParam {@link GetBulkloadStateParam}
      * @return {status:result code, data:GetImportStateResponse{status,state}}
      * @return {status:result code, data:GetImportStateResponse{status,state}}
      */
      */
-    R<GetImportStateResponse> getImportState(GetImportStateParam requestParam);
+    R<GetImportStateResponse> getBulkloadState(GetBulkloadStateParam requestParam);
 
 
     /**
     /**
-     * List all import tasks
+     * List all bulk load tasks
      *
      *
-     * @param requestParam {@link ListImportTasksParam}
+     * @param requestParam {@link ListBulkloadTasksParam}
      * @return {status:result code, data:ListImportTasksResponse{status,state}}
      * @return {status:result code, data:ListImportTasksResponse{status,state}}
      */
      */
-    R<ListImportTasksResponse> listImportTasks(ListImportTasksParam requestParam);
+    R<ListImportTasksResponse> listBulkloadTasks(ListBulkloadTasksParam requestParam);
 
 
     /**
     /**
      * Conducts ANN search on a vector field. Use expression to do filtering before search.
      * Conducts ANN search on a vector field. Use expression to do filtering before search.

+ 6 - 6
src/main/java/io/milvus/client/MilvusMultiServiceClient.java

@@ -285,21 +285,21 @@ public class MilvusMultiServiceClient implements MilvusClient {
     }
     }
 
 
     @Override
     @Override
-    public R<ImportResponse> importData(@NonNull ImportParam requestParam) {
+    public R<ImportResponse> bulkload(@NonNull BulkloadParam requestParam) {
         List<R<ImportResponse>> response = this.clusterFactory.getAvailableServerSettings().stream()
         List<R<ImportResponse>> response = this.clusterFactory.getAvailableServerSettings().stream()
-                .map(serverSetting -> serverSetting.getClient().importData(requestParam))
+                .map(serverSetting -> serverSetting.getClient().bulkload(requestParam))
                 .collect(Collectors.toList());
                 .collect(Collectors.toList());
         return handleResponse(response);
         return handleResponse(response);
     }
     }
 
 
     @Override
     @Override
-    public R<GetImportStateResponse> getImportState(GetImportStateParam requestParam) {
-        return this.clusterFactory.getMaster().getClient().getImportState(requestParam);
+    public R<GetImportStateResponse> getBulkloadState(GetBulkloadStateParam requestParam) {
+        return this.clusterFactory.getMaster().getClient().getBulkloadState(requestParam);
     }
     }
 
 
     @Override
     @Override
-    public R<ListImportTasksResponse> listImportTasks(@NonNull ListImportTasksParam requestParam) {
-        return this.clusterFactory.getMaster().getClient().listImportTasks(requestParam);
+    public R<ListImportTasksResponse> listBulkloadTasks(@NonNull ListBulkloadTasksParam requestParam) {
+        return this.clusterFactory.getMaster().getClient().listBulkloadTasks(requestParam);
     }
     }
 
 
     @Override
     @Override

+ 10 - 10
src/main/java/io/milvus/param/dml/ImportParam.java → src/main/java/io/milvus/param/dml/BulkloadParam.java

@@ -32,17 +32,17 @@ import java.util.List;
 import java.util.Map;
 import java.util.Map;
 
 
 /**
 /**
- * Parameters for <code>importData</code> interface.
+ * Parameters for <code>bulkload</code> interface.
  */
  */
 @Getter
 @Getter
-public class ImportParam {
+public class BulkloadParam {
     private final String collectionName;
     private final String collectionName;
     private final String partitionName;
     private final String partitionName;
     private final boolean rowBased;
     private final boolean rowBased;
     private final List<String> files;
     private final List<String> files;
     private final Map<String, String> options = new HashMap<>();
     private final Map<String, String> options = new HashMap<>();
 
 
-    private ImportParam(@NonNull Builder builder) {
+    private BulkloadParam(@NonNull Builder builder) {
         this.collectionName = builder.collectionName;
         this.collectionName = builder.collectionName;
         this.partitionName = builder.partitionName;
         this.partitionName = builder.partitionName;
         this.rowBased = builder.rowBased;
         this.rowBased = builder.rowBased;
@@ -55,7 +55,7 @@ public class ImportParam {
     }
     }
 
 
     /**
     /**
-     * Builder for {@link ImportParam} class.
+     * Builder for {@link BulkloadParam} class.
      */
      */
     public static class Builder {
     public static class Builder {
         private String collectionName;
         private String collectionName;
@@ -137,25 +137,25 @@ public class ImportParam {
         }
         }
 
 
         /**
         /**
-         * Verifies parameters and creates a new {@link ImportParam} instance.
+         * Verifies parameters and creates a new {@link BulkloadParam} instance.
          *
          *
-         * @return {@link ImportParam}
+         * @return {@link BulkloadParam}
          */
          */
-        public ImportParam build() throws ParamException {
+        public BulkloadParam build() throws ParamException {
             ParamUtils.CheckNullEmptyString(collectionName, "Collection name");
             ParamUtils.CheckNullEmptyString(collectionName, "Collection name");
 
 
-            return new ImportParam(this);
+            return new BulkloadParam(this);
         }
         }
     }
     }
 
 
     /**
     /**
-     * Constructs a <code>String</code> by {@link ImportParam} instance.
+     * Constructs a <code>String</code> by {@link BulkloadParam} instance.
      *
      *
      * @return <code>String</code>
      * @return <code>String</code>
      */
      */
     @Override
     @Override
     public String toString() {
     public String toString() {
-        return "ImportParam{" +
+        return "BulkloadParam{" +
                 "collectionName='" + collectionName + '\'' +
                 "collectionName='" + collectionName + '\'' +
                 ", partitionName='" + partitionName + '\'' +
                 ", partitionName='" + partitionName + '\'' +
                 ", files='" + files.toString() + '\'' +
                 ", files='" + files.toString() + '\'' +

+ 11 - 11
src/main/java/io/milvus/param/dml/GetImportStateParam.java → src/main/java/io/milvus/param/dml/GetBulkloadStateParam.java

@@ -5,13 +5,13 @@ import lombok.Getter;
 import lombok.NonNull;
 import lombok.NonNull;
 
 
 /**
 /**
- * Parameters for <code>getImportState</code> interface.
+ * Parameters for <code>getBulkloadState</code> interface.
  */
  */
 @Getter
 @Getter
-public class GetImportStateParam {
+public class GetBulkloadStateParam {
     private final long taskID;
     private final long taskID;
 
 
-    private GetImportStateParam(@NonNull Builder builder) {
+    private GetBulkloadStateParam(@NonNull Builder builder) {
         this.taskID = builder.taskID;
         this.taskID = builder.taskID;
     }
     }
 
 
@@ -20,7 +20,7 @@ public class GetImportStateParam {
     }
     }
 
 
     /**
     /**
-     * Builder for {@link GetImportStateParam} class.
+     * Builder for {@link GetBulkloadStateParam} class.
      */
      */
     public static class Builder {
     public static class Builder {
         private Long taskID;
         private Long taskID;
@@ -29,7 +29,7 @@ public class GetImportStateParam {
         }
         }
 
 
         /**
         /**
-         * Sets an import task id. The id is returned from importData() interface.
+         * Sets an import task id. The id is returned from bulkload() interface.
          *
          *
          * @param taskID id of the task
          * @param taskID id of the task
          * @return <code>Builder</code>
          * @return <code>Builder</code>
@@ -40,27 +40,27 @@ public class GetImportStateParam {
         }
         }
 
 
         /**
         /**
-         * Verifies parameters and creates a new {@link GetImportStateParam} instance.
+         * Verifies parameters and creates a new {@link GetBulkloadStateParam} instance.
          *
          *
-         * @return {@link GetImportStateParam}
+         * @return {@link GetBulkloadStateParam}
          */
          */
-        public GetImportStateParam build() throws ParamException {
+        public GetBulkloadStateParam build() throws ParamException {
             if (this.taskID == null) {
             if (this.taskID == null) {
                 throw new ParamException("Task ID not specified");
                 throw new ParamException("Task ID not specified");
             }
             }
 
 
-            return new GetImportStateParam(this);
+            return new GetBulkloadStateParam(this);
         }
         }
     }
     }
 
 
     /**
     /**
-     * Constructs a <code>String</code> by {@link GetImportStateParam} instance.
+     * Constructs a <code>String</code> by {@link GetBulkloadStateParam} instance.
      *
      *
      * @return <code>String</code>
      * @return <code>String</code>
      */
      */
     @Override
     @Override
     public String toString() {
     public String toString() {
-        return "GetImportStateParam{" +
+        return "GetBulkloadStateParam{" +
                 "taskID='" + taskID + '\'' +
                 "taskID='" + taskID + '\'' +
                 '}';
                 '}';
     }
     }

+ 41 - 0
src/main/java/io/milvus/param/dml/ListBulkloadTasksParam.java

@@ -0,0 +1,41 @@
+package io.milvus.param.dml;
+
+import io.milvus.exception.ParamException;
+import lombok.Getter;
+import lombok.NonNull;
+
+/**
+ * Parameters for <code>getBulkloadState</code> interface.
+ */
+@Getter
+public class ListBulkloadTasksParam {
+
+    private ListBulkloadTasksParam(@NonNull Builder builder) {
+    }
+
+    public static Builder newBuilder() {
+        return new Builder();
+    }
+
+    /**
+     * Builder for {@link ListBulkloadTasksParam} class.
+     */
+    public static class Builder {
+        private Builder() {
+        }
+
+        public ListBulkloadTasksParam build() throws ParamException {
+            return new ListBulkloadTasksParam(this);
+        }
+    }
+
+    /**
+     * Constructs a <code>String</code> by {@link ListBulkloadTasksParam} instance.
+     *
+     * @return <code>String</code>
+     */
+    @Override
+    public String toString() {
+        return "ListBulkloadTasksParam";
+    }
+}

+ 0 - 41
src/main/java/io/milvus/param/dml/ListImportTasksParam.java

@@ -1,41 +0,0 @@
-package io.milvus.param.dml;
-
-import io.milvus.exception.ParamException;
-import lombok.Getter;
-import lombok.NonNull;
-
-/**
- * Parameters for <code>getImportState</code> interface.
- */
-@Getter
-public class ListImportTasksParam {
-
-    private ListImportTasksParam(@NonNull Builder builder) {
-    }
-
-    public static Builder newBuilder() {
-        return new Builder();
-    }
-
-    /**
-     * Builder for {@link ListImportTasksParam} class.
-     */
-    public static class Builder {
-        private Builder() {
-        }
-
-        public ListImportTasksParam build() throws ParamException {
-            return new ListImportTasksParam(this);
-        }
-    }
-
-    /**
-     * Constructs a <code>String</code> by {@link ListImportTasksParam} instance.
-     *
-     * @return <code>String</code>
-     */
-    @Override
-    public String toString() {
-        return "ListImportTasksParam";
-    }
-}

+ 16 - 16
src/main/java/io/milvus/response/GetImportStateWrapper.java → src/main/java/io/milvus/response/GetBulkloadStateWrapper.java

@@ -9,64 +9,64 @@ import lombok.NonNull;
 import java.util.List;
 import java.util.List;
 
 
 /**
 /**
- * Util class to wrap response of <code>getImportState</code> interface.
+ * Util class to wrap response of <code>getBulkloadState</code> interface.
  */
  */
-public class GetImportStateWrapper {
+public class GetBulkloadStateWrapper {
     private final GetImportStateResponse response;
     private final GetImportStateResponse response;
 
 
-    public GetImportStateWrapper(@NonNull GetImportStateResponse response) {
+    public GetBulkloadStateWrapper(@NonNull GetImportStateResponse response) {
         this.response = response;
         this.response = response;
     }
     }
 
 
     /**
     /**
-     * Gets ID of the import task.
+     * Gets ID of the bulk load task.
      *
      *
-     * @return Long ID of the import task
+     * @return Long ID of the bulk load task
      */
      */
     public long getTaskID() {
     public long getTaskID() {
         return response.getId();
         return response.getId();
     }
     }
 
 
     /**
     /**
-     * Gets the long ID array for auto-id primary key, generated by import task.
+     * Gets the long ID array for auto-id primary key, generated by bulk load task.
      *
      *
-     * @return List&lt;Long&gt; ID array returned by import task
+     * @return List&lt;Long&gt; ID array returned by bulk load task
      */
      */
     public List<Long> getAutoGeneratedIDs() {
     public List<Long> getAutoGeneratedIDs() {
         return response.getIdListList();
         return response.getIdListList();
     }
     }
 
 
     /**
     /**
-     * Gets state of the import task.
+     * Gets state of the bulk load task.
      *
      *
-     * @return ImportState state of the import task
+     * @return ImportState state of the bulk load task
      */
      */
     public ImportState getState() {
     public ImportState getState() {
         return response.getState();
         return response.getState();
     }
     }
 
 
     /**
     /**
-     * Gets how many rows were imported by the import task.
+     * Gets how many rows were imported by the bulk load task.
      *
      *
-     * @return Long how many rows were imported by the import task
+     * @return Long how many rows were imported by the bulk load task
      */
      */
     public long getImportedCount() {
     public long getImportedCount() {
         return response.getRowCount();
         return response.getRowCount();
     }
     }
 
 
     /**
     /**
-     * Gets failed reason of the import task.
+     * Gets failed reason of the bulk load task.
      *
      *
-     * @return String failed reason of the import task
+     * @return String failed reason of the bulk load task
      */
      */
     public String getFailedReason() {
     public String getFailedReason() {
         return getInfo(Constant.FAILED_REASON);
         return getInfo(Constant.FAILED_REASON);
     }
     }
 
 
     /**
     /**
-     * Gets target files of the import task.
+     * Gets target files of the bulk load task.
      *
      *
-     * @return String target files of the import task
+     * @return String target files of the bulk load task
      */
      */
     public String getFiles() {
     public String getFiles() {
         return getInfo(Constant.IMPORT_FILES);
         return getInfo(Constant.IMPORT_FILES);
@@ -90,7 +90,7 @@ public class GetImportStateWrapper {
      */
      */
     @Override
     @Override
     public String toString() {
     public String toString() {
-        return "Import task state{" +
+        return "bulk load task state{" +
                 ", autoGeneratedIDs:" + getAutoGeneratedIDs() +
                 ", autoGeneratedIDs:" + getAutoGeneratedIDs() +
                 ", state:" + getState().name() +
                 ", state:" + getState().name() +
                 ", failed reason:" + getFailedReason() +
                 ", failed reason:" + getFailedReason() +

+ 11 - 11
src/test/java/io/milvus/client/MilvusServiceClientTest.java

@@ -1604,9 +1604,9 @@ class MilvusServiceClientTest {
     }
     }
 
 
     @Test
     @Test
-    void importData() {
+    void bulkload() {
         List<String> files = Collections.singletonList("f1");
         List<String> files = Collections.singletonList("f1");
-        ImportParam param = ImportParam.newBuilder()
+        BulkloadParam param = BulkloadParam.newBuilder()
                 .withCollectionName("collection1")
                 .withCollectionName("collection1")
                 .withPartitionName("partition1")
                 .withPartitionName("partition1")
                 .withRowBased(true)
                 .withRowBased(true)
@@ -1615,23 +1615,23 @@ class MilvusServiceClientTest {
                 .withBucket("myBucket")
                 .withBucket("myBucket")
                 .build();
                 .build();
 
 
-        testFuncByName("importData", param);
+        testFuncByName("bulkload", param);
     }
     }
 
 
     @Test
     @Test
-    void getImportState() {
-        GetImportStateParam param = GetImportStateParam.newBuilder()
+    void getBulkloadState() {
+        GetBulkloadStateParam param = GetBulkloadStateParam.newBuilder()
                 .withTaskID(100L)
                 .withTaskID(100L)
                 .build();
                 .build();
 
 
-        testFuncByName("getImportState", param);
+        testFuncByName("getBulkloadState", param);
     }
     }
 
 
     @Test
     @Test
-    void listImportTasks() {
-        ListImportTasksParam param = ListImportTasksParam.newBuilder().build();
+    void listBulkloadTasks() {
+        ListBulkloadTasksParam param = ListBulkloadTasksParam.newBuilder().build();
 
 
-        testFuncByName("listImportTasks", param);
+        testFuncByName("listBulkloadTasks", param);
     }
     }
 
 
     @Test
     @Test
@@ -2559,7 +2559,7 @@ class MilvusServiceClientTest {
     }
     }
 
 
     @Test
     @Test
-    void testGetImportStateWrapper() {
+    void testGetBulkloadStateWrapper() {
         long count = 1000;
         long count = 1000;
         long id = 88;
         long id = 88;
         ImportState state = ImportState.ImportStarted;
         ImportState state = ImportState.ImportStarted;
@@ -2579,7 +2579,7 @@ class MilvusServiceClientTest {
                         .build())
                         .build())
                 .build();
                 .build();
 
 
-        GetImportStateWrapper wrapper = new GetImportStateWrapper(reso);
+        GetBulkloadStateWrapper wrapper = new GetBulkloadStateWrapper(reso);
         assertEquals(count, wrapper.getImportedCount());
         assertEquals(count, wrapper.getImportedCount());
         assertEquals(1, wrapper.getAutoGeneratedIDs().size());
         assertEquals(1, wrapper.getAutoGeneratedIDs().size());
         assertEquals(id, wrapper.getAutoGeneratedIDs().get(0));
         assertEquals(id, wrapper.getAutoGeneratedIDs().get(0));