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 077e6c12..7eeae878 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 @@ -470,20 +470,26 @@ public class ExperimentServiceImpl implements ExperimentService { String sourceTaskId = (String) source.get("task_id"); Map datasetPreprocessMap = (Map) datasetPreprocess.get(sourceTaskId); //处理project数据 - Map projectMap = (Map) datasetPreprocessMap.get("project"); - Map datasets = (Map) datasetPreprocessMap.get("datasets"); - datasetTempStorage.setName((String) datasets.get("dataset_identifier")); - datasetTempStorage.setVersion((String) datasets.get("dataset_version")); // 拼接需要的参数 + Map projectMap = (Map) datasetPreprocessMap.get("project"); Map sourceParams = new HashMap<>(); sourceParams.put("experiment_name", experimentName); sourceParams.put("experiment_ins_id", experimentInsId); sourceParams.put("experiment_id", experimentId); sourceParams.put("train_name", sourceTaskId); sourceParams.put("preprocess_code", projectMap); - datasetTempStorage.setSource(JacksonUtil.toJSONString(sourceParams)); - datasetTempStorage.setState(1); - datasetTempStorageService.insert(datasetTempStorage); + 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); + } + } + + } }