LoadPartitions support replicas (#1307)

Signed-off-by: yhmo <[email protected]>
milvus-io · Feb 14, 2025 · bd8b543 · bd8b543
1 parent 93de6d6
commit bd8b543
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/sdk-core/src/main/java/io/milvus/v2/service/partition/PartitionService.java b/sdk-core/src/main/java/io/milvus/v2/service/partition/PartitionService.java
@@ -99,6 +99,7 @@ public Void loadPartitions(MilvusServiceGrpc.MilvusServiceBlockingStub blockingS
         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())

diff --git a/sdk-core/src/main/java/io/milvus/v2/service/partition/request/LoadPartitionsReq.java b/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<>();