Bläddra i källkod

Support resource groups for loadCollection/loadPartition (#1248)

Signed-off-by: yhmo <yihua.mo@zilliz.com>
groot 4 månader sedan
förälder
incheckning
843a1d2a45

+ 1 - 0
sdk-core/src/main/java/io/milvus/v2/service/collection/CollectionService.java

@@ -258,6 +258,7 @@ public class CollectionService extends BaseService {
                 .setRefresh(request.getRefresh())
                 .addAllLoadFields(request.getLoadFields())
                 .setSkipLoadDynamicField(request.getSkipLoadDynamicField())
+                .addAllResourceGroups(request.getResourceGroups())
                 .build();
         Status status = blockingStub.loadCollection(loadCollectionRequest);
         rpcUtils.handleResponse(title, status);

+ 2 - 0
sdk-core/src/main/java/io/milvus/v2/service/collection/request/LoadCollectionReq.java

@@ -42,4 +42,6 @@ public class LoadCollectionReq {
     private List<String> loadFields = new ArrayList<>();
     @Builder.Default
     private Boolean skipLoadDynamicField = Boolean.FALSE;
+    @Builder.Default
+    private List<String> resourceGroups = new ArrayList<>();
 }

+ 1 - 0
sdk-core/src/main/java/io/milvus/v2/service/partition/PartitionService.java

@@ -102,6 +102,7 @@ public class PartitionService extends BaseService {
                 .setRefresh(request.getRefresh())
                 .addAllLoadFields(request.getLoadFields())
                 .setSkipLoadDynamicField(request.getSkipLoadDynamicField())
+                .addAllResourceGroups(request.getResourceGroups())
                 .build();
         Status status = blockingStub.loadPartitions(loadPartitionsRequest);
         rpcUtils.handleResponse(title, status);

+ 2 - 0
sdk-core/src/main/java/io/milvus/v2/service/partition/request/LoadPartitionsReq.java

@@ -38,4 +38,6 @@ public class LoadPartitionsReq {
     private List<String> loadFields = new ArrayList<>();
     @Builder.Default
     private Boolean skipLoadDynamicField = Boolean.FALSE;
+    @Builder.Default
+    private List<String> resourceGroups = new ArrayList<>();
 }