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 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 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 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 getUploadStatus( @RequestBody @Valid UploadDto.StatusReq statusReq) { return ApiResponseDto.ok(uploadService.getUploadStatus(statusReq)); } */ }