chunk업로드 공통, 모델관리 수정

2026-01-08 18:35:19 +09:00
parent 4d2135ec4c
commit eb88717791
19 changed files with 1335 additions and 23 deletions
--- a/src/main/java/com/kamco/cd/kamcoback/upload/UploadApiController.java
+++ b/src/main/java/com/kamco/cd/kamcoback/upload/UploadApiController.java
@@ -0,0 +1,144 @@
+package com.kamco.cd.kamcoback.upload;
+
+import com.kamco.cd.kamcoback.config.api.ApiResponseDto;
+import com.kamco.cd.kamcoback.upload.dto.UploadDto;
+import com.kamco.cd.kamcoback.upload.service.UploadService;
+import io.swagger.v3.oas.annotations.Operation;
+import io.swagger.v3.oas.annotations.media.Content;
+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 java.util.UUID;
+import lombok.RequiredArgsConstructor;
+import org.springframework.beans.factory.annotation.Value;
+import org.springframework.http.MediaType;
+import org.springframework.web.bind.annotation.PathVariable;
+import org.springframework.web.bind.annotation.PostMapping;
+import org.springframework.web.bind.annotation.PutMapping;
+import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.RequestParam;
+import org.springframework.web.bind.annotation.RequestPart;
+import org.springframework.web.bind.annotation.RestController;
+import org.springframework.web.multipart.MultipartFile;
+
+@Tag(name = "파일 업로드", description = "대용량 파일 업로드 API")
+@RestController
+@RequestMapping("/api/upload")
+@RequiredArgsConstructor
+public class UploadApiController {
+
+  private final UploadService uploadService;
+
+  @Value("${file.sync-root-dir}")
+  private String syncRootDir;
+
+  @Value("${file.sync-tmp-dir}")
+  private String syncTmpDir;
+
+  @Value("${file.sync-file-extention}")
+  private String syncFileExtention;
+
+  @Value("${file.dataset-dir}")
+  private String datasetDir;
+
+  @Value("${file.dataset-tmp-dir}")
+  private String datasetTmpDir;
+
+  @Value("${file.model-dir}")
+  private String modelDir;
+
+  @Value("${file.model-tmp-dir}")
+  private String modelTmpDir;
+
+  /*
+    @Operation(summary = "데이터셋 대용량 업로드 세션 시작", description = "데이터셋 대용량 파일 업로드 세션을 시작합니다.")
+    @ApiResponses(
+        value = {
+          @ApiResponse(
+              responseCode = "201",
+              description = "세션 생성 성공",
+              content =
+                  @Content(
+                      mediaType = "application/json",
+                      schema = @Schema(implementation = UploadDto.InitRes.class))),
+          @ApiResponse(responseCode = "400", description = "잘못된 요청 데이터", content = @Content),
+          @ApiResponse(responseCode = "500", description = "서버 오류", content = @Content)
+        })
+    @PostMapping("/chunk-upload-init")
+    public ApiResponseDto<DmlReturn> initUpload(
+        @RequestBody @Valid UploadDto.InitReq initReq) {
+      return ApiResponseDto.createOK(uploadService.initUpload(initReq));
+    }
+  */
+
+  @Operation(summary = "데이터셋 대용량 파일 분할 전송", description = "데이터셋 파일 대용량 파일을 청크 단위로 전송합니다.")
+  @ApiResponses(
+      value = {
+        @ApiResponse(responseCode = "200", description = "청크 업로드 성공", content = @Content),
+        @ApiResponse(responseCode = "400", description = "잘못된 요청 데이터", content = @Content),
+        @ApiResponse(responseCode = "404", description = "업로드 세션을 찾을 수 없음", content = @Content),
+        @ApiResponse(responseCode = "500", description = "서버 오류", content = @Content)
+      })
+  @PostMapping(value = "/file-chunk-upload", consumes = MediaType.MULTIPART_FORM_DATA_VALUE)
+  public ApiResponseDto<UploadDto.UploadRes> fileChunkUpload(
+    @RequestParam("uuid") UUID uuid,
+    @RequestParam("fileName") String fileName,
+    @RequestParam("fileSize") long fileSize,
+    // @RequestParam("fileHash") String fileHash,
+    @RequestParam("chunkIndex") Integer chunkIndex,
+    @RequestParam("chunkTotalIndex") Integer chunkTotalIndex,
+    @RequestPart("chunkFile") MultipartFile chunkFile) {
+
+    String uploadDivi = "dataset";
+
+    UploadDto.UploadAddReq upAddReqDto = new UploadDto.UploadAddReq();
+    upAddReqDto.setDatasetId(0L);
+    upAddReqDto.setUuid(uuid);
+    upAddReqDto.setFileName(fileName);
+    upAddReqDto.setFileSize(fileSize);
+    upAddReqDto.setChunkIndex(chunkIndex);
+    upAddReqDto.setChunkTotalIndex(chunkTotalIndex);
+    upAddReqDto.setUploadDivi(uploadDivi);
+    upAddReqDto.setFinalPath(datasetDir);
+    upAddReqDto.setTempPath(datasetTmpDir);
+
+    System.out.println("uuid === "+ uuid);
+
+    return ApiResponseDto.ok(uploadService.uploadChunk(upAddReqDto, chunkFile));
+  }
+
+  @Operation(summary = "업로드 완료된 파일 병합", description = "업로드 완료 및 파일 병합을 요청합니다.")
+  @ApiResponses(
+      value = {
+        @ApiResponse(responseCode = "200", description = "업로드 완료 성공", content = @Content),
+        @ApiResponse(responseCode = "400", description = "잘못된 요청", content = @Content),
+        @ApiResponse(responseCode = "404", description = "업로드 세션을 찾을 수 없음", content = @Content),
+        @ApiResponse(responseCode = "500", description = "서버 오류", content = @Content)
+      })
+  @PutMapping("/chunk-upload-complete/{uuid}")
+  public ApiResponseDto<UploadDto.UploadRes> completeUpload(
+    @PathVariable UUID uuid) {
+    return ApiResponseDto.ok(uploadService.completeUpload(uuid));
+  }
+
+  /*
+  @Operation(summary = "업로드 상태 조회", description = "업로드 진행 상태를 조회합니다.")
+  @ApiResponses(
+      value = {
+        @ApiResponse(
+            responseCode = "200",
+            description = "조회 성공",
+            content =
+                @Content(
+                    mediaType = "application/json",
+                    schema = @Schema(implementation = UploadDto.Status.class))),
+        @ApiResponse(responseCode = "404", description = "업로드 세션을 찾을 수 없음", content = @Content),
+        @ApiResponse(responseCode = "500", description = "서버 오류", content = @Content)
+      })
+  @PostMapping("/status")
+  public ApiResponseDto<UploadDto.Status> getUploadStatus(
+      @RequestBody @Valid UploadDto.StatusReq statusReq) {
+    return ApiResponseDto.ok(uploadService.getUploadStatus(statusReq));
+  }
+  */
+}