Browse Source

Refine code (#1418)

Signed-off-by: yhmo <yihua.mo@zilliz.com>
groot 1 month ago
parent
commit
7d53a177f2

+ 12 - 2
sdk-core/src/main/java/io/milvus/common/clientenum/FunctionType.java

@@ -19,7 +19,17 @@
 
 package io.milvus.common.clientenum;
 
+import lombok.Getter;
+
 public enum FunctionType {
-    UNKNOWN,
-    BM25,
+    UNKNOWN(0),
+    BM25(1),
+    ;
+
+    @Getter
+    private final int code;
+
+    FunctionType(int i) {
+        code = i;
+    }
 }

+ 2 - 1
sdk-core/src/main/java/io/milvus/v2/service/collection/request/CreateCollectionReq.java

@@ -231,7 +231,8 @@ public class CreateCollectionReq {
         private String name;
         @Builder.Default
         private String description = "";
-        private FunctionType functionType;
+        @Builder.Default
+        private FunctionType functionType = FunctionType.UNKNOWN;
         @Builder.Default
         private List<String> inputFieldNames = new ArrayList<>();
         @Builder.Default

+ 1 - 1
sdk-core/src/main/java/io/milvus/v2/utils/SchemaUtils.java

@@ -129,7 +129,7 @@ public class SchemaUtils {
         FunctionSchema.Builder builder = FunctionSchema.newBuilder()
                 .setName(function.getName())
                 .setDescription(function.getDescription())
-                .setType(FunctionType.valueOf(function.getFunctionType().name()));
+                .setType(FunctionType.forNumber(function.getFunctionType().getCode()));
 
         for (String name : function.getInputFieldNames()) {
             builder.addInputFieldNames(name);