트랜젝션처리 임시폴더 uid업데이트

src/main/java/com/kamco/cd/training/train/service/DataSetCountersService.java

@@ -25,26 +25,32 @@ public class DataSetCountersService {
   @Value("${train.docker.basePath}")
   private String trainBaseDir;
 
-  public void getCount(Long modelId) {
+  public String getCount(Long modelId) {
     ModelTrainMngDto.Basic basic = modelTrainMngCoreService.findModelById(modelId);
     List<Long> datasetIds = modelTrainMngCoreService.findModelDatasetMapp(modelId);
     List<String> uids = modelTrainMngCoreService.findDatasetUid(datasetIds);
 
+    StringBuilder allLogs = new StringBuilder();
+
     try {
       // request 폴더
       for (String uid : uids) {
         Path path = Path.of(requestDir, uid);
         DatasetCounters counters = countTmpAfterBuild(path);
-        counters.prints(uid, "REQUEST");
+        allLogs.append(counters.prints(uid, "REQUEST")).append(System.lineSeparator());
       }
 
       // tmp
       Path tmpPath = Path.of(trainBaseDir, "tmp", basic.getRequestPath());
       DatasetCounters counters2 = countTmpAfterBuild(tmpPath);
-      counters2.prints(basic.getRequestPath(), "TMP");
+      allLogs
+          .append(counters2.prints(basic.getRequestPath(), "TMP"))
+          .append(System.lineSeparator());
     } catch (IOException e) {
       log.error(e.getMessage());
     }
+
+    return allLogs.toString();
   }
 
   private int countTif(Path dir) throws IOException {
@@ -114,7 +120,7 @@ public class DataSetCountersService {
       this.input2 = input2;
     }
 
-    public void prints(String uuid, String type) {
+    public String prints(String uuid, String type) {
       int train = 0, test = 0, val = 0;
       int train2 = 0, test2 = 0, val2 = 0;
 
@@ -138,6 +144,23 @@ public class DataSetCountersService {
           train + train2,
           val + val2,
           test + test2);
+
+      return String.format(
+          "======== UUID FOLDER COUNT %s : %s%n"
+              + "input 1 = train : %s | val : %s | test : %s%n"
+              + "input 2 = train : %s | val : %s | test : %s%n"
+              + "*total* = train : %s | val : %s | test : %s",
+          type,
+          uuid,
+          train,
+          val,
+          test,
+          train2,
+          val2,
+          test2,
+          train + train2,
+          val + val2,
+          test + test2);
     }
   }
 }

src/main/java/com/kamco/cd/training/train/service/TrainJobService.java

@@ -223,9 +223,9 @@ public class TrainJobService {
     UUID tmpUuid = UUID.randomUUID();
     String raw = tmpUuid.toString().toUpperCase().replace("-", "");
 
+    // MODELID 가져오기
     Long modelId = modelTrainMngCoreService.findModelIdByUuid(modelUuid);
     List<Long> datasetIds = modelTrainMngCoreService.findModelDatasetMapp(modelId);
-
     List<String> uids = modelTrainMngCoreService.findDatasetUid(datasetIds);
 
     try {
@@ -233,6 +233,7 @@ public class TrainJobService {
       String pathUid = tmpDatasetService.buildTmpDatasetSymlink(raw, uids);
       ModelTrainMngDto.UpdateReq updateReq = new ModelTrainMngDto.UpdateReq();
       updateReq.setRequestPath(pathUid);
+      // 학습모델을 수정한다.
       modelTrainMngCoreService.updateModelMaster(modelId, updateReq);
     } catch (IOException e) {
       throw new RuntimeException(e);