自动机器学习开发

ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/AutoMlInsService.java

@@ -1,7 +1,8 @@
 package com.ruoyi.platform.service;
+
+import com.ruoyi.platform.domain.AutoMlIns;
 import org.springframework.data.domain.Page;
 import org.springframework.data.domain.PageRequest;
-import com.ruoyi.platform.domain.AutoMlIns;
 
 import java.io.IOException;
 import java.util.List;
@@ -23,4 +24,6 @@ public interface AutoMlInsService {
     boolean terminateAutoMlIns(Long id);
 
     AutoMlIns getDetailById(Long id);
+
+    void updateAutoMlStatus(Long autoMlId);
 }

ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/AutoMlServiceImpl.java

@@ -6,6 +6,7 @@ import com.ruoyi.platform.domain.AutoMl;
 import com.ruoyi.platform.domain.AutoMlIns;
 import com.ruoyi.platform.mapper.AutoMlDao;
 import com.ruoyi.platform.mapper.AutoMlInsDao;
+import com.ruoyi.platform.service.AutoMlInsService;
 import com.ruoyi.platform.service.AutoMlService;
 import com.ruoyi.platform.utils.FileUtil;
 import com.ruoyi.platform.utils.HttpUtils;
@@ -53,6 +54,9 @@ public class AutoMlServiceImpl implements AutoMlService {
     @Resource
     private AutoMlInsDao autoMlInsDao;
 
+    @Resource
+    private AutoMlInsService autoMlInsService;
+
     @Override
     public Page<AutoMl> queryByPage(String mlName, PageRequest pageRequest) {
         long total = autoMlDao.count(mlName);
@@ -203,7 +207,7 @@ public class AutoMlServiceImpl implements AutoMlService {
             String seed = autoMl.getSeed() != null ? String.valueOf(autoMl.getSeed()) : "1";
             autoMlIns.setRunHistoryPath(outputPath + "smac3-output/run_" + seed + "/runhistory.json");
             autoMlInsDao.insert(autoMlIns);
-
+            autoMlInsService.updateAutoMlStatus(id);
         } catch (Exception e) {
             throw new RuntimeException(e);
         }