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

Update describeAlias to use request (#754)

Signed-off-by: Nian Liu <nian.liu@zilliz.com>
Nian Liu 1 жил өмнө
parent
commit
ea92b224c5

+ 3 - 2
src/main/java/io/milvus/v2/client/MilvusClientV2.java

@@ -22,6 +22,7 @@ import io.milvus.v2.service.rbac.response.DescribeUserResp;
 import io.milvus.v2.service.utility.UtilityService;
 import io.milvus.v2.service.utility.UtilityService;
 import io.milvus.v2.service.utility.request.AlterAliasReq;
 import io.milvus.v2.service.utility.request.AlterAliasReq;
 import io.milvus.v2.service.utility.request.CreateAliasReq;
 import io.milvus.v2.service.utility.request.CreateAliasReq;
+import io.milvus.v2.service.utility.request.DescribeAliasReq;
 import io.milvus.v2.service.utility.request.DropAliasReq;
 import io.milvus.v2.service.utility.request.DropAliasReq;
 import io.milvus.v2.service.utility.response.DescribeAliasResp;
 import io.milvus.v2.service.utility.response.DescribeAliasResp;
 import io.milvus.v2.service.utility.response.ListAliasResp;
 import io.milvus.v2.service.utility.response.ListAliasResp;
@@ -487,8 +488,8 @@ public class MilvusClientV2 {
      *
      *
      * @return DescribeAliasResp
      * @return DescribeAliasResp
      */
      */
-    public DescribeAliasResp describeAlias(String alias) {
-        return utilityService.describeAlias(this.blockingStub, alias);
+    public DescribeAliasResp describeAlias(DescribeAliasReq request) {
+        return utilityService.describeAlias(this.blockingStub, request);
     }
     }
 
 
     /**
     /**

+ 4 - 7
src/main/java/io/milvus/v2/service/utility/UtilityService.java

@@ -5,10 +5,7 @@ import io.milvus.grpc.MilvusServiceGrpc;
 import io.milvus.param.R;
 import io.milvus.param.R;
 import io.milvus.param.RpcStatus;
 import io.milvus.param.RpcStatus;
 import io.milvus.v2.service.BaseService;
 import io.milvus.v2.service.BaseService;
-import io.milvus.v2.service.utility.request.AlterAliasReq;
-import io.milvus.v2.service.utility.request.CreateAliasReq;
-import io.milvus.v2.service.utility.request.DropAliasReq;
-import io.milvus.v2.service.utility.request.FlushReq;
+import io.milvus.v2.service.utility.request.*;
 import io.milvus.v2.service.utility.response.DescribeAliasResp;
 import io.milvus.v2.service.utility.response.DescribeAliasResp;
 import io.milvus.v2.service.utility.response.ListAliasResp;
 import io.milvus.v2.service.utility.response.ListAliasResp;
 
 
@@ -52,10 +49,10 @@ public class UtilityService extends BaseService {
         rpcUtils.handleResponse(title, status);
         rpcUtils.handleResponse(title, status);
     }
     }
 
 
-    public DescribeAliasResp describeAlias(MilvusServiceGrpc.MilvusServiceBlockingStub blockingStub, String alias) {
-        String title = String.format("Describe alias %s", alias);
+    public DescribeAliasResp describeAlias(MilvusServiceGrpc.MilvusServiceBlockingStub blockingStub, DescribeAliasReq request) {
+        String title = String.format("Describe alias %s", request.getAlias());
         io.milvus.grpc.DescribeAliasRequest describeAliasRequest = io.milvus.grpc.DescribeAliasRequest.newBuilder()
         io.milvus.grpc.DescribeAliasRequest describeAliasRequest = io.milvus.grpc.DescribeAliasRequest.newBuilder()
-                .setAlias(alias)
+                .setAlias(request.getAlias())
                 .build();
                 .build();
         io.milvus.grpc.DescribeAliasResponse response = blockingStub.describeAlias(describeAliasRequest);
         io.milvus.grpc.DescribeAliasResponse response = blockingStub.describeAlias(describeAliasRequest);
 
 

+ 10 - 0
src/main/java/io/milvus/v2/service/utility/request/DescribeAliasReq.java

@@ -0,0 +1,10 @@
+package io.milvus.v2.service.utility.request;
+
+import lombok.Data;
+import lombok.experimental.SuperBuilder;
+
+@Data
+@SuperBuilder
+public class DescribeAliasReq {
+    private String alias;
+}

+ 5 - 1
src/test/java/io/milvus/v2/service/utility/UtilityTest.java

@@ -3,6 +3,7 @@ package io.milvus.v2.service.utility;
 import io.milvus.v2.BaseTest;
 import io.milvus.v2.BaseTest;
 import io.milvus.v2.service.utility.request.AlterAliasReq;
 import io.milvus.v2.service.utility.request.AlterAliasReq;
 import io.milvus.v2.service.utility.request.CreateAliasReq;
 import io.milvus.v2.service.utility.request.CreateAliasReq;
+import io.milvus.v2.service.utility.request.DescribeAliasReq;
 import io.milvus.v2.service.utility.request.DropAliasReq;
 import io.milvus.v2.service.utility.request.DropAliasReq;
 import io.milvus.v2.service.utility.response.DescribeAliasResp;
 import io.milvus.v2.service.utility.response.DescribeAliasResp;
 import io.milvus.v2.service.utility.response.ListAliasResp;
 import io.milvus.v2.service.utility.response.ListAliasResp;
@@ -40,7 +41,10 @@ class UtilityTest extends BaseTest {
 
 
     @Test
     @Test
     void describeAlias() {
     void describeAlias() {
-        DescribeAliasResp statusR = client_v2.describeAlias("test_alias");
+        DescribeAliasReq req = DescribeAliasReq.builder()
+                .alias("test_alias")
+                .build();
+        DescribeAliasResp statusR = client_v2.describeAlias(req);
     }
     }
     @Test
     @Test
     void listAliases() {
     void listAliases() {