Retry for V2 (#1007)

Signed-off-by: yhmo <yihua.mo@zilliz.com>

docker-compose.yml

@@ -32,7 +32,7 @@ services:
 
   standalone:
     container_name: milvus-javasdk-test-standalone
-    image: milvusdb/milvus:v2.3.13
+    image: milvusdb/milvus:v2.3.20
     command: ["milvus", "run", "standalone"]
     environment:
       ETCD_ENDPOINTS: etcd:2379
@@ -77,7 +77,7 @@ services:
 
   standaloneslave:
     container_name: milvus-javasdk-test-slave-standalone
-    image: milvusdb/milvus:v2.3.13
+    image: milvusdb/milvus:v2.3.20
     command: ["milvus", "run", "standalone"]
     environment:
       ETCD_ENDPOINTS: etcdslave:2379

examples/main/java/io/milvus/v2/IteratorExample.java

@@ -64,6 +64,12 @@ public class IteratorExample {
                 .metricType(IndexParam.MetricType.L2)
                 .build());
 
+        // drop collection if exists
+        client.dropCollection(DropCollectionReq.builder()
+                .collectionName(COLLECTION_NAME)
+                .build());
+
+        // create collection
         CreateCollectionReq requestCreate = CreateCollectionReq.builder()
                 .collectionName(COLLECTION_NAME)
                 .collectionSchema(collectionSchema)
@@ -97,7 +103,7 @@ public class IteratorExample {
                 .consistencyLevel(ConsistencyLevel.STRONG)
                 .build());
         List<QueryResp.QueryResult> queryResults = queryResp.getQueryResults();
-        System.out.printf("Inserted row count: %d\n", queryResults.size());
+        System.out.printf("Inserted row count: %d\n", queryResults.get(0).getEntity().get("count(*)"));
 
         // search iterator
         SearchIterator searchIterator = client.searchIterator(SearchIteratorReq.builder()
@@ -127,7 +133,7 @@ public class IteratorExample {
                 counter++;
             }
         }
-        System.out.println(String.format("%d search results returned\n", counter));
+        System.out.printf("%d search results returned\n%n", counter);
 
         // query iterator
         QueryIterator queryIterator = client.queryIterator(QueryIteratorReq.builder()
@@ -154,8 +160,6 @@ public class IteratorExample {
                 counter++;
             }
         }
-        System.out.println(String.format("%d query results returned", counter));
-
-        client.dropCollection(DropCollectionReq.builder().collectionName(COLLECTION_NAME).build());
+        System.out.printf("%d query results returned%n", counter);
     }
 }

src/main/java/io/milvus/client/MilvusServiceClient.java

@@ -323,7 +323,8 @@ public class MilvusServiceClient extends AbstractMilvusGrpcClient {
                             || code == Status.UNAUTHENTICATED.getCode()
                             || code == Status.INVALID_ARGUMENT.getCode()
                             || code == Status.ALREADY_EXISTS.getCode()
-                            || code == Status.RESOURCE_EXHAUSTED.getCode()) {
+                            || code == Status.RESOURCE_EXHAUSTED.getCode()
+                            || code == Status.UNIMPLEMENTED.getCode()) {
                         return resp;
                     }
 
