From 7fd38baf4a93894b2f88c354e7ff717f1ef2f280 Mon Sep 17 00:00:00 2001
From: chenzhihang <709011834@qq.com>
Date: Tue, 24 Sep 2024 08:49:57 +0800
Subject: [PATCH] =?UTF-8?q?=E6=A8=A1=E5=9E=8B=E5=85=83=E6=95=B0=E6=8D=AE?=
 =?UTF-8?q?=E8=AE=B0=E5=BD=95=E5=AE=9E=E9=AA=8C=E5=8F=82=E6=95=B0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../service/impl/ExperimentServiceImpl.java     | 17 ++++++++++++++---
 1 file changed, 14 insertions(+), 3 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 27d34ae0..f7b743eb 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
@@ -1,6 +1,8 @@
 package com.ruoyi.platform.service.impl;
 
 import com.alibaba.fastjson2.JSON;
+import com.alibaba.fastjson2.JSONArray;
+import com.alibaba.fastjson2.JSONObject;
 import com.ruoyi.common.security.utils.SecurityUtils;
 import com.ruoyi.platform.annotations.CheckDuplicate;
 import com.ruoyi.platform.constant.Constant;
@@ -294,7 +296,7 @@ public class ExperimentServiceImpl implements ExperimentService {
             ExperimentIns insert = experimentInsService.insert(experimentIns);
             //插入到模型依赖关系表
             if (dependendcy != null && trainInfo != null) {
-                insertModelDependencyNew(dependendcy, trainInfo, output, insert.getId(), experiment.getName());
+                insertModelDependencyNew(dependendcy, trainInfo, output, insert.getId(), experiment.getName(), experiment.getGlobalParam());
             }
 
             Map<String, Object> datasetDependendcy = (Map<String, Object>) converMap2.get("dataset_dependency");
@@ -488,7 +490,7 @@ public class ExperimentServiceImpl implements ExperimentService {
 
     }
 
-    private void insertModelDependencyNew(Map<String, Object> dependendcy, Map<String, Object> trainInfo, Map<String, Object> output, Integer experimentInsId, String experimentName) throws Exception {
+    private void insertModelDependencyNew(Map<String, Object> dependendcy, Map<String, Object> trainInfo, Map<String, Object> output, Integer experimentInsId, String experimentName, String globalParam) throws Exception {
         Iterator<Map.Entry<String, Object>> dependendcyIterator = dependendcy.entrySet().iterator();
         Map<String, Object> modelTrain = (Map<String, Object>) trainInfo.get("model_train");
         Map<String, Object> modelEvaluate = (Map<String, Object>) trainInfo.get("model_evaluate");
@@ -518,8 +520,17 @@ public class ExperimentServiceImpl implements ExperimentService {
             //处理镜像
             Map<String, String> imagesMap = (Map<String, String>) modelTrainMap.get("image");
             modelMetaVo.setImage(imagesMap.get("name"));
+
             //处理训练参数 todo
-//            HashMap<String, Object> trainParam = (HashMap<String, Object>) modelTrainMap.get("params");
+            HashMap<String, Object> trainParam = new HashMap<>();
+            JSONArray jsonArray = JSONArray.parseArray(globalParam);
+            for (int i = 0; i < jsonArray.size(); i++) {
+                JSONObject jsonObject = jsonArray.getJSONObject(i);
+                String paramName = jsonObject.getString("param_name");
+                String paramValue = jsonObject.getString("param_value");
+                trainParam.put(paramName, paramValue);
+            }
+            modelMetaVo.setParams(trainParam);
 
             //处理source数据
             List<Map<String, Object>> modelsList = (List<Map<String, Object>>) modelTrainMap.get("models");