2
0
Эх сурвалжийг харах

LoadPartitions support replicas (#1307)

Signed-off-by: yhmo <yihua.mo@zilliz.com>
groot 2 сар өмнө
parent
commit
bd8b543b42

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

@@ -99,6 +99,7 @@ public class PartitionService extends BaseService {
         io.milvus.grpc.LoadPartitionsRequest loadPartitionsRequest = io.milvus.grpc.LoadPartitionsRequest.newBuilder()
                 .setCollectionName(request.getCollectionName())
                 .addAllPartitionNames(request.getPartitionNames())
+                .setReplicaNumber(request.getNumReplicas())
                 .setRefresh(request.getRefresh())
                 .addAllLoadFields(request.getLoadFields())
                 .setSkipLoadDynamicField(request.getSkipLoadDynamicField())

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

@@ -33,6 +33,8 @@ public class LoadPartitionsReq {
     @Builder.Default
     private List<String> partitionNames = new ArrayList<>();
     @Builder.Default
+    private Integer numReplicas = 1;
+    @Builder.Default
     private Boolean refresh = Boolean.FALSE;
     @Builder.Default
     private List<String> loadFields = new ArrayList<>();