업로드 성공 후 COMPLETED 해주기

2026-02-10 18:28:42 +09:00
parent 025b573859
commit 6352f74b08
2 changed files with 11 additions and 0 deletions
--- 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());
+  }
 }