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

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<String, Object> datasetPreprocessMap = (Map<String, Object>) datasetPreprocess.get(sourceTaskId);
             //处理project数据
-            Map<String, Object> projectMap = (Map<String, Object>) datasetPreprocessMap.get("project");
-            Map<String, Object> datasets = (Map<String, Object>) datasetPreprocessMap.get("datasets");
-            datasetTempStorage.setName((String) datasets.get("dataset_identifier"));
-            datasetTempStorage.setVersion((String) datasets.get("dataset_version"));
             // 拼接需要的参数
+            Map<String, Object> projectMap = (Map<String, Object>) datasetPreprocessMap.get("project");
             Map<String, Object> 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<Map> datasets = (List<Map>) datasetPreprocessMap.get("datasets");
+            if (datasets != null) {
+                for (Map<String, Object> 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);
+                }
+            }
+
+
         }
 
     }