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()); + } }