From 789108aaa3ce1f92eccebdcffe9a2dd323ecfa2c Mon Sep 17 00:00:00 2001
From: NingHuan <2139614357@qq.com>
Date: Fri, 30 May 2025 13:09:47 +0800
Subject: [PATCH] =?UTF-8?q?[=E6=8F=90=E4=BA=A4]:=E5=A2=9E=E5=8A=A0?=
 =?UTF-8?q?=E4=BA=86=E6=9F=A5=E8=AF=A2=E6=95=B0=E6=8D=AE=E9=9B=86=E5=88=97?=
 =?UTF-8?q?=E8=A1=A8=E5=92=8C=E7=94=9F=E5=91=BD=E5=91=A8=E6=9C=9F=E5=88=97?=
 =?UTF-8?q?=E8=A1=A8?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../controller/ModelController.java           | 16 +++++++++++++++
 .../mapper/ModelMapper.java                   | 12 +++++++----
 .../service/Impl/ModelServiceImpl.java        | 20 +++++++++++++++++++
 .../service/ModelService.java                 | 10 ++++++----
 4 files changed, 50 insertions(+), 8 deletions(-)

diff --git a/src/main/java/com/bipt/intelligentapplicationorchestrationservice/controller/ModelController.java b/src/main/java/com/bipt/intelligentapplicationorchestrationservice/controller/ModelController.java
index 11012ae..f78cebd 100644
--- a/src/main/java/com/bipt/intelligentapplicationorchestrationservice/controller/ModelController.java
+++ b/src/main/java/com/bipt/intelligentapplicationorchestrationservice/controller/ModelController.java
@@ -9,6 +9,7 @@ import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.web.bind.annotation.*;
 
 import java.util.List;
+import java.util.Map;
 
 @Slf4j
 @RestController
@@ -75,5 +76,20 @@ public class ModelController {
         return OptResult.success();
     }
 
+    @Operation(summary = "查询生命周期列表")
+    @GetMapping("/listLifeCycle")
+    public OptResult listLifeCycle(){
+        log.info("查询生命周期列表");
+        List<Map<String, String>> lifeCycleList = modelService.listLifeCycle();
+        return OptResult.success(lifeCycleList);
+    }
+
+    @Operation(summary = "查询数据集列表")
+    @GetMapping("/listDataset")
+    public OptResult listDataset(){
+        List<DatasetEntity> datasetList = modelService.listDataset();
+        return OptResult.success(datasetList);
+    }
+
 
 }
diff --git a/src/main/java/com/bipt/intelligentapplicationorchestrationservice/mapper/ModelMapper.java b/src/main/java/com/bipt/intelligentapplicationorchestrationservice/mapper/ModelMapper.java
index fb3106f..c8d5e03 100644
--- a/src/main/java/com/bipt/intelligentapplicationorchestrationservice/mapper/ModelMapper.java
+++ b/src/main/java/com/bipt/intelligentapplicationorchestrationservice/mapper/ModelMapper.java
@@ -1,9 +1,6 @@
 package com.bipt.intelligentapplicationorchestrationservice.mapper;
 
-import com.bipt.intelligentapplicationorchestrationservice.pojo.ModelInfo;
-import com.bipt.intelligentapplicationorchestrationservice.pojo.ModelVO;
-import com.bipt.intelligentapplicationorchestrationservice.pojo.ModelVersion;
-import com.bipt.intelligentapplicationorchestrationservice.pojo.ModelVersionDTO;
+import com.bipt.intelligentapplicationorchestrationservice.pojo.*;
 import org.apache.ibatis.annotations.*;
 
 import java.util.List;
@@ -72,4 +69,11 @@ public interface ModelMapper {
      */
     @Update("update model_version set life_cycle=#{dbValue} where id=#{id}")
     int updateLifeCycleById(Long id, String dbValue);
+
+    /**
+     * 查询数据集列表
+     * @return
+     */
+    @Select("select dataset_id,dataset_name from dataset")
+    List<DatasetEntity> listDataset();
 }
diff --git a/src/main/java/com/bipt/intelligentapplicationorchestrationservice/service/Impl/ModelServiceImpl.java b/src/main/java/com/bipt/intelligentapplicationorchestrationservice/service/Impl/ModelServiceImpl.java
index cc8d32a..8b25ff1 100644
--- a/src/main/java/com/bipt/intelligentapplicationorchestrationservice/service/Impl/ModelServiceImpl.java
+++ b/src/main/java/com/bipt/intelligentapplicationorchestrationservice/service/Impl/ModelServiceImpl.java
@@ -12,7 +12,10 @@ import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
 
 import java.time.LocalDateTime;
+import java.util.Arrays;
 import java.util.List;
+import java.util.Map;
+import java.util.stream.Collectors;
 
 @Slf4j
 @Service
@@ -145,4 +148,21 @@ public class ModelServiceImpl implements ModelService {
         log.info("模型生命周期更新成功，新状态为: {}", targetLifeCycle);
     }
 
+    @Override
+    public List<Map<String, String>> listLifeCycle() {
+        return Arrays.stream(ModelLifecycle.values())
+                .map(lifecycle -> Map.of(
+                        "code", lifecycle.name(),           // 枚举名称（如 "DESIGNING"）
+                        "description", lifecycle.getDescription() // 中文描述（如 "设计中"）
+                ))
+                .collect(Collectors.toList());
+    }
+
+    @Override
+    public List<DatasetEntity> listDataset() {
+        List<DatasetEntity> datasetEntityList = modelMapper.listDataset();
+        return datasetEntityList;
+
+    }
+
 }
diff --git a/src/main/java/com/bipt/intelligentapplicationorchestrationservice/service/ModelService.java b/src/main/java/com/bipt/intelligentapplicationorchestrationservice/service/ModelService.java
index 0933085..dc468f0 100644
--- a/src/main/java/com/bipt/intelligentapplicationorchestrationservice/service/ModelService.java
+++ b/src/main/java/com/bipt/intelligentapplicationorchestrationservice/service/ModelService.java
@@ -1,11 +1,9 @@
 package com.bipt.intelligentapplicationorchestrationservice.service;
 
-import com.bipt.intelligentapplicationorchestrationservice.pojo.ModelDTO;
-import com.bipt.intelligentapplicationorchestrationservice.pojo.ModelVO;
-import com.bipt.intelligentapplicationorchestrationservice.pojo.ModelVersion;
-import com.bipt.intelligentapplicationorchestrationservice.pojo.ModelVersionDTO;
+import com.bipt.intelligentapplicationorchestrationservice.pojo.*;
 
 import java.util.List;
+import java.util.Map;
 
 public interface ModelService {
     void createModel(ModelDTO dto);
@@ -21,4 +19,8 @@ public interface ModelService {
     void deleteModelVersion(Long id);
 
     void updateLifeCycle(Long id, String lifeCycle);
+
+    List<Map<String, String>> listLifeCycle();
+
+    List<DatasetEntity> listDataset();
 }
-- 
2.25.1