From 583918f8ac0f67dd4ea3eaeb93bb82bae7763441 Mon Sep 17 00:00:00 2001
From: fanshuai <1141904845@qq.com>
Date: Fri, 27 Sep 2024 10:15:46 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9=E4=B8=BAgeneral-data-process?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../service/impl/ExperimentServiceImpl.java   | 29 +++++++++++--------
 .../src/main/resources/bootstrap.yml          |  2 +-
 2 files changed, 18 insertions(+), 13 deletions(-)

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 b86865f2..8e0eefcf 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
@@ -457,8 +457,6 @@ public class ExperimentServiceImpl implements ExperimentService {
      * 存储数据集元数据到临时表
      */
     private void insertDatasetTempStorage(Map<String, Object> datasetDependendcy, Map<String, Object> trainInfo, Integer experimentId, Integer experimentInsId, String experimentName) {
-        DatasetTempStorage datasetTempStorage = new DatasetTempStorage();
-
         Iterator<Map.Entry<String, Object>> dependendcyIterator = datasetDependendcy.entrySet().iterator();
         Map<String, Object> datasetExport = (Map<String, Object>) trainInfo.get("dataset_export");
         Map<String, Object> datasetPreprocess = (Map<String, Object>) trainInfo.get("general-data-process");
@@ -480,18 +478,25 @@ public class ExperimentServiceImpl implements ExperimentService {
             sourceParams.put("experiment_id", experimentId);
             sourceParams.put("train_name", sourceTaskId);
             sourceParams.put("preprocess_code", projectMap);
-            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);
-                }
-            }
 
+            if (target != null && target.size() > 0){
+                for (Map<String, Object> targetMap : target){
+                    String targetTaskId = (String)targetMap.get("task_id");
+                    Map<String, Object>  datasetExportMap =  (Map<String, Object>)datasetExport.get(targetTaskId);
+                    List<Map> datasets = (List<Map>) datasetExportMap.get("datasets");
+                    if (datasets != null) {
+                        for (Map<String, Object> 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: