!34 dev to main

Merge pull request !34 from xgc/dev
This commit is contained in:
xgc 2024-08-30 08:41:10 +00:00 committed by Gitee
commit 863607ca37
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
7 changed files with 31 additions and 10 deletions

View File

@ -31,7 +31,7 @@
<dependency> <dependency>
<groupId>org.dromara</groupId> <groupId>org.dromara</groupId>
<artifactId>milvus-plus-core</artifactId> <artifactId>milvus-plus-core</artifactId>
<version>2.1.3</version> <version>2.1.4</version>
</dependency> </dependency>
``` ```
@ -41,7 +41,7 @@ Spring应用支持
<dependency> <dependency>
<groupId>org.dromara</groupId> <groupId>org.dromara</groupId>
<artifactId>milvus-plus-boot-starter</artifactId> <artifactId>milvus-plus-boot-starter</artifactId>
<version>2.1.3</version> <version>2.1.4</version>
</dependency> </dependency>
``` ```
@ -51,7 +51,7 @@ Solon应用支持
<dependency> <dependency>
<groupId>org.dromara</groupId> <groupId>org.dromara</groupId>
<artifactId>milvus-plus-solon-plugin</artifactId> <artifactId>milvus-plus-solon-plugin</artifactId>
<version>2.1.3</version> <version>2.1.4</version>
</dependency> </dependency>
``` ```

View File

@ -27,7 +27,7 @@ Custom extension support:
<dependency> <dependency>
<groupId>org.dromara</groupId> <groupId>org.dromara</groupId>
<artifactId>milvus-plus-core</artifactId> <artifactId>milvus-plus-core</artifactId>
<version>2.1.3</version> <version>2.1.4</version>
</dependency> </dependency>
``` ```
@ -37,7 +37,7 @@ Spring application support:
<dependency> <dependency>
<groupId>org.dromara</groupId> <groupId>org.dromara</groupId>
<artifactId>milvus-plus-boot-starter</artifactId> <artifactId>milvus-plus-boot-starter</artifactId>
<version>2.1.3</version> <version>2.1.4</version>
</dependency> </dependency>
``` ```
@ -47,7 +47,7 @@ Solon application support:
<dependency> <dependency>
<groupId>org.dromara</groupId> <groupId>org.dromara</groupId>
<artifactId>milvus-plus-solon-plugin</artifactId> <artifactId>milvus-plus-solon-plugin</artifactId>
<version>2.1.3</version> <version>2.1.4</version>
</dependency> </dependency>
``` ```

View File

@ -20,9 +20,19 @@ public @interface MilvusCollection {
String name(); String name();
/** /**
* * 别名
*/ */
String[] alias() default {}; String[] alias() default {};
/**
* 一致性级别
*/
ConsistencyLevel level() default ConsistencyLevel.BOUNDED; ConsistencyLevel level() default ConsistencyLevel.BOUNDED;
/**
* 禁用动态字段
*/
boolean enableDynamicField() default false;
} }

View File

@ -16,7 +16,14 @@ public class CollectionSchemaBuilder {
private final MilvusClientV2 wrapper; private final MilvusClientV2 wrapper;
private final CreateCollectionReq.CollectionSchema schema; private final CreateCollectionReq.CollectionSchema schema;
private ConsistencyLevel consistencyLevel=ConsistencyLevel.BOUNDED; private ConsistencyLevel consistencyLevel=ConsistencyLevel.BOUNDED;
private Boolean enableDynamicField=false;
public CollectionSchemaBuilder(Boolean enableDynamicField,String collectionName, MilvusClientV2 wrapper) {
this.collectionName = collectionName;
this.wrapper = wrapper;
this.schema = wrapper.createSchema();
this.enableDynamicField=enableDynamicField;
}
public CollectionSchemaBuilder(String collectionName, MilvusClientV2 wrapper) { public CollectionSchemaBuilder(String collectionName, MilvusClientV2 wrapper) {
this.collectionName = collectionName; this.collectionName = collectionName;
this.wrapper = wrapper; this.wrapper = wrapper;
@ -44,7 +51,8 @@ public class CollectionSchemaBuilder {
CreateCollectionReq req=CreateCollectionReq.builder(). CreateCollectionReq req=CreateCollectionReq.builder().
collectionName(this.collectionName). collectionName(this.collectionName).
collectionSchema(this.schema). collectionSchema(this.schema).
consistencyLevel(this.consistencyLevel) consistencyLevel(this.consistencyLevel).
enableDynamicField(this.enableDynamicField)
.build(); .build();
wrapper.createCollection(req); wrapper.createCollection(req);
} }

View File

@ -69,6 +69,8 @@ public class MilvusConverter {
// 集合名称 // 集合名称
String collectionName = collectionAnnotation.name(); String collectionName = collectionAnnotation.name();
milvus.setCollectionName(collectionName); milvus.setCollectionName(collectionName);
boolean enableDynamicField = collectionAnnotation.enableDynamicField();
milvus.setEnableDynamicField(enableDynamicField);
//一致性级别 //一致性级别
ConsistencyLevel level = collectionAnnotation.level(); ConsistencyLevel level = collectionAnnotation.level();
milvus.setConsistencyLevel(level); milvus.setConsistencyLevel(level);
@ -220,7 +222,7 @@ public class MilvusConverter {
} }
// 创建新集合 // 创建新集合
CollectionSchemaBuilder schemaBuilder = new CollectionSchemaBuilder( CollectionSchemaBuilder schemaBuilder = new CollectionSchemaBuilder(
milvusEntity.getCollectionName(), client milvusEntity.getEnableDynamicField(),milvusEntity.getCollectionName(), client
); );
schemaBuilder.addField(milvusEntity.getMilvusFields().toArray(new AddFieldReq[0])); schemaBuilder.addField(milvusEntity.getMilvusFields().toArray(new AddFieldReq[0]));
schemaBuilder.addConsistencyLevel(milvusEntity.getConsistencyLevel()); schemaBuilder.addConsistencyLevel(milvusEntity.getConsistencyLevel());

View File

@ -18,4 +18,5 @@ public class MilvusEntity {
private List<AddFieldReq> milvusFields; private List<AddFieldReq> milvusFields;
private List<String> partitionName; private List<String> partitionName;
private ConsistencyLevel consistencyLevel; private ConsistencyLevel consistencyLevel;
private Boolean enableDynamicField;
} }

View File

@ -30,7 +30,7 @@
</developer> </developer>
</developers> </developers>
<properties> <properties>
<revision>2.1.3</revision> <revision>2.1.4</revision>
<maven.compiler.source>${java.version}</maven.compiler.source> <maven.compiler.source>${java.version}</maven.compiler.source>
<maven.compiler.target>${java.version}</maven.compiler.target> <maven.compiler.target>${java.version}</maven.compiler.target>
<maven-compiler.version>3.11.0</maven-compiler.version> <maven-compiler.version>3.11.0</maven-compiler.version>