Add forcePromote to UpdateReplicateConfigurationRequest and forceDrop to DropRoleRequest (#1790)

Signed-off-by: lentitude2tk <xushuang.hu@zilliz.com>
Co-authored-by: Claude Opus 4.6 (1M context) <noreply@anthropic.com>

Author: lentitude2tk

sdk-core/src/main/java/io/milvus/client/AbstractMilvusGrpcClient.java

@@ -2634,6 +2634,7 @@ public R<RpcStatus> dropRole(DropRoleParam requestParam) {
         try {
             DropRoleRequest request = DropRoleRequest.newBuilder()
                     .setRoleName(requestParam.getRoleName())
+                    .setForceDrop(requestParam.isForceDrop())
                     .build();
 
             Status response = blockingStub().dropRole(request);

sdk-core/src/main/java/io/milvus/param/role/DropRoleParam.java

@@ -25,12 +25,14 @@
 public class DropRoleParam {
 
     private final String roleName;
+    private final boolean forceDrop;
 
     private DropRoleParam(DropRoleParam.Builder builder) {
         if (builder == null) {
             throw new IllegalArgumentException("Builder cannot be null");
         }
         this.roleName = builder.roleName;
+        this.forceDrop = builder.forceDrop;
     }
 
     public static DropRoleParam.Builder newBuilder() {
@@ -41,10 +43,15 @@ public String getRoleName() {
         return roleName;
     }
 
+    public boolean isForceDrop() {
+        return forceDrop;
+    }
+
     @Override
     public String toString() {
         return "DropRoleParam{" +
                 "roleName='" + roleName + '\'' +
+                ", forceDrop=" + forceDrop +
                 '}';
     }
 
@@ -53,6 +60,7 @@ public String toString() {
      */
     public static final class Builder {
         private String roleName;
+        private boolean forceDrop;
 
         private Builder() {
         }
@@ -71,6 +79,17 @@ public DropRoleParam.Builder withRoleName(String roleName) {
             return this;
         }
 
+        /**
+         * Sets the forceDrop flag. If true, the role will be force dropped.
+         *
+         * @param forceDrop forceDrop
+         * @return <code>Builder</code>
+         */
+        public DropRoleParam.Builder withForceDrop(boolean forceDrop) {
+            this.forceDrop = forceDrop;
+            return this;
+        }
+
         /**
          * Verifies parameters and creates a new {@link DropRoleParam} instance.
          *

sdk-core/src/main/java/io/milvus/v2/service/cdc/CDCService.java

@@ -30,6 +30,7 @@ public class CDCService extends BaseService {
     public UpdateReplicateConfigurationResp updateReplicateConfiguration(MilvusServiceGrpc.MilvusServiceBlockingStub blockingStub, UpdateReplicateConfigurationReq requestParam) {
         UpdateReplicateConfigurationRequest request = UpdateReplicateConfigurationRequest.newBuilder()
                 .setReplicateConfiguration(requestParam.getReplicateConfiguration().toGRPC())
+                .setForcePromote(requestParam.isForcePromote())
                 .build();
 
         String title = "UpdateReplicateConfiguration";

sdk-core/src/main/java/io/milvus/v2/service/cdc/request/UpdateReplicateConfigurationReq.java

@@ -21,13 +21,15 @@
 
 public class UpdateReplicateConfigurationReq {
     private ReplicateConfiguration replicateConfiguration;
+    private boolean forcePromote;
 
     public static UpdateReplicateConfigurationReqBuilder builder() {
         return new UpdateReplicateConfigurationReqBuilder();
     }
 
     private UpdateReplicateConfigurationReq(UpdateReplicateConfigurationReqBuilder builder) {
         this.replicateConfiguration = builder.replicateConfiguration;
+        this.forcePromote = builder.forcePromote;
     }
 
     public ReplicateConfiguration getReplicateConfiguration() {
@@ -38,21 +40,36 @@ public void setReplicateConfiguration(ReplicateConfiguration replicateConfigurat
         this.replicateConfiguration = replicateConfiguration;
     }
 
+    public boolean isForcePromote() {
+        return forcePromote;
+    }
+
+    public void setForcePromote(boolean forcePromote) {
+        this.forcePromote = forcePromote;
+    }
+
     @Override
     public String toString() {
         return "UpdateReplicateConfigurationReq{" +
                 "replicateConfiguration=" + replicateConfiguration +
+                ", forcePromote=" + forcePromote +
                 '}';
     }
 
     public static class UpdateReplicateConfigurationReqBuilder {
         private ReplicateConfiguration replicateConfiguration;
+        private boolean forcePromote;
 
         public UpdateReplicateConfigurationReqBuilder replicateConfiguration(ReplicateConfiguration replicateConfiguration) {
             this.replicateConfiguration = replicateConfiguration;
             return this;
         }
 
+        public UpdateReplicateConfigurationReqBuilder forcePromote(boolean forcePromote) {
+            this.forcePromote = forcePromote;
+            return this;
+        }
+
         public UpdateReplicateConfigurationReq build() {
             return new UpdateReplicateConfigurationReq(this);
         }

sdk-core/src/main/java/io/milvus/v2/service/rbac/RBACService.java

@@ -88,6 +88,7 @@ public Void dropRole(MilvusServiceGrpc.MilvusServiceBlockingStub blockingStub, D
         String title = String.format("Drop role: '%s'", request.getRoleName());
         DropRoleRequest dropRoleRequest = DropRoleRequest.newBuilder()
                 .setRoleName(request.getRoleName())
+                .setForceDrop(request.isForceDrop())
                 .build();
         Status status = blockingStub.dropRole(dropRoleRequest);
         rpcUtils.handleResponse(title, status);

sdk-core/src/main/java/io/milvus/v2/service/rbac/request/DropRoleReq.java

@@ -21,9 +21,11 @@
 
 public class DropRoleReq {
     private String roleName;
+    private boolean forceDrop;
 
     private DropRoleReq(DropRoleReqBuilder builder) {
         this.roleName = builder.roleName;
+        this.forceDrop = builder.forceDrop;
     }
 
     public String getRoleName() {
@@ -34,10 +36,19 @@ public void setRoleName(String roleName) {
         this.roleName = roleName;
     }
 
+    public boolean isForceDrop() {
+        return forceDrop;
+    }
+
+    public void setForceDrop(boolean forceDrop) {
+        this.forceDrop = forceDrop;
+    }
+
     @Override
     public String toString() {
         return "DropRoleReq{" +
                 "roleName='" + roleName + '\'' +
+                ", forceDrop=" + forceDrop +
                 '}';
     }
 
@@ -47,6 +58,7 @@ public static DropRoleReqBuilder builder() {
 
     public static class DropRoleReqBuilder {
         private String roleName;
+        private boolean forceDrop;
 
         private DropRoleReqBuilder() {
         }
@@ -56,6 +68,11 @@ public DropRoleReqBuilder roleName(String roleName) {
             return this;
         }
 
+        public DropRoleReqBuilder forceDrop(boolean forceDrop) {
+            this.forceDrop = forceDrop;
+            return this;
+        }
+
         public DropRoleReq build() {
             return new DropRoleReq(this);
         }