@@ -342,7 +343,7 @@ public class MilvusServiceClient extends AbstractMilvusGrpcClient {
 
                     // for server-side returned error, only retry for rate limit
                     // in new error codes of v2.3, rate limit error value is 8
-                    if (retryParam.isRetryOnRateLimie() &&
+                    if (retryParam.isRetryOnRateLimit() &&
                             (serverException.getCompatibleCode() == ErrorCode.RateLimit ||
                                     serverException.getStatus() == 8)) {
                     } else {

src/main/java/io/milvus/param/RetryParam.java

@@ -37,14 +37,14 @@ public class RetryParam {
     private long initialBackOffMs;
     private long maxBackOffMs;
     private int backOffMultiplier;
-    private boolean retryOnRateLimie;
+    private boolean retryOnRateLimit;
 
     protected RetryParam(@NonNull Builder builder) {
         this.maxRetryTimes = builder.maxRetryTimes;
         this.initialBackOffMs = builder.initialBackOffMs;
         this.maxBackOffMs = builder.maxBackOffMs;
         this.backOffMultiplier = builder.backOffMultiplier;
-        this.retryOnRateLimie = builder.retryOnRateLimie;
+        this.retryOnRateLimit = builder.retryOnRateLimit;
     }
 
     public static Builder newBuilder() {
@@ -60,7 +60,7 @@ public class RetryParam {
         private long initialBackOffMs = 10;
         private long maxBackOffMs = 3000;
         private int backOffMultiplier = 3;
-        private boolean retryOnRateLimie = true;
+        private boolean retryOnRateLimit = true;
 
         protected Builder() {
         }
@@ -112,11 +112,11 @@ public class RetryParam {
         /**
          * Sets whether to retry when the returned error is rate limit.Default value is true.
          *
-         * @param retryOnRateLimie whether to retry when the returned error is rate limit
+         * @param retryOnRateLimit whether to retry when the returned error is rate limit
          * @return <code>Builder</code>
          */
-        public Builder withRetryOnRateLimie(boolean retryOnRateLimie) {
-            this.retryOnRateLimie = retryOnRateLimie;
+        public Builder withRetryOnRateLimit(boolean retryOnRateLimit) {
+            this.retryOnRateLimit = retryOnRateLimit;
             return this;
         }
 

src/main/java/io/milvus/v2/client/MilvusClientV2.java

@@ -20,20 +20,20 @@
 package io.milvus.v2.client;
 
 import io.grpc.ManagedChannel;
+import io.grpc.Status;
+import io.grpc.StatusRuntimeException;
 import io.milvus.grpc.MilvusServiceGrpc;
 import io.milvus.orm.iterator.QueryIterator;
 import io.milvus.orm.iterator.SearchIterator;
+
+import io.milvus.v2.exception.ErrorCode;
+import io.milvus.v2.exception.MilvusClientException;
 import io.milvus.v2.service.collection.CollectionService;
 import io.milvus.v2.service.collection.request.*;
-import io.milvus.v2.service.collection.response.DescribeCollectionResp;
-import io.milvus.v2.service.collection.response.GetCollectionStatsResp;
-import io.milvus.v2.service.collection.response.ListCollectionsResp;
+import io.milvus.v2.service.collection.response.*;
 import io.milvus.v2.service.index.IndexService;
-import io.milvus.v2.service.index.request.CreateIndexReq;
-import io.milvus.v2.service.index.request.DescribeIndexReq;
-import io.milvus.v2.service.index.request.DropIndexReq;
-import io.milvus.v2.service.index.request.ListIndexesReq;
-import io.milvus.v2.service.index.response.DescribeIndexResp;
+import io.milvus.v2.service.index.request.*;
+import io.milvus.v2.service.index.response.*;
 import io.milvus.v2.service.partition.PartitionService;
 import io.milvus.v2.service.partition.request.*;
 import io.milvus.v2.service.rbac.RoleService;
@@ -55,6 +55,7 @@ import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
 import java.util.List;
+import java.util.concurrent.Callable;
 import java.util.concurrent.TimeUnit;
 
 public class MilvusClientV2 {
@@ -71,6 +72,7 @@ public class MilvusClientV2 {
     private final RoleService roleService = new RoleService();
     private final UtilityService utilityService = new UtilityService();
     private ConnectConfig connectConfig;
+    private RetryConfig retryConfig = RetryConfig.builder().build();
 
     /**
      * Creates a Milvus client instance.
@@ -111,6 +113,116 @@ public class MilvusClientV2 {
         }
     }
 
+    public void retryConfig(RetryConfig retryConfig) {
+        this.retryConfig = retryConfig;
+    }
+
+    private <T> T retry(Callable<T> callable) {
+        int maxRetryTimes = retryConfig.getMaxRetryTimes();
+        // no retry, direct call the method
+        if (maxRetryTimes <= 1) {
+            try {
+                return callable.call();
+            } catch (StatusRuntimeException e) {
+                throw new MilvusClientException(ErrorCode.RPC_ERROR, e.getMessage()); // rpc error
+            } catch (MilvusClientException e) {
+                throw e; // server error or client error
+            } catch (Exception e) {
+                throw new MilvusClientException(ErrorCode.CLIENT_ERROR, e.getMessage()); // others error treated as client error
+            }
+        }
+
+        // method to check timeout
+        long begin = System.currentTimeMillis();
+        long maxRetryTimeoutMs = retryConfig.getMaxRetryTimeoutMs();
+        Callable<Boolean> timeoutChecker = ()->{
+            long current = System.currentTimeMillis();
+            long cost = (current - begin);
+            if (maxRetryTimeoutMs > 0 && cost >= maxRetryTimeoutMs) {
+                return Boolean.TRUE;
+            }
+            return Boolean.FALSE;
+        };
+
+        // retry within timeout
+        long retryIntervalMs = retryConfig.getInitialBackOffMs();
+        for (int k = 1; k <= maxRetryTimes; k++) {
+            try {
+                return callable.call();
+            } catch (StatusRuntimeException e) {
+                Status.Code code = e.getStatus().getCode();
+                if (code == Status.DEADLINE_EXCEEDED.getCode()
+                        || code == Status.PERMISSION_DENIED.getCode()
+                        || code == Status.UNAUTHENTICATED.getCode()
+                        || code == Status.INVALID_ARGUMENT.getCode()
+                        || code == Status.ALREADY_EXISTS.getCode()
+                        || code == Status.RESOURCE_EXHAUSTED.getCode()
+                        || code == Status.UNIMPLEMENTED.getCode()) {
+                    String msg = String.format("Encounter rpc error that cannot be retried, reason: %s", e.getMessage());
+                    logger.error(msg);
+                    throw new MilvusClientException(ErrorCode.RPC_ERROR, msg); // throw rpc error
+                }
+
+                try {
+                    if (timeoutChecker.call() == Boolean.TRUE) {
+                        String msg = String.format("Retry timeout: %dms, maxRetry:%d, retries: %d, reason: %s",
+                                maxRetryTimeoutMs, maxRetryTimes, k, e.getMessage());
+                        logger.warn(msg);
+                        throw new MilvusClientException(ErrorCode.TIMEOUT, msg); // exit retry for timeout
+                    }
+                } catch (Exception ignored) {
+                }
+            } catch (MilvusClientException e) {
+                try {
+                    if (timeoutChecker.call() == Boolean.TRUE) {
+                        String msg = String.format("Retry timeout: %dms, maxRetry:%d, retries: %d, reason: %s",
+                                maxRetryTimeoutMs, maxRetryTimes, k, e.getMessage());
+                        logger.warn(msg);
+                        throw new MilvusClientException(ErrorCode.TIMEOUT, msg); // exit retry for timeout
+                    }
+                } catch (Exception ignored) {
+                }
+
+                // for server-side returned error, only retry for rate limit
+                // in new error codes of v2.3, rate limit error value is 8
+                if (retryConfig.isRetryOnRateLimit() &&
+                        (e.getLegacyServerCode() == io.milvus.grpc.ErrorCode.RateLimit.getNumber() ||
+                                e.getServerErrCode() == 8)) {
+                    // cannot be retried
+                } else {
+                    throw e; // exit retry, throw the error
+                }
+            } catch (Exception e) {
+                throw new MilvusClientException(ErrorCode.CLIENT_ERROR, e.getMessage()); // others error treated as client error
+            }
+
+            try {
+                if (k >= maxRetryTimes) {
+                    // finish retry loop, return the response of the last retry
+                    String msg = String.format("Finish %d retry times, stop retry", maxRetryTimes);
+                    logger.warn(msg);
+                    throw new MilvusClientException(ErrorCode.TIMEOUT, msg); // exceed max time, exit retry
+                } else {
+                    // sleep for interval
+                    // print log, follow the pymilvus logic
+                    if (k > 3) {
+                        logger.warn(String.format("Retry(%d) with interval %dms", k, retryIntervalMs));
+                    }
+                    TimeUnit.MILLISECONDS.sleep(retryIntervalMs);
+                }
+
+                // reset the next interval value
+                retryIntervalMs = retryIntervalMs*retryConfig.getBackOffMultiplier();
+                if (retryIntervalMs > retryConfig.getMaxBackOffMs()) {
+                    retryIntervalMs = retryConfig.getMaxBackOffMs();
+                }
+            } catch (Exception ignored) {
+            }
+        }
+
+        return null;
+    }
+
     /**
      * use Database
      * @param dbName databaseName
@@ -133,7 +245,7 @@ public class MilvusClientV2 {
      * @param request create collection request
      */
     public void createCollection(CreateCollectionReq request) {
-        collectionService.createCollection(this.blockingStub, request);
+        retry(()-> collectionService.createCollection(this.blockingStub, request));
     }
 
     /**
@@ -150,7 +262,7 @@ public class MilvusClientV2 {
      * @return List of String collection names
      */
     public ListCollectionsResp listCollections() {
-        return collectionService.listCollections(this.blockingStub);
+        return retry(()-> collectionService.listCollections(this.blockingStub));
     }
 
     /**
@@ -159,7 +271,7 @@ public class MilvusClientV2 {
      * @param request drop collection request
      */
     public void dropCollection(DropCollectionReq request) {
-        collectionService.dropCollection(this.blockingStub, request);
+        retry(()-> collectionService.dropCollection(this.blockingStub, request));
     }
     /**
      * Checks whether a collection exists in Milvus.
@@ -168,7 +280,7 @@ public class MilvusClientV2 {
      * @return Boolean
      */
     public Boolean hasCollection(HasCollectionReq request) {
-        return collectionService.hasCollection(this.blockingStub, request);
+        return retry(()-> collectionService.hasCollection(this.blockingStub, request));
     }
     /**
      * Gets the collection info in Milvus.
@@ -177,7 +289,7 @@ public class MilvusClientV2 {
      * @return DescribeCollectionResp
      */
     public DescribeCollectionResp describeCollection(DescribeCollectionReq request) {
-        return collectionService.describeCollection(this.blockingStub, request);
+        return retry(()-> collectionService.describeCollection(this.blockingStub, request));
     }
     /**
      * get collection stats for a collection in Milvus.
@@ -186,7 +298,7 @@ public class MilvusClientV2 {
      * @return GetCollectionStatsResp
      */
     public GetCollectionStatsResp getCollectionStats(GetCollectionStatsReq request) {
-        return collectionService.getCollectionStats(this.blockingStub, request);
+        return retry(()-> collectionService.getCollectionStats(this.blockingStub, request));
     }
     /**
      * rename collection in a collection in Milvus.
@@ -194,7 +306,7 @@ public class MilvusClientV2 {
      * @param request rename collection request
      */
     public void renameCollection(RenameCollectionReq request) {
-        collectionService.renameCollection(this.blockingStub, request);
+        retry(()-> collectionService.renameCollection(this.blockingStub, request));
     }
     /**
      * Loads a collection into memory in Milvus.
@@ -202,7 +314,7 @@ public class MilvusClientV2 {
      * @param request load collection request
      */
     public void loadCollection(LoadCollectionReq request) {
-        collectionService.loadCollection(this.blockingStub, request);
+        retry(()-> collectionService.loadCollection(this.blockingStub, request));
     }
     /**
      * Releases a collection from memory in Milvus.
@@ -210,7 +322,7 @@ public class MilvusClientV2 {
      * @param request release collection request
      */
     public void releaseCollection(ReleaseCollectionReq request) {
-        collectionService.releaseCollection(this.blockingStub, request);
+        retry(()-> collectionService.releaseCollection(this.blockingStub, request));
     }
     /**
      * Checks whether a collection is loaded in Milvus.
@@ -219,7 +331,7 @@ public class MilvusClientV2 {
      * @return Boolean
      */
     public Boolean getLoadState(GetLoadStateReq request) {
-        return collectionService.getLoadState(this.blockingStub, request);
+        return retry(()->collectionService.getLoadState(this.blockingStub, request));
     }
 
     //Index Operations
@@ -229,7 +341,7 @@ public class MilvusClientV2 {
      * @param request create index request
      */
     public void createIndex(CreateIndexReq request) {
-        indexService.createIndex(this.blockingStub, request);
+        retry(()->indexService.createIndex(this.blockingStub, request));
     }
     /**
      * Drops an index for a specified field in a collection in Milvus.
@@ -237,7 +349,7 @@ public class MilvusClientV2 {
      * @param request drop index request
      */
     public void dropIndex(DropIndexReq request) {
-        indexService.dropIndex(this.blockingStub, request);
+        retry(()->indexService.dropIndex(this.blockingStub, request));
     }
     /**
      * Checks whether an index exists for a specified field in a collection in Milvus.
@@ -246,7 +358,7 @@ public class MilvusClientV2 {
      * @return DescribeIndexResp
      */
     public DescribeIndexResp describeIndex(DescribeIndexReq request) {
-        return indexService.describeIndex(this.blockingStub, request);
+        return retry(()->indexService.describeIndex(this.blockingStub, request));
     }
 
     /**
@@ -256,7 +368,7 @@ public class MilvusClientV2 {
      * @return List of String indexes names
      */
     public List<String> listIndexes(ListIndexesReq request) {
-        return indexService.listIndexes(this.blockingStub, request);
+        return retry(()->indexService.listIndexes(this.blockingStub, request));
     }
     // Vector Operations
 
@@ -267,7 +379,7 @@ public class MilvusClientV2 {
      * @return InsertResp
      */
     public InsertResp insert(InsertReq request) {
-        return vectorService.insert(this.blockingStub, request);
+        return retry(()->vectorService.insert(this.blockingStub, request));
     }
     /**
      * Upsert vectors into a collection in Milvus.
@@ -276,7 +388,7 @@ public class MilvusClientV2 {
      * @return UpsertResp
      */
     public UpsertResp upsert(UpsertReq request) {
-        return vectorService.upsert(this.blockingStub, request);
+        return retry(()->vectorService.upsert(this.blockingStub, request));
     }
     /**
      * Deletes vectors in a collection in Milvus.
@@ -285,7 +397,7 @@ public class MilvusClientV2 {
      * @return DeleteResp
      */
     public DeleteResp delete(DeleteReq request) {
-        return vectorService.delete(this.blockingStub, request);
+        return retry(()->vectorService.delete(this.blockingStub, request));
     }
     /**
      * Gets vectors in a collection in Milvus.
@@ -294,7 +406,7 @@ public class MilvusClientV2 {
      * @return GetResp
      */
     public GetResp get(GetReq request) {
-        return vectorService.get(this.blockingStub, request);
+        return retry(()->vectorService.get(this.blockingStub, request));
     }
 
     /**
@@ -304,7 +416,7 @@ public class MilvusClientV2 {
      * @return QueryResp
      */
     public QueryResp query(QueryReq request) {
-        return vectorService.query(this.blockingStub, request);
+        return retry(()->vectorService.query(this.blockingStub, request));
     }
     /**
      * Searches vectors in a collection in Milvus.
@@ -313,7 +425,7 @@ public class MilvusClientV2 {
      * @return SearchResp
      */
     public SearchResp search(SearchReq request) {
-        return vectorService.search(this.blockingStub, request);
+        return retry(()->vectorService.search(this.blockingStub, request));
     }
 
     /**
@@ -324,7 +436,7 @@ public class MilvusClientV2 {
      * @return {status:result code,data: QueryIterator}
      */
     public QueryIterator queryIterator(QueryIteratorReq request) {
-        return vectorService.queryIterator(this.blockingStub, request);
+        return retry(()->vectorService.queryIterator(this.blockingStub, request));
     }
 
     /**
@@ -334,7 +446,7 @@ public class MilvusClientV2 {
      * @return {status:result code, data: SearchIterator}
      */
     public SearchIterator searchIterator(SearchIteratorReq request) {
-        return vectorService.searchIterator(this.blockingStub, request);
+        return retry(()->vectorService.searchIterator(this.blockingStub, request));
     }
 
     // Partition Operations
@@ -344,7 +456,7 @@ public class MilvusClientV2 {
      * @param request create partition request
      */
     public void createPartition(CreatePartitionReq request) {
-        partitionService.createPartition(this.blockingStub, request);
+        retry(()->partitionService.createPartition(this.blockingStub, request));
     }
 
     /**
@@ -353,7 +465,7 @@ public class MilvusClientV2 {
      * @param request drop partition request
      */
     public void dropPartition(DropPartitionReq request) {
-        partitionService.dropPartition(this.blockingStub, request);
+        retry(()->partitionService.dropPartition(this.blockingStub, request));
     }
 
     /**
@@ -363,7 +475,7 @@ public class MilvusClientV2 {
      * @return Boolean
      */
     public Boolean hasPartition(HasPartitionReq request) {
-        return partitionService.hasPartition(this.blockingStub, request);
+        return retry(()->partitionService.hasPartition(this.blockingStub, request));
     }
 
     /**
@@ -373,7 +485,7 @@ public class MilvusClientV2 {
      * @return List of String partition names
      */
     public List<String> listPartitions(ListPartitionsReq request) {
-        return partitionService.listPartitions(this.blockingStub, request);
+        return retry(()->partitionService.listPartitions(this.blockingStub, request));
     }
 
     /**
@@ -382,7 +494,7 @@ public class MilvusClientV2 {
      * @param request load partitions request
      */
     public void loadPartitions(LoadPartitionsReq request) {
-        partitionService.loadPartitions(this.blockingStub, request);
+        retry(()->partitionService.loadPartitions(this.blockingStub, request));
     }
     /**
      * Releases partitions in a collection in Milvus.
@@ -390,7 +502,7 @@ public class MilvusClientV2 {
      * @param request release partitions request
      */
     public void releasePartitions(ReleasePartitionsReq request) {
-        partitionService.releasePartitions(this.blockingStub, request);
+        retry(()->partitionService.releasePartitions(this.blockingStub, request));
     }
     // rbac operations
     // user operations
@@ -400,7 +512,7 @@ public class MilvusClientV2 {
      * @return List of String usernames
      */
     public List<String> listUsers() {
-        return userService.listUsers(this.blockingStub);
+        return retry(()->userService.listUsers(this.blockingStub));
     }
     /**
      * describe user
@@ -409,7 +521,7 @@ public class MilvusClientV2 {
      * @return DescribeUserResp
      */
     public DescribeUserResp describeUser(DescribeUserReq request) {
-        return userService.describeUser(this.blockingStub, request);
+        return retry(()->userService.describeUser(this.blockingStub, request));
     }
     /**
      * create user
@@ -417,7 +529,7 @@ public class MilvusClientV2 {
      * @param request create user request
      */
     public void createUser(CreateUserReq request) {
-        userService.createUser(this.blockingStub, request);
+        retry(()->userService.createUser(this.blockingStub, request));
     }
     /**
      * change password
@@ -425,7 +537,7 @@ public class MilvusClientV2 {
      * @param request change password request
      */
     public void updatePassword(UpdatePasswordReq request) {
-        userService.updatePassword(this.blockingStub, request);
+        retry(()->userService.updatePassword(this.blockingStub, request));
     }
     /**
      * drop user
@@ -433,7 +545,7 @@ public class MilvusClientV2 {
      * @param request drop user request
      */
     public void dropUser(DropUserReq request) {
-        userService.dropUser(this.blockingStub, request);
+        retry(()->userService.dropUser(this.blockingStub, request));
     }
     // role operations
     /**
@@ -442,7 +554,7 @@ public class MilvusClientV2 {
      * @return List of String role names
      */
     public List<String> listRoles() {
-        return roleService.listRoles(this.blockingStub);
+        return retry(()->roleService.listRoles(this.blockingStub));
     }
     /**
      * describe role
@@ -451,7 +563,7 @@ public class MilvusClientV2 {
      * @return DescribeRoleResp
      */
     public DescribeRoleResp describeRole(DescribeRoleReq request) {
-        return roleService.describeRole(this.blockingStub, request);
+        return retry(()->roleService.describeRole(this.blockingStub, request));
     }
     /**
      * create role
@@ -459,7 +571,7 @@ public class MilvusClientV2 {
      * @param request create role request
      */
     public void createRole(CreateRoleReq request) {
-        roleService.createRole(this.blockingStub, request);
+        retry(()->roleService.createRole(this.blockingStub, request));
     }
     /**
      * drop role
@@ -467,7 +579,7 @@ public class MilvusClientV2 {
      * @param request drop role request
      */
     public void dropRole(DropRoleReq request) {
-        roleService.dropRole(this.blockingStub, request);
+        retry(()->roleService.dropRole(this.blockingStub, request));
     }
     /**
      * grant privilege
@@ -475,7 +587,7 @@ public class MilvusClientV2 {
      * @param request grant privilege request
      */
     public void grantPrivilege(GrantPrivilegeReq request) {
-        roleService.grantPrivilege(this.blockingStub, request);
+        retry(()->roleService.grantPrivilege(this.blockingStub, request));
     }
     /**
      * revoke privilege
@@ -483,7 +595,7 @@ public class MilvusClientV2 {
      * @param request revoke privilege request
      */
     public void revokePrivilege(RevokePrivilegeReq request) {
-        roleService.revokePrivilege(this.blockingStub, request);
+        retry(()->roleService.revokePrivilege(this.blockingStub, request));
     }
     /**
      * grant role
@@ -491,7 +603,7 @@ public class MilvusClientV2 {
      * @param request grant role request
      */
     public void grantRole(GrantRoleReq request) {
-        roleService.grantRole(this.blockingStub, request);
+        retry(()->roleService.grantRole(this.blockingStub, request));
     }
     /**
      * revoke role
@@ -499,7 +611,7 @@ public class MilvusClientV2 {
      * @param request revoke role request
      */
     public void revokeRole(RevokeRoleReq request) {
-        roleService.revokeRole(this.blockingStub, request);
+        retry(()->roleService.revokeRole(this.blockingStub, request));
     }
 
     // Utility Operations
@@ -510,7 +622,7 @@ public class MilvusClientV2 {
      * @param request create alias request
      */
     public void createAlias(CreateAliasReq request) {
-        utilityService.createAlias(this.blockingStub, request);
+        retry(()->utilityService.createAlias(this.blockingStub, request));
     }
     /**
      * drop aliases
@@ -518,7 +630,7 @@ public class MilvusClientV2 {
      * @param request drop alias request
      */
     public void dropAlias(DropAliasReq request) {
-        utilityService.dropAlias(this.blockingStub, request);
+        retry(()->utilityService.dropAlias(this.blockingStub, request));
     }
     /**
      * alter aliases
@@ -526,7 +638,7 @@ public class MilvusClientV2 {
      * @param request alter alias request
      */
     public void alterAlias(AlterAliasReq request) {
-        utilityService.alterAlias(this.blockingStub, request);
+        retry(()->utilityService.alterAlias(this.blockingStub, request));
     }
     /**
      * list aliases
@@ -535,7 +647,7 @@ public class MilvusClientV2 {
      * @return List of String aliases names
      */
     public ListAliasResp listAliases(ListAliasesReq request) {
-        return utilityService.listAliases(this.blockingStub, request);
+        return retry(()->utilityService.listAliases(this.blockingStub, request));
     }
     /**
      * describe aliases
@@ -544,7 +656,7 @@ public class MilvusClientV2 {
      * @return DescribeAliasResp
      */
     public DescribeAliasResp describeAlias(DescribeAliasReq request) {
-        return utilityService.describeAlias(this.blockingStub, request);
+        return retry(()->utilityService.describeAlias(this.blockingStub, request));
     }
 
     /**

src/main/java/io/milvus/v2/client/RetryConfig.java

@@ -0,0 +1,41 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements.  See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership.  The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License.  You may obtain a copy of the License at
+ *
+ *   http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the License is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ * KIND, either express or implied.  See the License for the
+ * specific language governing permissions and limitations
+ * under the License.
+ */
+
+package io.milvus.v2.client;
+
+import lombok.Builder;
+import lombok.Data;
+import lombok.experimental.SuperBuilder;
+
+@Data
+@SuperBuilder
+public class RetryConfig {
+    @Builder.Default
+    private int maxRetryTimes = 75;
+    @Builder.Default
+    private long initialBackOffMs = 10;
+    @Builder.Default
+    private long maxBackOffMs = 3000;
+    @Builder.Default
+    private int backOffMultiplier = 3;
+    @Builder.Default
+    private boolean retryOnRateLimit = true;
+    @Builder.Default
+    private long maxRetryTimeoutMs = 0;
+}

src/main/java/io/milvus/v2/exception/ErrorCode.java

@@ -27,7 +27,9 @@ public enum ErrorCode {
     COLLECTION_NOT_FOUND(1),
     SERVER_ERROR(2),
     INVALID_PARAMS(3),
-    CLIENT_ERROR(4);
+    CLIENT_ERROR(4),
+    RPC_ERROR(5),
+    TIMEOUT(6);
 
     private final int code;
 

src/main/java/io/milvus/v2/exception/MilvusClientException.java

@@ -26,10 +26,18 @@ public class MilvusClientException extends RuntimeException {
 
     private final ErrorCode errorCode;
 
+    private int serverErrCode = 0;
+    private int legacyServerCode = 0;
+
     public MilvusClientException(ErrorCode errorCode, String message) {
         super(message);
         this.errorCode = errorCode;
     }
 
-
+    public MilvusClientException(ErrorCode errorCode, String message, int serverErrCode, int legacyServerCode) {
+        super(message);
+        this.errorCode = errorCode;
+        this.serverErrCode = serverErrCode;
+        this.legacyServerCode = legacyServerCode;
+    }
 }

src/main/java/io/milvus/v2/service/collection/CollectionService.java

@@ -38,11 +38,11 @@ import java.util.Collections;
 public class CollectionService extends BaseService {
     public IndexService indexService = new IndexService();
 
-    public void createCollection(MilvusServiceGrpc.MilvusServiceBlockingStub blockingStub, CreateCollectionReq request) {
+    public Void createCollection(MilvusServiceGrpc.MilvusServiceBlockingStub blockingStub, CreateCollectionReq request) {
         if (request.getCollectionSchema() != null) {
             //create collections with schema
             createCollectionWithSchema(blockingStub, request);
-            return;
+            return null;
         }
 
         if (request.getDimension() == null) {
@@ -102,9 +102,10 @@ public class CollectionService extends BaseService {
         } catch (Exception e) {
             throw new MilvusClientException(ErrorCode.SERVER_ERROR, "Load collection failed" + e.getMessage());
         }
+        return null;
     }
 
-    public void createCollectionWithSchema(MilvusServiceGrpc.MilvusServiceBlockingStub blockingStub, CreateCollectionReq request) {
+    public Void createCollectionWithSchema(MilvusServiceGrpc.MilvusServiceBlockingStub blockingStub, CreateCollectionReq request) {
         String title = String.format("CreateCollectionRequest collectionName:%s", request.getCollectionName());
 
         //convert CollectionSchema to io.milvus.grpc.CollectionSchema
@@ -141,6 +142,8 @@ public class CollectionService extends BaseService {
             //load collection
             loadCollection(blockingStub, LoadCollectionReq.builder().collectionName(request.getCollectionName()).build());
         }
+
+        return null;
     }
 
     public ListCollectionsResp listCollections(MilvusServiceGrpc.MilvusServiceBlockingStub blockingStub) {
@@ -154,7 +157,7 @@ public class CollectionService extends BaseService {
         return listCollectionsResp;
     }
 
-    public void dropCollection(MilvusServiceGrpc.MilvusServiceBlockingStub blockingStub, DropCollectionReq request) {
+    public Void dropCollection(MilvusServiceGrpc.MilvusServiceBlockingStub blockingStub, DropCollectionReq request) {
 
         String title = String.format("DropCollectionRequest collectionName:%s", request.getCollectionName());
         DropCollectionRequest dropCollectionRequest = DropCollectionRequest.newBuilder()
@@ -166,6 +169,8 @@ public class CollectionService extends BaseService {
         if (request.getAsync()) {
             WaitForDropCollection(blockingStub, request);
         }
+
+        return null;
     }
 
     public Boolean hasCollection(MilvusServiceGrpc.MilvusServiceBlockingStub blockingStub, HasCollectionReq request) {
@@ -206,7 +211,7 @@ public class CollectionService extends BaseService {
         return describeCollectionResp;
     }
 
-    public void renameCollection(MilvusServiceGrpc.MilvusServiceBlockingStub blockingStub, RenameCollectionReq request) {
+    public Void renameCollection(MilvusServiceGrpc.MilvusServiceBlockingStub blockingStub, RenameCollectionReq request) {
         String title = String.format("RenameCollectionRequest collectionName:%s", request.getCollectionName());
         RenameCollectionRequest renameCollectionRequest = RenameCollectionRequest.newBuilder()
                 .setOldName(request.getCollectionName())
@@ -214,9 +219,11 @@ public class CollectionService extends BaseService {
                 .build();
         Status status = blockingStub.renameCollection(renameCollectionRequest);
         rpcUtils.handleResponse(title, status);
+
+        return null;
     }
 
-    public void loadCollection(MilvusServiceGrpc.MilvusServiceBlockingStub blockingStub, LoadCollectionReq request) {
+    public Void loadCollection(MilvusServiceGrpc.MilvusServiceBlockingStub blockingStub, LoadCollectionReq request) {
         String title = String.format("LoadCollectionRequest collectionName:%s", request.getCollectionName());
         LoadCollectionRequest loadCollectionRequest = LoadCollectionRequest.newBuilder()
                 .setCollectionName(request.getCollectionName())
@@ -227,9 +234,11 @@ public class CollectionService extends BaseService {
         if (request.getAsync()) {
             WaitForLoadCollection(blockingStub, request);
         }
+
+        return null;
     }
 
-    public void releaseCollection(MilvusServiceGrpc.MilvusServiceBlockingStub blockingStub, ReleaseCollectionReq request) {
+    public Void releaseCollection(MilvusServiceGrpc.MilvusServiceBlockingStub blockingStub, ReleaseCollectionReq request) {
         String title = String.format("ReleaseCollectionRequest collectionName:%s", request.getCollectionName());
         ReleaseCollectionRequest releaseCollectionRequest = ReleaseCollectionRequest.newBuilder()
                 .setCollectionName(request.getCollectionName())
@@ -239,6 +248,8 @@ public class CollectionService extends BaseService {
         if (request.getAsync()) {
             waitForCollectionRelease(blockingStub, request);
         }
+
+        return null;
     }
 
     public Boolean getLoadState(MilvusServiceGrpc.MilvusServiceBlockingStub blockingStub, GetLoadStateReq request) {

src/main/java/io/milvus/v2/service/index/IndexService.java

@@ -36,7 +36,7 @@ import java.util.stream.Collectors;
 
 public class IndexService extends BaseService {
 
-    public void createIndex(MilvusServiceGrpc.MilvusServiceBlockingStub blockingStub, CreateIndexReq request) {
+    public Void createIndex(MilvusServiceGrpc.MilvusServiceBlockingStub blockingStub, CreateIndexReq request) {
         for(IndexParam indexParam : request.getIndexParams()) {
             String title = String.format("CreateIndexRequest collectionName:%s, fieldName:%s",
                     request.getCollectionName(), indexParam.getFieldName());
@@ -72,9 +72,11 @@ public class IndexService extends BaseService {
             Status status = blockingStub.createIndex(createIndexRequest);
             rpcUtils.handleResponse(title, status);
         }
+
+        return null;
     }
 
-    public void dropIndex(MilvusServiceGrpc.MilvusServiceBlockingStub blockingStub, DropIndexReq request) {
+    public Void dropIndex(MilvusServiceGrpc.MilvusServiceBlockingStub blockingStub, DropIndexReq request) {
         String title = String.format("DropIndexRequest collectionName:%s, fieldName:%s, indexName:%s",
                 request.getCollectionName(), request.getFieldName(), request.getIndexName());
         DropIndexRequest dropIndexRequest = DropIndexRequest.newBuilder()
@@ -85,6 +87,8 @@ public class IndexService extends BaseService {
 
         Status status = blockingStub.dropIndex(dropIndexRequest);
         rpcUtils.handleResponse(title, status);
+
+        return null;
     }
 
     public DescribeIndexResp describeIndex(MilvusServiceGrpc.MilvusServiceBlockingStub blockingStub, DescribeIndexReq request) {

src/main/java/io/milvus/v2/service/partition/PartitionService.java

@@ -28,7 +28,7 @@ import io.milvus.v2.service.partition.request.*;
 import java.util.List;
 
 public class PartitionService extends BaseService {
-    public void createPartition(MilvusServiceGrpc.MilvusServiceBlockingStub blockingStub, CreatePartitionReq request) {
+    public Void createPartition(MilvusServiceGrpc.MilvusServiceBlockingStub blockingStub, CreatePartitionReq request) {
         String title = String.format("Create partition %s in collection %s", request.getPartitionName(), request.getCollectionName());
 
         CreatePartitionRequest createPartitionRequest = io.milvus.grpc.CreatePartitionRequest.newBuilder()
@@ -38,9 +38,10 @@ public class PartitionService extends BaseService {
         Status status = blockingStub.createPartition(createPartitionRequest);
         rpcUtils.handleResponse(title, status);
 
+        return null;
     }
 
-    public void dropPartition(MilvusServiceGrpc.MilvusServiceBlockingStub blockingStub, DropPartitionReq request) {
+    public Void dropPartition(MilvusServiceGrpc.MilvusServiceBlockingStub blockingStub, DropPartitionReq request) {
         String title = String.format("Drop partition %s in collection %s", request.getPartitionName(), request.getCollectionName());
 
         io.milvus.grpc.DropPartitionRequest dropPartitionRequest = io.milvus.grpc.DropPartitionRequest.newBuilder()
@@ -50,6 +51,7 @@ public class PartitionService extends BaseService {
         Status status = blockingStub.dropPartition(dropPartitionRequest);
         rpcUtils.handleResponse(title, status);
 
+        return null;
     }
 
     public Boolean hasPartition(MilvusServiceGrpc.MilvusServiceBlockingStub blockingStub, HasPartitionReq request) {
@@ -77,7 +79,7 @@ public class PartitionService extends BaseService {
         return showPartitionsResponse.getPartitionNamesList();
     }
 
-    public void loadPartitions(MilvusServiceGrpc.MilvusServiceBlockingStub blockingStub, LoadPartitionsReq request) {
+    public Void loadPartitions(MilvusServiceGrpc.MilvusServiceBlockingStub blockingStub, LoadPartitionsReq request) {
         String title = String.format("Load partitions %s in collection %s", request.getPartitionNames(), request.getCollectionName());
 
         io.milvus.grpc.LoadPartitionsRequest loadPartitionsRequest = io.milvus.grpc.LoadPartitionsRequest.newBuilder()
@@ -85,9 +87,11 @@ public class PartitionService extends BaseService {
                 .addAllPartitionNames(request.getPartitionNames()).build();
         Status status = blockingStub.loadPartitions(loadPartitionsRequest);
         rpcUtils.handleResponse(title, status);
+
+        return null;
     }
 
-    public void releasePartitions(MilvusServiceGrpc.MilvusServiceBlockingStub blockingStub, ReleasePartitionsReq request) {
+    public Void releasePartitions(MilvusServiceGrpc.MilvusServiceBlockingStub blockingStub, ReleasePartitionsReq request) {
         String title = String.format("Release partitions %s in collection %s", request.getPartitionNames(), request.getCollectionName());
 
         io.milvus.grpc.ReleasePartitionsRequest releasePartitionsRequest = io.milvus.grpc.ReleasePartitionsRequest.newBuilder()
@@ -95,5 +99,7 @@ public class PartitionService extends BaseService {
                 .addAllPartitionNames(request.getPartitionNames()).build();
         Status status = blockingStub.releasePartitions(releasePartitionsRequest);
         rpcUtils.handleResponse(title, status);
+
+        return null;
     }
 }

src/main/java/io/milvus/v2/service/rbac/RoleService.java

@@ -38,7 +38,7 @@ public class RoleService extends BaseService {
         return response.getResultsList().stream().map(roleResult -> roleResult.getRole().getName()).collect(Collectors.toList());
     }
 
-    public void createRole(MilvusServiceGrpc.MilvusServiceBlockingStub blockingStub, CreateRoleReq request) {
+    public Void createRole(MilvusServiceGrpc.MilvusServiceBlockingStub blockingStub, CreateRoleReq request) {
         String title = "createRole";
         CreateRoleRequest createRoleRequest = CreateRoleRequest.newBuilder()
                 .setEntity(RoleEntity.newBuilder()
@@ -47,6 +47,8 @@ public class RoleService extends BaseService {
                 .build();
         Status status = blockingStub.createRole(createRoleRequest);
         rpcUtils.handleResponse(title, status);
+
+        return null;
     }
 
     public DescribeRoleResp describeRole(MilvusServiceGrpc.MilvusServiceBlockingStub blockingStub, DescribeRoleReq request) {
@@ -72,16 +74,18 @@ public class RoleService extends BaseService {
         return describeRoleResp;
     }
 
-    public void dropRole(MilvusServiceGrpc.MilvusServiceBlockingStub blockingStub, DropRoleReq request) {
+    public Void dropRole(MilvusServiceGrpc.MilvusServiceBlockingStub blockingStub, DropRoleReq request) {
         String title = "dropRole";
         DropRoleRequest dropRoleRequest = DropRoleRequest.newBuilder()
                 .setRoleName(request.getRoleName())
                 .build();
         Status status = blockingStub.dropRole(dropRoleRequest);
         rpcUtils.handleResponse(title, status);
+
+        return null;
     }
 
-    public void grantPrivilege(MilvusServiceGrpc.MilvusServiceBlockingStub blockingStub, GrantPrivilegeReq request) {
+    public Void grantPrivilege(MilvusServiceGrpc.MilvusServiceBlockingStub blockingStub, GrantPrivilegeReq request) {
         String title = "grantPrivilege";
         GrantEntity entity = GrantEntity.newBuilder()
                 .setRole(RoleEntity.newBuilder()
@@ -98,9 +102,11 @@ public class RoleService extends BaseService {
                 .build();
         Status status = blockingStub.operatePrivilege(operatePrivilegeRequest);
         rpcUtils.handleResponse(title, status);
+
+        return null;
     }
 
-    public void revokePrivilege(MilvusServiceGrpc.MilvusServiceBlockingStub blockingStub, RevokePrivilegeReq request) {
+    public Void revokePrivilege(MilvusServiceGrpc.MilvusServiceBlockingStub blockingStub, RevokePrivilegeReq request) {
         String title = "revokePrivilege";
         GrantEntity entity = GrantEntity.newBuilder()
                 .setRole(RoleEntity.newBuilder()
@@ -117,9 +123,11 @@ public class RoleService extends BaseService {
                 .build();
         Status status = blockingStub.operatePrivilege(operatePrivilegeRequest);
         rpcUtils.handleResponse(title, status);
+
+        return null;
     }
 
-    public void grantRole(MilvusServiceGrpc.MilvusServiceBlockingStub blockingStub, GrantRoleReq request) {
+    public Void grantRole(MilvusServiceGrpc.MilvusServiceBlockingStub blockingStub, GrantRoleReq request) {
         String title = "grantRole";
         OperateUserRoleRequest operateUserRoleRequest = OperateUserRoleRequest.newBuilder()
                 .setUsername(request.getUserName())
@@ -128,9 +136,11 @@ public class RoleService extends BaseService {
                 .build();
         Status status = blockingStub.operateUserRole(operateUserRoleRequest);
         rpcUtils.handleResponse(title, status);
+
+        return null;
     }
 
-    public void revokeRole(MilvusServiceGrpc.MilvusServiceBlockingStub blockingStub, RevokeRoleReq request) {
+    public Void revokeRole(MilvusServiceGrpc.MilvusServiceBlockingStub blockingStub, RevokeRoleReq request) {
         String title = "grantRole";
         OperateUserRoleRequest operateUserRoleRequest = OperateUserRoleRequest.newBuilder()
                 .setUsername(request.getUserName())
@@ -139,5 +149,7 @@ public class RoleService extends BaseService {
                 .build();
         Status status = blockingStub.operateUserRole(operateUserRoleRequest);
         rpcUtils.handleResponse(title, status);
+
+        return null;
     }
 }

src/main/java/io/milvus/v2/service/rbac/UserService.java

@@ -57,7 +57,7 @@ public class UserService extends BaseService {
         return describeUserResp;
     }
 
-    public void createUser(MilvusServiceGrpc.MilvusServiceBlockingStub blockingStub, CreateUserReq request) {
+    public Void createUser(MilvusServiceGrpc.MilvusServiceBlockingStub blockingStub, CreateUserReq request) {
         String title = String.format("create user %s", request.getUserName());
         CreateCredentialRequest createCredentialRequest = CreateCredentialRequest.newBuilder()
                 .setUsername(request.getUserName())
@@ -65,10 +65,12 @@ public class UserService extends BaseService {
                 .build();
         Status response = blockingStub.createCredential(createCredentialRequest);
         rpcUtils.handleResponse(title, response);
+
+        return null;
     }
 
 
-    public void updatePassword(MilvusServiceGrpc.MilvusServiceBlockingStub blockingStub, UpdatePasswordReq request) {
+    public Void updatePassword(MilvusServiceGrpc.MilvusServiceBlockingStub blockingStub, UpdatePasswordReq request) {
         String title = String.format("update password for user %s", request.getUserName());
         UpdateCredentialRequest updateCredentialRequest = UpdateCredentialRequest.newBuilder()
                 .setUsername(request.getUserName())
@@ -77,14 +79,18 @@ public class UserService extends BaseService {
                 .build();
         Status response = blockingStub.updateCredential(updateCredentialRequest);
         rpcUtils.handleResponse(title, response);
+
+        return null;
     }
 
-    public void dropUser(MilvusServiceGrpc.MilvusServiceBlockingStub blockingStub, DropUserReq request) {
+    public Void dropUser(MilvusServiceGrpc.MilvusServiceBlockingStub blockingStub, DropUserReq request) {
         String title = String.format("drop user %s", request.getUserName());
         DeleteCredentialRequest deleteCredentialRequest = DeleteCredentialRequest.newBuilder()
                 .setUsername(request.getUserName())
                 .build();
         Status response = blockingStub.deleteCredential(deleteCredentialRequest);
         rpcUtils.handleResponse(title, response);
+
+        return null;
     }
 }

src/main/java/io/milvus/v2/service/utility/UtilityService.java

@@ -39,7 +39,7 @@ public class UtilityService extends BaseService {
         return R.success(new RpcStatus(RpcStatus.SUCCESS_MSG));
     }
 
-    public void createAlias(MilvusServiceGrpc.MilvusServiceBlockingStub blockingStub, CreateAliasReq request) {
+    public Void createAlias(MilvusServiceGrpc.MilvusServiceBlockingStub blockingStub, CreateAliasReq request) {
         String title = String.format("Create alias %s for collection %s", request.getAlias(), request.getCollectionName());
         io.milvus.grpc.CreateAliasRequest createAliasRequest = io.milvus.grpc.CreateAliasRequest.newBuilder()
                 .setCollectionName(request.getCollectionName())
@@ -47,18 +47,22 @@ public class UtilityService extends BaseService {
                 .build();
         io.milvus.grpc.Status status = blockingStub.createAlias(createAliasRequest);
         rpcUtils.handleResponse(title, status);
+
+        return null;
     }
 
-    public void dropAlias(MilvusServiceGrpc.MilvusServiceBlockingStub blockingStub, DropAliasReq request) {
+    public Void dropAlias(MilvusServiceGrpc.MilvusServiceBlockingStub blockingStub, DropAliasReq request) {
         String title = String.format("Drop alias %s", request.getAlias());
         io.milvus.grpc.DropAliasRequest dropAliasRequest = io.milvus.grpc.DropAliasRequest.newBuilder()
                 .setAlias(request.getAlias())
                 .build();
         io.milvus.grpc.Status status = blockingStub.dropAlias(dropAliasRequest);
         rpcUtils.handleResponse(title, status);
+
+        return null;
     }
 
-    public void alterAlias(MilvusServiceGrpc.MilvusServiceBlockingStub blockingStub, AlterAliasReq request) {
+    public Void alterAlias(MilvusServiceGrpc.MilvusServiceBlockingStub blockingStub, AlterAliasReq request) {
         String title = String.format("Alter alias %s for collection %s", request.getAlias(), request.getCollectionName());
         io.milvus.grpc.AlterAliasRequest alterAliasRequest = io.milvus.grpc.AlterAliasRequest.newBuilder()
                 .setCollectionName(request.getCollectionName())
@@ -66,6 +70,8 @@ public class UtilityService extends BaseService {
                 .build();
         io.milvus.grpc.Status status = blockingStub.alterAlias(alterAliasRequest);
         rpcUtils.handleResponse(title, status);
+
+        return null;
     }
 
     public DescribeAliasResp describeAlias(MilvusServiceGrpc.MilvusServiceBlockingStub blockingStub, DescribeAliasReq request) {

src/main/java/io/milvus/v2/utils/RpcUtils.java

@@ -49,7 +49,8 @@ public class RpcUtils {
             }
             logger.error("{} failed, error code: {}, reason: {}", requestInfo, ErrorCode.SERVER_ERROR.getCode(),
                     status.getReason());
-            throw new MilvusClientException(ErrorCode.SERVER_ERROR, status.getReason());
+            throw new MilvusClientException(ErrorCode.SERVER_ERROR, status.getReason(),
+                    code, status.getErrorCode().getNumber());
         }
 
         logger.debug("{} successfully!", requestInfo);

src/test/java/io/milvus/client/MilvusClientDockerTest.java

@@ -79,7 +79,7 @@ class MilvusClientDockerTest {
     protected static final Gson GSON_INSTANCE = new Gson();
 
     @Container
-    private static final MilvusContainer milvus = new MilvusContainer("milvusdb/milvus:v2.3.13");
+    private static final MilvusContainer milvus = new MilvusContainer("milvusdb/milvus:v2.3.20");
 
     @BeforeAll
     public static void setUp() {

src/test/java/io/milvus/v2/client/MilvusClientV2DockerTest.java

@@ -70,7 +70,7 @@ class MilvusClientV2DockerTest {
     private static final Random RANDOM = new Random();
 
     @Container
-    private static final MilvusContainer milvus = new MilvusContainer("milvusdb/milvus:v2.3.13");
+    private static final MilvusContainer milvus = new MilvusContainer("milvusdb/milvus:v2.3.20");
 
     @BeforeAll
     public static void setUp() {