Browse Source

fix issue 64

Zhiru Zhu 5 years ago
parent
commit
8867d5f7be

+ 10 - 4
src/main/java/io/milvus/client/MilvusGrpcClient.java

@@ -741,11 +741,17 @@ public class MilvusGrpcClient implements MilvusClient {
 
     final int numQueries = (int) topKQueryResult.getRowNum();
     final int topK =
-        topKQueryResult.getIdsCount() / numQueries; // Guaranteed to be disable from server side
+        numQueries == 0
+            ? 0
+            : topKQueryResult.getIdsCount()
+                / numQueries; // Guaranteed to be divisible from server side
 
-    List<List<Long>> resultIdsList = ListUtils.partition(topKQueryResult.getIdsList(), topK);
-    List<List<Float>> resultDistancesList =
-        ListUtils.partition(topKQueryResult.getDistancesList(), topK);
+    List<List<Long>> resultIdsList = new ArrayList<>();
+    List<List<Float>> resultDistancesList = new ArrayList<>();
+    if (topK > 0) {
+      resultIdsList = ListUtils.partition(topKQueryResult.getIdsList(), topK);
+      resultDistancesList = ListUtils.partition(topKQueryResult.getDistancesList(), topK);
+    }
 
     SearchResponse searchResponse = new SearchResponse();
     searchResponse.setNumQueries(numQueries);

+ 1 - 1
src/test/java/io/milvus/client/MilvusGrpcClientTest.java

@@ -67,7 +67,7 @@ class MilvusClientTest {
 
     client = new MilvusGrpcClient();
     ConnectParam connectParam =
-        new ConnectParam.Builder().withHost("192.168.1.113").withPort("19530").build();
+        new ConnectParam.Builder().withHost("localhost").withPort("19530").build();
     client.connect(connectParam);
 
     generator = new RandomStringGenerator.Builder().withinRange('a', 'z').build();