fix: 状态刷新

taais-modules/taais-biz/src/main/java/com/taais/biz/controller/DataAugmentationController.java

@@ -20,6 +20,7 @@ import com.taais.biz.domain.bo.DataAugmentationBo;
 import com.taais.biz.domain.bo.DataAugmentationResultBo;
 import com.taais.biz.domain.bo.VideoStableStartResultBo;
 import com.taais.biz.domain.vo.DataAugmentationVo;
+import com.taais.biz.domain.vo.TaskDictDataVo;
 import com.taais.biz.service.IVideoStableService;
 import com.taais.common.core.service.OssService;
 import com.taais.system.domain.SysDictData;
@@ -30,6 +31,7 @@ import jakarta.validation.Valid;
 import lombok.RequiredArgsConstructor;
 import jakarta.servlet.http.HttpServletResponse;
 import cn.dev33.satoken.annotation.SaCheckPermission;
+import org.apache.commons.compress.archivers.ar.ArArchiveEntry;
 import org.apache.commons.io.IOUtils;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.http.HttpHeaders;
@@ -341,6 +343,30 @@ public class DataAugmentationController extends BaseController {
         } catch (Exception e) {
             return CommonResult.fail("日志文件" + filePath + "读取失败！");
         }
+    }
 
+    @SaCheckPermission("demo:dataAugmentation:query")
+    @GetMapping(value = "/getTaskDictData")
+    public CommonResult<List<TaskDictDataVo>> getTaskDictData() {
+        ArrayList<TaskDictDataVo> taskDictDataVos = new ArrayList<>();
+        QueryWrapper queryWrapper = new QueryWrapper();
+        queryWrapper.eq("dict_type", "biz_data_augmentation");
+        queryWrapper.eq("dict_label", "任务类型");
+        List<SysDictData> taskTypeList = sysDictDataService.list(queryWrapper);
+        for (SysDictData sysDictData: taskTypeList) {
+            String dictValue = sysDictData.getDictValue();
+            queryWrapper.clear();
+            queryWrapper.eq("dict_type", "biz_data_augmentation");
+            queryWrapper.eq("dict_label", dictValue + "超参配置");
+            List<SysDictData> hyparamList = sysDictDataService.list(queryWrapper);
+            if (hyparamList.isEmpty()) {
+                return CommonResult.fail("未在type为biz_data_augmentation的字典数据中设置" + dictValue + "的超参配置");
+            }
+            TaskDictDataVo taskDictDataVo = new TaskDictDataVo();
+            taskDictDataVo.setTaskType(dictValue);
+            taskDictDataVo.setHyperparameterConfiguration(hyparamList.get(0).getDictValue());
+            taskDictDataVos.add(taskDictDataVo);
+        }
+        return CommonResult.success(taskDictDataVos);
     }
 }

taais-modules/taais-biz/src/main/java/com/taais/biz/domain/vo/TaskDictDataVo.java

@@ -0,0 +1,30 @@
+package com.taais.biz.domain.vo;
+
+import java.util.Date;
+import com.alibaba.excel.annotation.ExcelIgnoreUnannotated;
+import com.alibaba.excel.annotation.ExcelProperty;
+import com.taais.biz.domain.DataAugmentation;
+import io.github.linpeilie.annotations.AutoMapper;
+import lombok.Data;
+import lombok.EqualsAndHashCode;
+import java.io.Serial;
+import java.io.Serializable;
+import com.taais.common.orm.core.domain.BaseEntity;
+
+/**
+ * 任务字典数据
+ *
+ * @author tanlun
+ * 2024-11-14
+ */
+@Data
+@ExcelIgnoreUnannotated
+@EqualsAndHashCode(callSuper = true)
+public class TaskDictDataVo extends BaseEntity implements Serializable {
+
+    @Serial
+    private static final long serialVersionUID = 1L;
+    private String taskType;
+    private String hyperparameterConfiguration;
+
+}