JavaProjectRepo/模型训练面板编码实现规划.md

模型训练面板编码实现规划

本规划基于《模型训练面板实现过程.md》，旨在提供具体的编码指导。核心原则：简化开发，直接使用 Domain 实体，避免繁琐的 DTO 转换。

1. 数据库与实体层

1.1 数据库表 (model_train_task)

请确保数据库已执行以下 SQL：

CREATE TABLE IF NOT EXISTS `model_train_task` (
  `task_id` char(36) NOT NULL COMMENT '任务ID',
  `task_name` varchar(100) DEFAULT NULL COMMENT '任务名称',
  `algorithm_type` varchar(50) DEFAULT NULL COMMENT '算法类型',
  `device_type` varchar(50) DEFAULT NULL COMMENT '设备类型',
  `dataset_path` varchar(255) DEFAULT NULL COMMENT '数据集路径',
  `train_params` json DEFAULT NULL COMMENT '训练参数',
  `status` varchar(20) DEFAULT 'PENDING' COMMENT '状态',
  `metrics` json DEFAULT NULL COMMENT '训练指标',
  `model_output_path` varchar(255) DEFAULT NULL COMMENT '临时模型路径',
  `feature_map_snapshot` json DEFAULT NULL COMMENT '特征映射',
  `metrics_image_path` varchar(255) DEFAULT NULL COMMENT '指标图路径',
  `error_log` text DEFAULT NULL COMMENT '错误日志',
  `created_at` datetime DEFAULT CURRENT_TIMESTAMP,
  `updated_at` datetime DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP,
  PRIMARY KEY (`task_id`)
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_general_ci;

1.2 Domain 实体 (ModelTrainTask.java)

• 位置: com.yfd.business.css.domain
• 注解: @TableName("model_train_task"), @TableId(type = IdType.ASSIGN_UUID)
• JSON 字段处理: 使用 MyBatis-Plus 的 @TableField(typeHandler = JacksonTypeHandler.class) 处理 trainParams, metrics, featureMapSnapshot 等 JSON 字段。建议直接映射为 Map<String, Object> 或 JsonNode，或者简单的 String（前端解析）。为了简化，使用 String 存储 JSON 字符串。

@Data
@TableName(value = "model_train_task", autoResultMap = true)
public class ModelTrainTask implements Serializable {
    @TableId(type = IdType.ASSIGN_UUID)
    private String taskId;
    private String taskName;
    private String algorithmType;
    private String deviceType;
    private String datasetPath;
    
    @TableField(typeHandler = JacksonTypeHandler.class)
    private Map<String, Object> trainParams; // 或 String
    
    private String status; // PENDING, TRAINING, SUCCESS, FAILED
    
    @TableField(typeHandler = JacksonTypeHandler.class)
    private Map<String, Object> metrics;
    
    private String modelOutputPath;
    
    @TableField(typeHandler = JacksonTypeHandler.class)
    private Map<String, Object> featureMapSnapshot;
    
    private String metricsImagePath;
    private String errorLog;
    
    @TableField(fill = FieldFill.INSERT)
    private LocalDateTime createdAt;
    
    @TableField(fill = FieldFill.INSERT_UPDATE)
    private LocalDateTime updatedAt;
}

2. Mapper 层 (ModelTrainTaskMapper.java)

• 位置: com.yfd.business.css.mapper
• 继承: BaseMapper<ModelTrainTask>

3. Service 层

3.1 接口 (ModelTrainService.java)

• 位置: com.yfd.business.css.service
• 继承: IService<ModelTrainTask>
• 核心方法:
  • String uploadDataset(MultipartFile file): 上传文件，返回路径。
  • String submitTask(ModelTrainTask task): 提交任务（接收前端传来的实体，补充初始状态后保存，并异步调用 Python）。
  • ModelTrainTask syncTaskStatus(String taskId): 同步并返回最新状态。
  • boolean publishModel(String taskId, String versionTag, String description): 发布模型。

3.2 实现 (ModelTrainServiceImpl.java)

• 位置: com.yfd.business.css.service.impl
• 依赖: ModelTrainTaskMapper, RestTemplate (调用 Python), AlgorithmModelService (发布模型)。

4. Controller 层 (ModelTrainController.java)

• 位置: com.yfd.business.css.controller
• 原则: 直接接收和返回 ModelTrainTask 实体或 Map。

4.1 接口定义

1. 上传数据集

   • POST /train/upload
   • 参数: @RequestParam("file") MultipartFile file
   • 返回: Result<String> (文件路径)

2. 提交任务

   • POST /train/submit
   • 参数: @RequestBody ModelTrainTask task
   • 逻辑: 前端构建好 task 对象（包含 name, algoType, deviceType, datasetPath, trainParams），传给后端。后端设置 status=PENDING, save, async call python。

3. 查询任务列表

   • GET /train/list
   • 参数: PageQuery, ModelTrainTask query (查询条件)
   • 返回: Result<Page<ModelTrainTask>>

4. 查询任务详情/状态

   • GET /train/status/{taskId}
   • 逻辑: 调用 Service 的 syncTaskStatus，触发一次 Python 状态查询，更新数据库，然后返回最新实体。

5. 发布模型

   • POST /train/publish
   • 参数: Map<String, String> body (包含 taskId, versionTag, description)
   • 逻辑: 检查状态 -> 移动文件 -> 插入 algorithm_model 表。

5. Python 服务交互

• 调用方式: RestTemplate
• API 对应:
  • Java submitTask -> Python POST /v1/train
  • Java syncTaskStatus -> Python GET /v1/train/status/{taskId}

6. 开发步骤建议

1. 创建表：执行 SQL。
2. 生成代码：创建 Domain, Mapper, Service, Controller。
3. 实现上传接口：确保文件能保存到指定目录（如 data/upload/）。
4. 实现提交接口：
   • 先实现保存到数据库。
   • 再实现 RestTemplate 调用 Python（Mock 阶段可先跳过 Python 调用，直接设为 SUCCESS）。
5. 实现状态查询：对接 Python 查询接口。
6. 实现发布接口：操作 algorithm_model 表。