diff --git a/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ExperimentServiceImpl.java b/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ExperimentServiceImpl.java index 7eeae878..52ac55da 100644 --- a/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ExperimentServiceImpl.java +++ b/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ExperimentServiceImpl.java @@ -455,8 +455,6 @@ public class ExperimentServiceImpl implements ExperimentService { * 存储数据集元数据到临时表 */ private void insertDatasetTempStorage(Map datasetDependendcy, Map trainInfo, Integer experimentId, Integer experimentInsId, String experimentName) { - DatasetTempStorage datasetTempStorage = new DatasetTempStorage(); - Iterator> dependendcyIterator = datasetDependendcy.entrySet().iterator(); Map datasetExport = (Map) trainInfo.get("dataset_export"); Map datasetPreprocess = (Map) trainInfo.get("general-data-process"); @@ -478,18 +476,25 @@ public class ExperimentServiceImpl implements ExperimentService { sourceParams.put("experiment_id", experimentId); sourceParams.put("train_name", sourceTaskId); sourceParams.put("preprocess_code", projectMap); - List datasets = (List) datasetPreprocessMap.get("datasets"); - if (datasets != null) { - for (Map dataset : datasets) { - datasetTempStorage.setName((String) dataset.get("dataset_identifier")); - datasetTempStorage.setVersion((String) dataset.get("dataset_version")); - datasetTempStorage.setSource(JacksonUtil.toJSONString(sourceParams)); - datasetTempStorage.setState(1); - datasetTempStorageService.insert(datasetTempStorage); - } - } + if (target != null && target.size() > 0){ + for (Map targetMap : target){ + String targetTaskId = (String)targetMap.get("task_id"); + Map datasetExportMap = (Map)datasetExport.get(targetTaskId); + List datasets = (List) datasetExportMap.get("datasets"); + if (datasets != null) { + for (Map dataset : datasets) { + DatasetTempStorage datasetTempStorage = new DatasetTempStorage(); + datasetTempStorage.setName((String) dataset.get("dataset_identifier")); + datasetTempStorage.setVersion((String) dataset.get("dataset_version")); + datasetTempStorage.setSource(JacksonUtil.toJSONString(sourceParams)); + datasetTempStorage.setState(1); + datasetTempStorageService.insert(datasetTempStorage); + } + } + } + } } } diff --git a/ruoyi-modules/management-platform/src/main/resources/bootstrap.yml b/ruoyi-modules/management-platform/src/main/resources/bootstrap.yml index 25460d4c..1815cde0 100644 --- a/ruoyi-modules/management-platform/src/main/resources/bootstrap.yml +++ b/ruoyi-modules/management-platform/src/main/resources/bootstrap.yml @@ -9,7 +9,7 @@ spring: name: management-platform profiles: # 环境配置 - active: dev + active: dev1 cloud: nacos: discovery: