From 2e42c6fbf97c5c5d8fa5569837abd30ae8a4e17f Mon Sep 17 00:00:00 2001
From: Jiale <cjl601523@gmail.com>
Date: Tue, 27 May 2025 20:52:55 +0800
Subject: [PATCH] =?UTF-8?q?feat(model):=20=E6=B7=BB=E5=8A=A0=E6=A8=A1?=
 =?UTF-8?q?=E5=9E=8B=E8=AE=AD=E7=BB=83=E6=97=B6=E9=97=B4=E4=BF=A1=E6=81=AF?=
 =?UTF-8?q?=E5=B9=B6=E4=BC=98=E5=8C=96=E6=A8=A1=E5=9E=8B=E4=BF=A1=E6=81=AF?=
 =?UTF-8?q?=E6=9B=B4=E6=96=B0=E5=8A=9F=E8=83=BD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- 在 ModelInfoVO 中添加 TrainTime 类，用于记录模型训练时间相关信息
- 移除 ModelController 中未使用的 CreateModelInfo 导入
- 实现 ModelServiceImpl 中的 updateModelInfo 方法，使用 ModelCfg 对象更新模型信息
---
 .../admin/model/ModelController.java          |  1 -
 .../admin/model/vo/ModelInfoVO.java           | 34 +++++++++++++++++++
 .../service/model/impl/ModelServiceImpl.java  |  7 +++-
 3 files changed, 40 insertions(+), 2 deletions(-)

diff --git a/yudao-module-alert/yudao-module-alert-biz/src/main/java/cn/iocoder/yudao/module/alert/controller/admin/model/ModelController.java b/yudao-module-alert/yudao-module-alert-biz/src/main/java/cn/iocoder/yudao/module/alert/controller/admin/model/ModelController.java
index fb825dc..f06d08c 100644
--- a/yudao-module-alert/yudao-module-alert-biz/src/main/java/cn/iocoder/yudao/module/alert/controller/admin/model/ModelController.java
+++ b/yudao-module-alert/yudao-module-alert-biz/src/main/java/cn/iocoder/yudao/module/alert/controller/admin/model/ModelController.java
@@ -1,7 +1,6 @@
 package cn.iocoder.yudao.module.alert.controller.admin.model;
 
 import cn.iocoder.yudao.framework.common.pojo.CommonResult;
-import cn.iocoder.yudao.module.alert.controller.admin.model.vo.CreateModelInfo;
 import cn.iocoder.yudao.module.alert.controller.admin.model.vo.ModelInfoVO;
 import cn.iocoder.yudao.module.alert.controller.admin.model.vo.ModelSelectQuery;
 import cn.iocoder.yudao.module.alert.controller.admin.model.vo.ModelSimpleVO;
diff --git a/yudao-module-alert/yudao-module-alert-biz/src/main/java/cn/iocoder/yudao/module/alert/controller/admin/model/vo/ModelInfoVO.java b/yudao-module-alert/yudao-module-alert-biz/src/main/java/cn/iocoder/yudao/module/alert/controller/admin/model/vo/ModelInfoVO.java
index a54c1c9..b1f485c 100644
--- a/yudao-module-alert/yudao-module-alert-biz/src/main/java/cn/iocoder/yudao/module/alert/controller/admin/model/vo/ModelInfoVO.java
+++ b/yudao-module-alert/yudao-module-alert-biz/src/main/java/cn/iocoder/yudao/module/alert/controller/admin/model/vo/ModelInfoVO.java
@@ -46,11 +46,45 @@ public class ModelInfoVO {
 
     private List<Point> pointInfo;
 
+    private List<TrainTime> trainTime;
+
     private Integer algorithmId;
 
     private Integer systemId;
 
     private Integer sampling;
 
+    @Data
+    public static class TrainTime {
+        /**
+         * 开始时间
+         */
+        private String st;
+
+        /**
+         * 结束时间
+         */
+        private String et;
+
+        /**
+         * 时长（秒）
+         */
+        private Integer duration;
+
+        /**
+         * 采样数量
+         */
+        private Integer number;
+
+        /**
+         * 清洗样本数
+         */
+        private Integer filter;
+
+        /**
+         * 有效样本数
+         */
+        private Integer mode;
+    }
 
 }
diff --git a/yudao-module-alert/yudao-module-alert-biz/src/main/java/cn/iocoder/yudao/module/alert/service/model/impl/ModelServiceImpl.java b/yudao-module-alert/yudao-module-alert-biz/src/main/java/cn/iocoder/yudao/module/alert/service/model/impl/ModelServiceImpl.java
index 9e2b92a..5c6addd 100644
--- a/yudao-module-alert/yudao-module-alert-biz/src/main/java/cn/iocoder/yudao/module/alert/service/model/impl/ModelServiceImpl.java
+++ b/yudao-module-alert/yudao-module-alert-biz/src/main/java/cn/iocoder/yudao/module/alert/service/model/impl/ModelServiceImpl.java
@@ -115,6 +115,11 @@ public class ModelServiceImpl implements ModelService {
 
     @Override
     public Boolean updateModelInfo(ModelInfoVO modelInfo) {
-        return false;
+        ModelCfg modelCfg = ModelCfg.builder()
+                .id(modelInfo.getId())
+                .modelInfo(JsonUtils.toJsonString(modelInfo))
+                .updateTime(new Date())
+                .build();
+        return modelCfgService.updateById(modelCfg);
     }
 }