2024/07/19 ai标注枚举修改

middleware-automatic-center-client/src/main/java/com/zhiwei/middleware/automatic/server/pojo/AIMark.java

@@ -34,6 +34,8 @@ public class AIMark implements Serializable {
 
     private List<String> fields;
 
+    private String methodType;
+
     /**
      * 项目名
      */
@@ -99,6 +101,14 @@ public class AIMark implements Serializable {
         this.projectId = projectId;
     }
 
+    public String getMethodType() {
+        return methodType;
+    }
+
+    public void setMethodType(String methodType) {
+        this.methodType = methodType;
+    }
+
     public List<Integer> getMonitorLevelIds() {
         return monitorLevelIds;
     }

middleware-automatic-center-client/src/main/java/com/zhiwei/middleware/automatic/server/pojo/enums/AiInterfaceEnums.java

@@ -6,24 +6,19 @@ package com.zhiwei.middleware.automatic.server.pojo.enums;
  **/
 public enum AiInterfaceEnums {
 
-    FEI_HE("飞鹤", "62baa19f4d9a553ac3e7cd9b"),
+    PRODUCT("PRODUCT"),
 
-    COMMON("常用", "常用");
+    COMMON("COMMON");
 
-    final String project;
+    final String type;
 
-    final String projectId;
 
-    AiInterfaceEnums(String project, String projectId) {
-        this.project = project;
-        this.projectId = projectId;
+    AiInterfaceEnums(String type) {
+        this.type = type;
     }
 
-    public String getProject() {
-        return project;
+    public String getType() {
+        return type;
     }
 
-    public String getProjectId() {
-        return projectId;
-    }
 }

middleware-automatic-center-son/src/main/java/com/zhiwei/middleware/automatic/son/task/aiTask/AiTaskServiceHandler.java

@@ -18,7 +18,7 @@ public class AiTaskServiceHandler {
     private static final Map<String, AiTaskService> SERVICE_MAP = new HashMap<>();
     private AiTaskServiceHandler() {
         Map<String, AiTaskService> beansOfType = ApplicationContextHolder.getInstance().getBeansOfType(AiTaskService.class);
-        beansOfType.forEach((k, v) -> SERVICE_MAP.put(v.getType().getProjectId(), v));
+        beansOfType.forEach((k, v) -> SERVICE_MAP.put(v.getType().getType(), v));
     }
 
     public static AiTaskServiceHandler getInstance() {
@@ -26,9 +26,9 @@ public class AiTaskServiceHandler {
     }
 
     public MarkInfo sendAIRequest(AiInterfaceParam aiParam, String timeKey) {
-        AiTaskService aiTaskService = SERVICE_MAP.get(aiParam.getAiMark().getProjectId());
+        AiTaskService aiTaskService = SERVICE_MAP.get(aiParam.getAiMark().getMethodType());
         if (Objects.isNull(aiTaskService)) {
-            aiTaskService = SERVICE_MAP.get(AiInterfaceEnums.COMMON.getProjectId());
+            aiTaskService = SERVICE_MAP.get(AiInterfaceEnums.COMMON.getType());
         }
         return aiTaskService.executor(aiParam, timeKey);
     }

middleware-automatic-center-son/src/main/java/com/zhiwei/middleware/automatic/son/task/aiTask/AiTaskServiceFeiHe.java → middleware-automatic-center-son/src/main/java/com/zhiwei/middleware/automatic/son/task/aiTask/AiTaskServiceProduct.java

@@ -12,9 +12,9 @@ import org.springframework.stereotype.Service;
  * date: 2024/7/9 14:16
  **/
 @Service
-public class AiTaskServiceFeiHe extends AbstractAiTaskService {
-    public AiTaskServiceFeiHe(RedissonUtil redissonUtil, AiApiResultDao aiApiResultDao) {
-        super(AiInterfaceEnums.FEI_HE, redissonUtil, aiApiResultDao);
+public class AiTaskServiceProduct extends AbstractAiTaskService {
+    public AiTaskServiceProduct(RedissonUtil redissonUtil, AiApiResultDao aiApiResultDao) {
+        super(AiInterfaceEnums.PRODUCT, redissonUtil, aiApiResultDao);
     }
 
     @Override