From 81b0b55d578bc0449403ef96bda0883895e5873b Mon Sep 17 00:00:00 2001 From: teddy Date: Wed, 11 Feb 2026 11:37:36 +0900 Subject: [PATCH] =?UTF-8?q?=EB=9D=BC=EB=B2=A8=20=EB=8B=A4=EC=9A=B4?= =?UTF-8?q?=EB=A1=9C=EB=93=9C=20=EC=88=98=EC=A0=95?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../label/LabelAllocateApiController.java | 7 +++---- .../postgres/core/LabelAllocateCoreService.java | 16 ++++++++-------- 2 files changed, 11 insertions(+), 12 deletions(-) diff --git a/src/main/java/com/kamco/cd/kamcoback/label/LabelAllocateApiController.java b/src/main/java/com/kamco/cd/kamcoback/label/LabelAllocateApiController.java index 963396d5..1c893adc 100644 --- a/src/main/java/com/kamco/cd/kamcoback/label/LabelAllocateApiController.java +++ b/src/main/java/com/kamco/cd/kamcoback/label/LabelAllocateApiController.java @@ -31,7 +31,6 @@ import java.util.List; import java.util.UUID; import lombok.RequiredArgsConstructor; import lombok.extern.slf4j.Slf4j; -import org.apache.coyote.BadRequestException; import org.springframework.beans.factory.annotation.Value; import org.springframework.data.domain.Page; import org.springframework.http.MediaType; @@ -386,9 +385,9 @@ public class LabelAllocateApiController { @Parameter(example = "6d8d49dc-0c9d-4124-adc7-b9ca610cc394") @PathVariable UUID uuid) throws IOException { - if (!labelAllocateService.isDownloadable(uuid)) { - throw new BadRequestException(); - } + // if (!labelAllocateService.isDownloadable(uuid)) { + // throw new BadRequestException(); + // } String uid = labelAllocateService.findLearnUid(uuid); Path zipPath = Paths.get(responsePath).resolve(uid + ".zip"); diff --git a/src/main/java/com/kamco/cd/kamcoback/postgres/core/LabelAllocateCoreService.java b/src/main/java/com/kamco/cd/kamcoback/postgres/core/LabelAllocateCoreService.java index 482e991e..20f2f4b1 100644 --- a/src/main/java/com/kamco/cd/kamcoback/postgres/core/LabelAllocateCoreService.java +++ b/src/main/java/com/kamco/cd/kamcoback/postgres/core/LabelAllocateCoreService.java @@ -257,16 +257,16 @@ public class LabelAllocateCoreService { // 파일이 있는지만 확인 Path path = Paths.get(responsePath).resolve(dto.getLearnUid() + ".zip"); + if (!Files.isRegularFile(path)) return false; // exists 포함 - if (!Files.exists(path) || !Files.isRegularFile(path)) { - // 실제 파일만 true (디렉터리는 제외) - return false; - } + String state = dto.getAnalState(); + boolean isLabelingIng = + LabelMngState.ASSIGNED.getId().equals(state) || LabelMngState.ING.getId().equals(state); - // 다운로드 확인할 학습데이터가 라벨링중인 경우 파일 생성여부가 정상인지 확인 - if (dto.getAnalState().equals(LabelMngState.ASSIGNED.getId()) - || dto.getAnalState().equals(LabelMngState.ING.getId())) { - return batchStepHistoryRepository.isDownloadable(dto.getAnalId()); + if (isLabelingIng) { + Long analId = dto.getAnalId(); + if (analId == null) return false; + return batchStepHistoryRepository.isDownloadable(analId); } return true;