浏览代码

Add withPartitionName for DeleteIdsParam (#718)

Signed-off-by: yhmo <yihua.mo@zilliz.com>
groot 1 年之前
父节点
当前提交
f7ad863b49

+ 1 - 0
src/main/java/io/milvus/client/AbstractMilvusGrpcClient.java

@@ -3076,6 +3076,7 @@ public abstract class AbstractMilvusGrpcClient implements MilvusClient {
             String expr = VectorUtils.convertPksExpr(requestParam.getPrimaryIds(), wrapper);
             DeleteParam deleteParam = DeleteParam.newBuilder()
                     .withCollectionName(requestParam.getCollectionName())
+                    .withPartitionName(requestParam.getPartitionName())
                     .withExpr(expr)
                     .build();
             R<MutationResult> resultR = delete(deleteParam);

+ 14 - 1
src/main/java/io/milvus/param/highlevel/dml/DeleteIdsParam.java

@@ -36,12 +36,13 @@ import java.util.List;
 @Getter
 @ToString
 public class DeleteIdsParam {
-
     private final String collectionName;
+    private final String partitionName;
     private final List<?> primaryIds;
 
     private DeleteIdsParam(@NonNull Builder builder) {
         this.collectionName = builder.collectionName;
+        this.partitionName = builder.partitionName;
         this.primaryIds = builder.primaryIds;
     }
 
@@ -54,6 +55,7 @@ public class DeleteIdsParam {
      */
     public static class Builder<T> {
         private String collectionName;
+        private String partitionName = "";
         private List<T> primaryIds = new ArrayList<>();
 
         private Builder() {
@@ -70,6 +72,17 @@ public class DeleteIdsParam {
             return this;
         }
 
+        /**
+         * Sets the partition name (Optional).
+         *
+         * @param partitionName partition name
+         * @return <code>Builder</code>
+         */
+        public Builder withPartitionName(@NonNull String partitionName) {
+            this.partitionName = partitionName;
+            return this;
+        }
+
         /**
          * Specifies primaryId fields. PrimaryIds cannot be empty or null.
          *