업로드 성공 후 COMPLETED 해주기

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, "업로드 성공하였습니다.");
   }
 

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