Ver Fonte

Fix a bug of SearchResutsWrapper (#758)

Signed-off-by: yhmo <yihua.mo@zilliz.com>
groot há 1 ano atrás
pai
commit
0b9568ee1f

+ 8 - 2
src/main/java/io/milvus/response/SearchResultsWrapper.java

@@ -60,9 +60,15 @@ public class SearchResultsWrapper extends RowRecordWrapper {
         long topK = results.getTopK();
         List<IDScore> idScore = getIDScore(indexOfTarget);
         for (int i = 0; i < topK; ++i) {
+            IDScore score = idScore.get(i);
             QueryResultsWrapper.RowRecord record = new QueryResultsWrapper.RowRecord();
-            record.put("id", idScore.get(i).getLongID());
-            record.put("distance", idScore.get(i).getScore());
+            if (score.getStrID().isEmpty()) {
+                record.put("id", score.getLongID());
+            } else {
+                record.put("id", score.getStrID());
+            }
+
+            record.put("distance", score.getScore());
             buildRowRecord(record, i);
             records.add(record);
         }