diff --git a/src/main/java/com/kamco/cd/training/model/ModelTrainMngApiController.java b/src/main/java/com/kamco/cd/training/model/ModelTrainMngApiController.java index c6fd192..8068bd0 100644 --- a/src/main/java/com/kamco/cd/training/model/ModelTrainMngApiController.java +++ b/src/main/java/com/kamco/cd/training/model/ModelTrainMngApiController.java @@ -8,6 +8,8 @@ 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; import com.kamco.cd.training.model.service.ModelTrainMngService; +import com.kamco.cd.training.train.service.ModelTestMetricsJobService; +import com.kamco.cd.training.train.service.ModelTrainMetricsJobService; import io.swagger.v3.oas.annotations.Operation; import io.swagger.v3.oas.annotations.Parameter; import io.swagger.v3.oas.annotations.media.Content; @@ -16,6 +18,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.io.IOException; import java.util.List; import java.util.UUID; import lombok.RequiredArgsConstructor; @@ -35,6 +38,8 @@ import org.springframework.web.bind.annotation.RestController; @RequestMapping("/api/models") public class ModelTrainMngApiController { private final ModelTrainMngService modelTrainMngService; + private final ModelTrainMetricsJobService modelTrainMetricsJobService; + private final ModelTestMetricsJobService modelTestMetricsJobService; @Operation(summary = "모델학습 목록 조회", description = "모델학습 목록 조회 API") @ApiResponses( @@ -167,4 +172,44 @@ public class ModelTrainMngApiController { public ApiResponseDto findModelStep1InProgressCnt() { return ApiResponseDto.ok(modelTrainMngService.findModelStep1InProgressCnt()); } + + @Operation( + summary = "스케줄러 findTrainValidMetricCsvFiles", + description = "스케줄러 findTrainValidMetricCsvFiles") + @ApiResponses( + value = { + @ApiResponse( + responseCode = "200", + description = "검색 성공", + content = + @Content( + mediaType = "application/json", + schema = @Schema(implementation = Long.class))), + @ApiResponse(responseCode = "400", description = "잘못된 검색 조건", content = @Content), + @ApiResponse(responseCode = "500", description = "서버 오류", content = @Content) + }) + @GetMapping("/schedule-trainvalid") + public ApiResponseDto findTrainValidMetricCsvFiles() { + modelTrainMetricsJobService.findTrainValidMetricCsvFiles(); + return ApiResponseDto.ok(null); + } + + @Operation(summary = "스케줄러 findTestMetricCsvFiles", description = "스케줄러 findTestMetricCsvFiles") + @ApiResponses( + value = { + @ApiResponse( + responseCode = "200", + description = "검색 성공", + content = + @Content( + mediaType = "application/json", + schema = @Schema(implementation = Long.class))), + @ApiResponse(responseCode = "400", description = "잘못된 검색 조건", content = @Content), + @ApiResponse(responseCode = "500", description = "서버 오류", content = @Content) + }) + @GetMapping("/schedule-test") + public ApiResponseDto findTestValidMetricCsvFiles() throws IOException { + modelTestMetricsJobService.findTestValidMetricCsvFiles(); + return ApiResponseDto.ok(null); + } }