Browse Source

feat: 添加获取算法enums接口

WANGKANG 9 tháng trước cách đây
mục cha
commit
e2cf902e2e

+ 5 - 0
taais-modules/taais-biz/src/main/java/com/taais/biz/controller/AlgorithmConfigTrackController.java

@@ -106,4 +106,9 @@ public class AlgorithmConfigTrackController extends BaseController {
         }
         return CommonResult.success();
     }
+
+    @GetMapping("/enums")
+    public CommonResult<List<Object>> enums() {
+        return CommonResult.success(algorithmConfigTrackService.getEnums());
+    }
 }

+ 1 - 0
taais-modules/taais-biz/src/main/java/com/taais/biz/service/IAlgorithmConfigTrackService.java

@@ -63,4 +63,5 @@ public interface IAlgorithmConfigTrackService extends IBaseService<AlgorithmConf
      */
     boolean deleteByIds(Long[] ids);
 
+    List<Object> getEnums();
 }

+ 19 - 0
taais-modules/taais-biz/src/main/java/com/taais/biz/service/impl/AlgorithmConfigTrackServiceImpl.java

@@ -1,6 +1,8 @@
 package com.taais.biz.service.impl;
 
+import java.util.ArrayList;
 import java.util.Arrays;
+import java.util.HashMap;
 import java.util.List;
 
 import cn.hutool.core.util.ObjectUtil;
@@ -137,4 +139,21 @@ public class AlgorithmConfigTrackServiceImpl extends BaseServiceImpl<AlgorithmCo
         return this.removeByIds(Arrays.asList(ids));
     }
 
+    @Override
+    public List<Object> getEnums() {
+        algorithmConfigTrackMapper.selectAll();
+
+        List<Object> result = new ArrayList<>();
+
+        for (AlgorithmConfigTrack algorithmConfigTrack : algorithmConfigTrackMapper.selectAll()) {
+            HashMap<String, Object> map = new HashMap<>();
+            map.put("label",algorithmConfigTrack.getAlgorithmName());
+            map.put("value",algorithmConfigTrack.getId());
+
+            result.add(map);
+        }
+
+        return result;
+    }
+
 }