파일 count 기능 추가

src/main/java/com/kamco/cd/training/train/TrainApiController.java

@@ -1,6 +1,7 @@
 package com.kamco.cd.training.train;
 
 import com.kamco.cd.training.config.api.ApiResponseDto;
+import com.kamco.cd.training.train.service.DataSetCountersService;
 import com.kamco.cd.training.train.service.TestJobService;
 import com.kamco.cd.training.train.service.TrainJobService;
 import io.swagger.v3.oas.annotations.Operation;
@@ -25,6 +26,7 @@ public class TrainApiController {
 
   private final TrainJobService trainJobService;
   private final TestJobService testJobService;
+  private final DataSetCountersService dataSetCountersService;
 
   @Operation(summary = "학습 실행", description = "학습 실행 API")
   @ApiResponses(
@@ -45,7 +47,9 @@ public class TrainApiController {
           @PathVariable
           UUID uuid) {
     Long modelId = trainJobService.getModelIdByUuid(uuid);
+    trainJobService.createTmpFile(uuid);
     trainJobService.enqueue(modelId);
+
     return ApiResponseDto.ok("ok");
   }
 
@@ -186,4 +190,27 @@ public class TrainApiController {
 
     return ApiResponseDto.ok(trainJobService.createTmpFile(uuid));
   }
+
+  @Operation(summary = "getCount", description = "getCount 서버 로그확인")
+  @ApiResponses(
+      value = {
+        @ApiResponse(
+            responseCode = "200",
+            description = "데이터셋 tmp 파일생성 성공",
+            content =
+                @Content(
+                    mediaType = "application/json",
+                    schema = @Schema(implementation = String.class))),
+        @ApiResponse(responseCode = "400", description = "잘못된 검색 조건", content = @Content),
+        @ApiResponse(responseCode = "500", description = "서버 오류", content = @Content)
+      })
+  @PostMapping("/counts/{uuid}")
+  public ApiResponseDto<Void> getCount(
+      @Parameter(description = "uuid", example = "80a0e544-36ed-4999-b705-97427f23337d")
+          @PathVariable
+          UUID uuid) {
+    Long modelId = trainJobService.getModelIdByUuid(uuid);
+    dataSetCountersService.getCount(modelId);
+    return ApiResponseDto.ok(null);
+  }
 }