Browse Source

Fix a bug: collection consistency level is missed (#1365)

Signed-off-by: yhmo <yihua.mo@zilliz.com>
groot 3 weeks ago
parent
commit
681ada8e39

+ 2 - 1
src/main/java/io/milvus/v2/service/collection/CollectionService.java

@@ -123,7 +123,8 @@ public class CollectionService extends BaseService {
         CreateCollectionRequest.Builder builder = CreateCollectionRequest.newBuilder()
                 .setCollectionName(request.getCollectionName())
                 .setSchema(grpcSchema.toByteString())
-                .setShardsNum(request.getNumShards());
+                .setShardsNum(request.getNumShards())
+                .setConsistencyLevelValue(request.getConsistencyLevel().getCode());
         List<KeyValuePair> propertiesList = ParamUtils.AssembleKvPair(request.getProperties());
         if (CollectionUtils.isNotEmpty(propertiesList)) {
             propertiesList.forEach(builder::addProperties);

+ 1 - 0
src/main/java/io/milvus/v2/service/collection/response/DescribeCollectionResp.java

@@ -46,6 +46,7 @@ public class DescribeCollectionResp {
     private CreateCollectionReq.CollectionSchema collectionSchema;
     private Long createTime;
     private ConsistencyLevel consistencyLevel;
+    private Integer shardsNum;
     @Builder.Default
     private final Map<String, String> properties = new HashMap<>();
 }

+ 1 - 0
src/main/java/io/milvus/v2/utils/ConvertUtils.java

@@ -138,6 +138,7 @@ public class ConvertUtils {
                 .primaryFieldName(response.getSchema().getFieldsList().stream().filter(FieldSchema::getIsPrimaryKey).map(FieldSchema::getName).collect(java.util.stream.Collectors.toList()).get(0))
                 .createTime(response.getCreatedTimestamp())
                 .consistencyLevel(io.milvus.v2.common.ConsistencyLevel.valueOf(response.getConsistencyLevel().name().toUpperCase()))
+                .shardsNum(response.getShardsNum())
                 .properties(properties)
                 .build();
         return describeCollectionResp;