add shardNum field if listCollection (milvus-io#1585)

Signed-off-by: lentitude2tk <[email protected]>

Author: lentitude2tk

sdk-core/src/main/java/io/milvus/v2/service/collection/CollectionInfo.java

@@ -0,0 +1,11 @@
+package io.milvus.v2.service.collection;
+
+import lombok.Data;
+import lombok.experimental.SuperBuilder;
+
+@Data
+@SuperBuilder
+public class CollectionInfo {
+    private String collectionName;
+    private Integer shardNum;
+}

sdk-core/src/main/java/io/milvus/v2/service/collection/CollectionService.java

@@ -183,11 +183,24 @@ public ListCollectionsResp listCollections(MilvusServiceGrpc.MilvusServiceBlocki
         ShowCollectionsRequest showCollectionsRequest = ShowCollectionsRequest.newBuilder()
                 .build();
         ShowCollectionsResponse response = blockingStub.showCollections(showCollectionsRequest);
-        ListCollectionsResp listCollectionsResp = ListCollectionsResp.builder()
+
+        List<CollectionInfo> collectionInfos = new ArrayList<>();
+        for (int i = 0; i < response.getCollectionNamesCount(); i++) {
+            CollectionInfo collectionInfo = CollectionInfo.builder()
+                    .collectionName(response.getCollectionNames(i))
+                    .build();
+            // Milvus version >= 2.6.1 will additionally return shardNum
+            List<Integer> shardsNums = response.getShardsNumList();
+            if (CollectionUtils.isNotEmpty(shardsNums)) {
+                collectionInfo.setShardNum(response.getShardsNum(i));
+            }
+            collectionInfos.add(collectionInfo);
+        }
+
+        return ListCollectionsResp.builder()
                 .collectionNames(response.getCollectionNamesList())
+                .collectionInfos(collectionInfos)
                 .build();
-
-        return listCollectionsResp;
     }
 
     public Void dropCollection(MilvusServiceGrpc.MilvusServiceBlockingStub blockingStub, DropCollectionReq request) {

sdk-core/src/main/java/io/milvus/v2/service/collection/response/ListCollectionsResp.java

@@ -50,9 +50,9 @@ public void setCollectionNames(List<String> collectionNames) {
     public boolean equals(Object obj) {
         if (this == obj) return true;
         if (obj == null || getClass() != obj.getClass()) return false;
-        
+
         ListCollectionsResp that = (ListCollectionsResp) obj;
-        
+
         return new EqualsBuilder()
                 .append(collectionNames, that.collectionNames)
                 .isEquals();