QueryIterator/SearchIterator support retry

Signed-off-by: yhmo <[email protected]>
milvus-io · Dec 31, 2024 · c8d7933 · c8d7933
1 parent f5462d5
commit c8d7933
Show file tree

Hide file tree

Showing 4 changed files with 198 additions and 192 deletions.
diff --git a/sdk-core/src/main/java/io/milvus/orm/iterator/QueryIterator.java b/sdk-core/src/main/java/io/milvus/orm/iterator/QueryIterator.java
@@ -230,7 +230,7 @@ private QueryResults executeQuery(String expr, long offset, long limit, long ts)
         // set default consistency level
         builder.setUseDefaultConsistency(true);
 
-        QueryResults response = blockingStub.query(builder.build());
+        QueryResults response = rpcUtils.retry(()->blockingStub.query(builder.build()));
         String title = String.format("QueryRequest collectionName:%s", queryIteratorParam.getCollectionName());
         rpcUtils.handleResponse(title, response.getStatus());
         return response;

diff --git a/sdk-core/src/main/java/io/milvus/orm/iterator/SearchIterator.java b/sdk-core/src/main/java/io/milvus/orm/iterator/SearchIterator.java
@@ -282,7 +282,7 @@ private SearchResults executeSearch(Map<String, Object> params, String nextExpr,
         // set default consistency level
         builder.setUseDefaultConsistency(true);
 
-        SearchResults response = blockingStub.search(builder.build());
+        SearchResults response = rpcUtils.retry(()->blockingStub.search(builder.build()));
         String title = String.format("SearchRequest collectionName:%s", searchIteratorParam.getCollectionName());
         rpcUtils.handleResponse(title, response.getStatus());
         return response;