Browse Source

Merge remote-tracking branch 'origin/dev' into dev-czh

# Conflicts:
#	ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ExperimentServiceImpl.java
dev-DXTZYK
chenzhihang 1 year ago
parent
commit
7f112c3078
1 changed files with 7 additions and 9 deletions
  1. +7
    -9
      ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ExperimentServiceImpl.java

+ 7
- 9
ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ExperimentServiceImpl.java View File

@@ -9,7 +9,7 @@ import com.ruoyi.platform.domain.dependencydomain.TrainTaskDepency;
import com.ruoyi.platform.mapper.ExperimentDao;
import com.ruoyi.platform.mapper.ExperimentInsDao;
import com.ruoyi.platform.service.*;
import com.ruoyi.platform.utils.DVCUtils;
import com.ruoyi.platform.utils.HttpUtils;
import com.ruoyi.platform.utils.JacksonUtil;
import com.ruoyi.platform.utils.JsonUtils;
import com.ruoyi.platform.utils.NewHttpUtils;
@@ -59,8 +59,6 @@ public class ExperimentServiceImpl implements ExperimentService {
@Resource
@Lazy
private ExperimentInsService experimentInsService;
@Resource
private NewHttpUtils httpUtils;
@Value("${argo.url}")
private String argoUrl;
@Value("${argo.convert}")
@@ -225,7 +223,7 @@ public class ExperimentServiceImpl implements ExperimentService {
String dag = workflow.getDag();
// 调argo转换接口
try {
String convertRes = httpUtils.sendPost(argoUrl + argoConvert, null, dag);
String convertRes = HttpUtils.sendPost(argoUrl + argoConvert, dag);
if (convertRes == null || StringUtils.isEmpty(convertRes)) {
throw new RuntimeException("转换流水线失败");
}
@@ -244,7 +242,7 @@ public class ExperimentServiceImpl implements ExperimentService {

Map<String, Object> output = (Map<String, Object>) converMap.get("output");
// 调argo运行接口
String runRes = httpUtils.sendPost(argoUrl + argoWorkflowRun, null, JsonUtils.mapToJson(runReqMap));
String runRes = HttpUtils.sendPost(argoUrl + argoWorkflowRun, JsonUtils.mapToJson(runReqMap));

if (runRes == null || StringUtils.isEmpty(runRes)) {
throw new RuntimeException("Failed to run workflow.");
@@ -324,10 +322,10 @@ public class ExperimentServiceImpl implements ExperimentService {
List<Map<String, Object>> datasets = (List<Map<String, Object>>) trainInfoDetails.get("datasets");
if (datasets != null) {
//查询名字再回填
for (int i = 0; i < datasets.size(); i++) {
Dataset dataset = datasetService.queryById((Integer) datasets.get(i).get("dataset_id"));
datasets.get(i).put("dataset_name", dataset.getName());
}
// for (int i = 0; i < datasets.size(); i++) {
// Dataset dataset = datasetService.queryById(Integer.valueOf((String) datasets.get(i).get("dataset_id")));
// datasets.get(i).put("dataset_name", dataset.getName());
// }
metricRecordItem.put("datasets", datasets);
}
}


Loading…
Cancel
Save