diff --git a/sdk-core/src/main/java/io/milvus/v2/service/collection/CollectionService.java b/sdk-core/src/main/java/io/milvus/v2/service/collection/CollectionService.java index ac2b7be35..89c31ae01 100644 --- a/sdk-core/src/main/java/io/milvus/v2/service/collection/CollectionService.java +++ b/sdk-core/src/main/java/io/milvus/v2/service/collection/CollectionService.java @@ -100,10 +100,10 @@ public Void createCollection(MilvusServiceGrpc.MilvusServiceBlockingStub blockin .collectionName(request.getCollectionName()) .build(); indexService.createIndex(blockingStub, createIndexReq); - //load collection + //load collection, set async to true since no need to wait loading progress try { //TimeUnit.MILLISECONDS.sleep(1000); - loadCollection(blockingStub, LoadCollectionReq.builder().collectionName(request.getCollectionName()).build()); + loadCollection(blockingStub, LoadCollectionReq.builder().async(true).collectionName(request.getCollectionName()).build()); } catch (Exception e) { throw new MilvusClientException(ErrorCode.SERVER_ERROR, "Load collection failed: " + e); } @@ -160,8 +160,8 @@ public Void createCollectionWithSchema(MilvusServiceGrpc.MilvusServiceBlockingSt .build(); indexService.createIndex(blockingStub, createIndexReq); } - //load collection - loadCollection(blockingStub, LoadCollectionReq.builder().collectionName(request.getCollectionName()).build()); + //load collection, set async to true since no need to wait loading progress + loadCollection(blockingStub, LoadCollectionReq.builder().async(true).collectionName(request.getCollectionName()).build()); } return null; @@ -289,7 +289,7 @@ public Void loadCollection(MilvusServiceGrpc.MilvusServiceBlockingStub blockingS .build(); Status status = blockingStub.loadCollection(loadCollectionRequest); rpcUtils.handleResponse(title, status); - if (request.getAsync()) { + if (!request.getAsync()) { WaitForLoadCollection(blockingStub, request.getCollectionName(), request.getTimeout()); } diff --git a/sdk-core/src/main/java/io/milvus/v2/service/collection/request/LoadCollectionReq.java b/sdk-core/src/main/java/io/milvus/v2/service/collection/request/LoadCollectionReq.java index d6f9e2534..c9ed0cf5f 100644 --- a/sdk-core/src/main/java/io/milvus/v2/service/collection/request/LoadCollectionReq.java +++ b/sdk-core/src/main/java/io/milvus/v2/service/collection/request/LoadCollectionReq.java @@ -33,7 +33,7 @@ public class LoadCollectionReq { @Builder.Default private Integer numReplicas = 1; @Builder.Default - private Boolean async = Boolean.TRUE; + private Boolean async = Boolean.FALSE; @Builder.Default private Long timeout = 60000L; @Builder.Default diff --git a/sdk-core/src/test/java/io/milvus/v2/client/MilvusClientV2DockerTest.java b/sdk-core/src/test/java/io/milvus/v2/client/MilvusClientV2DockerTest.java index eb1738850..7a5df3033 100644 --- a/sdk-core/src/test/java/io/milvus/v2/client/MilvusClientV2DockerTest.java +++ b/sdk-core/src/test/java/io/milvus/v2/client/MilvusClientV2DockerTest.java @@ -1587,14 +1587,12 @@ void testDatabase() { .indexType(IndexParam.IndexType.FLAT) .metricType(IndexParam.MetricType.COSINE) .build(); - CreateCollectionReq requestCreate = CreateCollectionReq.builder() .collectionName(randomCollectionName) .collectionSchema(collectionSchema) .indexParams(Collections.singletonList(indexParam)) .build(); client.createCollection(requestCreate); - ListCollectionsResp listCollectionsResp = client.listCollections(); List collectionNames = listCollectionsResp.getCollectionNames(); Assertions.assertEquals(1, collectionNames.size());