[feat][sdk] Support vector count.

yuhaijun999 · Sep 27, 2023 · 5143f05 · 5143f05
1 parent 4f2baba
commit 5143f05
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/java/dingo-sdk/src/main/java/io/dingodb/sdk/service/index/IndexServiceClient.java b/java/dingo-sdk/src/main/java/io/dingodb/sdk/service/index/IndexServiceClient.java
@@ -280,6 +280,15 @@ public List<Boolean> vectorDelete(DingoCommonId indexId, DingoCommonId regionId,
         return response.getKeyStatesList();
     }
 
+    public Long vectorCount(DingoCommonId indexId, DingoCommonId regionId) {
+        Index.VectorCountRequest.Builder builder = Index.VectorCountRequest.newBuilder();
+
+        Index.VectorCountResponse response = exec(stub -> stub.vectorCount(
+                builder.setContext(mapping(contextCache.get(regionId))).build()), retryTimes, indexId, regionId);
+
+        return response.getCount();
+    }
+
     private <R> R exec(
             Function<IndexServiceGrpc.IndexServiceBlockingStub, R> function,
             int retryTimes,