From 6352f74b08da13da4a7bd54ec3ed7cd10a547007 Mon Sep 17 00:00:00 2001 From: "gayoun.park" Date: Tue, 10 Feb 2026 18:28:42 +0900 Subject: [PATCH] =?UTF-8?q?=EC=97=85=EB=A1=9C=EB=93=9C=20=EC=84=B1?= =?UTF-8?q?=EA=B3=B5=20=ED=9B=84=20COMPLETED=20=ED=95=B4=EC=A3=BC=EA=B8=B0?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../cd/training/dataset/service/DatasetService.java | 2 ++ .../cd/training/postgres/core/DatasetCoreService.java | 9 +++++++++ 2 files changed, 11 insertions(+) diff --git a/src/main/java/com/kamco/cd/training/dataset/service/DatasetService.java b/src/main/java/com/kamco/cd/training/dataset/service/DatasetService.java index 588982d..c27503f 100644 --- a/src/main/java/com/kamco/cd/training/dataset/service/DatasetService.java +++ b/src/main/java/com/kamco/cd/training/dataset/service/DatasetService.java @@ -194,6 +194,8 @@ public class DatasetService { log.error(e.getMessage()); return new ResponseObj(ApiResponseCode.INTERNAL_SERVER_ERROR, e.getMessage()); } + + datasetCoreService.updateDatasetUploadStatus(datasetUid); return new ResponseObj(ApiResponseCode.OK, "업로드 성공하였습니다."); } diff --git a/src/main/java/com/kamco/cd/training/postgres/core/DatasetCoreService.java b/src/main/java/com/kamco/cd/training/postgres/core/DatasetCoreService.java index 964b1c3..6dc0d72 100644 --- a/src/main/java/com/kamco/cd/training/postgres/core/DatasetCoreService.java +++ b/src/main/java/com/kamco/cd/training/postgres/core/DatasetCoreService.java @@ -233,4 +233,13 @@ public class DatasetCoreService public void insertDatasetTestObj(DatasetObjRegDto objRegDto) { datasetObjRepository.insertDatasetTestObj(objRegDto); } + + public void updateDatasetUploadStatus(Long datasetUid) { + DatasetEntity entity = + datasetRepository + .findById(datasetUid) + .orElseThrow(() -> new NotFoundException("데이터셋을 찾을 수 없습니다. ID: " + datasetUid)); + + entity.setStatus(LearnDataRegister.COMPLETED.getId()); + } }