모델학습 데이터셋 선택 목록 수정

src/main/java/com/kamco/cd/training/model/ModelTrainMngApiController.java

@@ -1,6 +1,8 @@
 package com.kamco.cd.training.model;
 
 import com.kamco.cd.training.config.api.ApiResponseDto;
+import com.kamco.cd.training.dataset.dto.DatasetDto;
+import com.kamco.cd.training.dataset.dto.DatasetDto.SelectDataSet;
 import com.kamco.cd.training.model.dto.ModelConfigDto;
 import com.kamco.cd.training.model.dto.ModelTrainMngDto;
 import com.kamco.cd.training.model.dto.ModelTrainMngDto.Basic;
@@ -13,6 +15,7 @@ import io.swagger.v3.oas.annotations.responses.ApiResponse;
 import io.swagger.v3.oas.annotations.responses.ApiResponses;
 import io.swagger.v3.oas.annotations.tags.Tag;
 import jakarta.validation.Valid;
+import java.util.List;
 import java.util.UUID;
 import lombok.RequiredArgsConstructor;
 import org.springframework.data.domain.Page;
@@ -110,4 +113,34 @@ public class ModelTrainMngApiController {
   public ApiResponseDto<ModelConfigDto.Basic> updateModelTrain(@PathVariable UUID uuid) {
     return ApiResponseDto.ok(modelTrainMngService.getModelConfigByModelId(uuid));
   }
+
+  @Operation(summary = "학습데이터관리 상세  조회", description = "학습데이터관리 상세 정보를 조회합니다.")
+  @ApiResponses(
+      value = {
+        @ApiResponse(
+            responseCode = "200",
+            description = "조회 성공",
+            content =
+                @Content(
+                    mediaType = "application/json",
+                    schema = @Schema(implementation = DatasetDto.Basic.class))),
+        @ApiResponse(responseCode = "404", description = "데이터셋을 찾을 수 없음", content = @Content),
+        @ApiResponse(responseCode = "500", description = "서버 오류", content = @Content)
+      })
+  @GetMapping("/select-dataset-list")
+  public ApiResponseDto<List<SelectDataSet>> getDatasetSelectList(
+      @Parameter(
+              description = "모델 구분",
+              example = "",
+              schema = @Schema(allowableValues = {"G1", "G2", "G3"}))
+          @RequestParam
+          String modelType,
+      @Parameter(
+              description = "선택 구분",
+              example = "",
+              schema = @Schema(allowableValues = {"CURRENT", "DELIVER", "PRODUCTION"}))
+          @RequestParam
+          String selectType) {
+    return ApiResponseDto.ok(modelTrainMngService.getDatasetSelectList(modelType, selectType));
+  }
 }

src/main/java/com/kamco/cd/training/model/service/ModelTrainMngService.java

@@ -2,12 +2,14 @@ package com.kamco.cd.training.model.service;
 
 import com.kamco.cd.training.common.dto.HyperParam;
 import com.kamco.cd.training.common.enums.HyperParamSelectType;
+import com.kamco.cd.training.dataset.dto.DatasetDto.SelectDataSet;
 import com.kamco.cd.training.hyperparam.dto.HyperParamDto;
 import com.kamco.cd.training.model.dto.ModelConfigDto;
 import com.kamco.cd.training.model.dto.ModelTrainMngDto;
 import com.kamco.cd.training.model.dto.ModelTrainMngDto.SearchReq;
 import com.kamco.cd.training.postgres.core.HyperParamCoreService;
 import com.kamco.cd.training.postgres.core.ModelTrainMngCoreService;
+import java.util.List;
 import java.util.UUID;
 import lombok.RequiredArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
@@ -85,4 +87,19 @@ public class ModelTrainMngService {
   public ModelConfigDto.Basic getModelConfigByModelId(UUID uuid) {
     return modelTrainMngCoreService.findModelConfigByModelId(uuid);
   }
+
+  /**
+   * 모델별 데이터셋 목록 조회
+   *
+   * @param modelType
+   * @param selectType
+   * @return
+   */
+  public List<SelectDataSet> getDatasetSelectList(String modelType, String selectType) {
+    if (modelType.equals("G1")) {
+      return modelTrainMngCoreService.getDatasetSelectM1List(modelType, selectType);
+    } else {
+      return modelTrainMngCoreService.getDatasetSelectM2M3List(modelType, selectType);
+    }
+  }
 }