diff --git a/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/AutoMlInsService.java b/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/AutoMlInsService.java index bb417ed2..2131fc44 100644 --- a/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/AutoMlInsService.java +++ b/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); } diff --git a/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/AutoMlServiceImpl.java b/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/AutoMlServiceImpl.java index 7b842299..d5e436e8 100644 --- a/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/AutoMlServiceImpl.java +++ b/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 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); }