From 8fc2a4b5cfccc506f7718f4c56bee29b2cbf2d76 Mon Sep 17 00:00:00 2001
From: chenzhihang <709011834@qq.com>
Date: Wed, 8 Jan 2025 16:46:04 +0800
Subject: [PATCH] =?UTF-8?q?=E8=87=AA=E5=8A=A8=E8=B6=85=E5=8F=82=E6=95=B0?=
 =?UTF-8?q?=E5=AF=BB=E4=BC=98=E5=AE=9E=E9=AA=8C=E5=8A=9F=E8=83=BD=E5=BC=80?=
 =?UTF-8?q?=E5=8F=91?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../src/main/java/com/ruoyi/platform/domain/Ray.java       | 3 +++
 .../src/main/java/com/ruoyi/platform/vo/RayVo.java         | 7 +++++--
 .../resources/mapper/managementPlatform/RayDaoMapper.xml   | 7 +++++--
 3 files changed, 13 insertions(+), 4 deletions(-)

diff --git a/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/domain/Ray.java b/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/domain/Ray.java
index 1cd7dbc8..47a68380 100644
--- a/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/domain/Ray.java
+++ b/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/domain/Ray.java
@@ -23,6 +23,9 @@ public class Ray {
     @ApiModelProperty(value = "数据集")
     private String dataset;
 
+    @ApiModelProperty(value = "数据集挂载路径")
+    private String datasetPath;
+
     @ApiModelProperty(value = "代码")
     private String code;
 
diff --git a/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/vo/RayVo.java b/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/vo/RayVo.java
index 4f6997c2..0f3a7eda 100644
--- a/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/vo/RayVo.java
+++ b/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/vo/RayVo.java
@@ -48,10 +48,10 @@ public class RayVo {
     @ApiModelProperty(value = "指标最大化或最小化，min or max")
     private String mode;
 
-    @ApiModelProperty(value = "搜索算法为ASHA，HyperBand时传入，每次试验的最大时间单位。测试将在max_t时间单位后停止。")
+    @ApiModelProperty(value = "单次试验最大时间：单位秒，搜索算法为ASHA，HyperBand时传入，每次试验的最大时间单位。测试将在max_t时间单位后停止。")
     private Integer maxT;
 
-    @ApiModelProperty(value = "搜索算法为MedianStopping时传入，计算中位数的最小试验数。")
+    @ApiModelProperty(value = "计算中位数的最小试验数：搜索算法为MedianStopping时传入，计算中位数的最小试验数。")
     private Integer minSamplesRequired;
 
     @ApiModelProperty(value = "使用cpu数")
@@ -76,4 +76,7 @@ public class RayVo {
     private Map<String, Object> code;
 
     private Map<String, Object> dataset;
+
+    @ApiModelProperty(value = "数据集挂载路径")
+    private String datasetPath;
 }
diff --git a/ruoyi-modules/management-platform/src/main/resources/mapper/managementPlatform/RayDaoMapper.xml b/ruoyi-modules/management-platform/src/main/resources/mapper/managementPlatform/RayDaoMapper.xml
index 2c2e6196..d592d444 100644
--- a/ruoyi-modules/management-platform/src/main/resources/mapper/managementPlatform/RayDaoMapper.xml
+++ b/ruoyi-modules/management-platform/src/main/resources/mapper/managementPlatform/RayDaoMapper.xml
@@ -2,10 +2,10 @@
 <!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
 <mapper namespace="com.ruoyi.platform.mapper.RayDao">
     <insert id="save">
-        insert into ray(name, description, dataset, code, main_py, num_samples, parameters, points_to_evaluate, storage_path,
+        insert into ray(name, description, dataset, dataset_path, code, main_py, num_samples, parameters, points_to_evaluate, storage_path,
                         search_alg, scheduler, metric, mode, max_t,
                         min_samples_required, cpu, gpu, create_by, update_by)
-        values (#{ray.name}, #{ray.dataset}, #{ray.code}, #{ray.mainPy}, #{ray.numSamples}, #{ray.parameters},
+        values (#{ray.name}, #{ray.dataset}, #{ray.datasetPath}, #{ray.code}, #{ray.mainPy}, #{ray.numSamples}, #{ray.parameters},
                 #{ray.pointsToEvaluate}, #{ray.storagePath},
                 #{ray.searchAlg}, #{ray.scheduler}, #{ray.metric}, #{ray.mode}, #{ray.maxT}, #{ray.minSamplesRequired},
                 #{ray.cpu}, #{ray.gpu}, #{ray.createBy}, #{ray.updateBy})
@@ -23,6 +23,9 @@
             <if test="ray.dataset != null and ray.dataset !=''">
                 dataset = #{ray.dataset},
             </if>
+            <if test="ray.datasetPath != null and ray.datasetPath !=''">
+                dataset_path = #{ray.datasetPath},
+            </if>
             <if test="ray.code != null and ray.code !=''">
                 code = #{ray.code},
             </if>