Merge pull request '학습실행 mount 경로 수정' (#60) from feat/training_260202 into develop

Reviewed-on: #60

src/main/java/com/kamco/cd/training/model/dto/ModelTrainMngDto.java

@@ -60,7 +60,7 @@ public class ModelTrainMngDto {
       }
     }
 
-    public String getStep2StatusNAme() {
+    public String getStep2StatusName() {
       if (this.step2Status == null || this.step2Status.isBlank()) return null;
       try {
         return TrainStatusType.valueOf(this.step2Status).getText(); // 또는 getName()

src/main/java/com/kamco/cd/training/model/service/TmpDatasetService.java

@@ -7,7 +7,6 @@ import lombok.RequiredArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.beans.factory.annotation.Value;
 import org.springframework.stereotype.Service;
-import org.springframework.transaction.annotation.Transactional;
 
 @Slf4j
 @Service
@@ -17,43 +16,73 @@ public class TmpDatasetService {
   @Value("${train.docker.requestDir}")
   private String requestDir;
 
-  @Transactional(readOnly = true)
   public Path buildTmpDatasetSymlink(String uid, List<String> datasetUids) throws IOException {
 
-    // 환경에 맞게 yml로 빼는 걸 추천
-    Path BASE = Paths.get(requestDir);
+    log.info("========== buildTmpDatasetSymlink START ==========");
+    log.info("uid={}", uid);
+    log.info("datasetUids={}", datasetUids);
+    log.info("requestDir(raw)={}", requestDir);
+
+    Path BASE = toPath(requestDir);
     Path tmp = BASE.resolve("tmp").resolve(uid);
 
-    // mkdir -p "$TMP"/train/{input1,input2,label} ...
+    log.info("BASE={}", BASE);
+    log.info("BASE exists? {}", Files.isDirectory(BASE));
+    log.info("tmp={}", tmp);
+
+    long noDir = 0, scannedDirs = 0, regularFiles = 0, symlinksMade = 0;
+
     for (String type : List.of("train", "val")) {
       for (String part : List.of("input1", "input2", "label")) {
-        Files.createDirectories(tmp.resolve(type).resolve(part));
+        Path dir = tmp.resolve(type).resolve(part);
+        Files.createDirectories(dir);
+        log.info("createDirectories: {}", dir);
       }
     }
 
     for (String id : datasetUids) {
       Path srcRoot = BASE.resolve(id);
+      log.info("---- dataset id={} srcRoot={} exists? {}", id, srcRoot, Files.isDirectory(srcRoot));
 
       for (String type : List.of("train", "val")) {
         for (String part : List.of("input1", "input2", "label")) {
 
           Path srcDir = srcRoot.resolve(type).resolve(part);
 
-          // zsh NULL_GLOB: 폴더가 없으면 그냥 continue
-          if (!Files.isDirectory(srcDir)) continue;
+          if (!Files.isDirectory(srcDir)) {
+            log.warn("SKIP (not directory): {}", srcDir);
+            noDir++;
+            continue;
+          }
+
+          scannedDirs++;
+          log.info("SCAN dir={}", srcDir);
 
           try (DirectoryStream<Path> stream = Files.newDirectoryStream(srcDir)) {
             for (Path f : stream) {
-              if (!Files.isRegularFile(f)) continue;
+              if (!Files.isRegularFile(f)) {
+                log.debug("skip non-regular file: {}", f);
+                continue;
+              }
+
+              regularFiles++;
 
               String dstName = id + "__" + f.getFileName();
               Path dst = tmp.resolve(type).resolve(part).resolve(dstName);
 
-              // 이미 있으면 스킵(원하면 덮어쓰기 로직으로 바꿀 수 있음)
-              if (Files.exists(dst)) continue;
+              try {
+                if (Files.exists(dst) || Files.isSymbolicLink(dst)) {
+                  Files.delete(dst);
+                  log.debug("deleted existing: {}", dst);
+                }
 
-              // ln -s "$f" "$dst" 와 동일
-              Files.createSymbolicLink(dst, f.toAbsolutePath());
+                Files.createSymbolicLink(dst, f.toAbsolutePath());
+                symlinksMade++;
+                log.debug("created symlink: {} -> {}", dst, f.toAbsolutePath());
+
+              } catch (Exception e) {
+                log.error("FAILED create symlink: {} -> {}", dst, f.toAbsolutePath(), e);
+              }
             }
           }
         }
@@ -61,6 +90,20 @@ public class TmpDatasetService {
     }
 
     log.info("tmp dataset created: {}", tmp);
+    log.info(
+        "summary: scannedDirs={}, noDir={}, regularFiles={}, symlinksMade={}",
+        scannedDirs,
+        noDir,
+        regularFiles,
+        symlinksMade);
+
     return tmp;
   }
+
+  private static Path toPath(String p) {
+    if (p.startsWith("~/")) {
+      return Paths.get(System.getProperty("user.home")).resolve(p.substring(2)).normalize();
+    }
+    return Paths.get(p).toAbsolutePath().normalize();
+  }
 }

src/main/java/com/kamco/cd/training/postgres/repository/model/ModelDatasetMappRepositoryImpl.java

@@ -16,10 +16,10 @@ public class ModelDatasetMappRepositoryImpl implements ModelDatasetMappRepositor
 
   @Override
   public List<ModelDatasetMappEntity> findByModelUid(Long modelId) {
-    queryFactory
+    return queryFactory
         .select(modelDatasetMappEntity)
         .from(modelDatasetMappEntity)
-        .where(modelDatasetMappEntity.modelUid.eq(modelId));
-    return List.of();
+        .where(modelDatasetMappEntity.modelUid.eq(modelId))
+        .fetch();
   }
 }

src/main/java/com/kamco/cd/training/postgres/repository/train/ModelTestMetricsJobRepositoryImpl.java

@@ -42,7 +42,10 @@ public class ModelTestMetricsJobRepositoryImpl extends QuerydslRepositorySupport
     return queryFactory
         .select(
             Projections.constructor(
-                ResponsePathDto.class, modelMasterEntity.id, modelMasterEntity.responsePath))
+                ResponsePathDto.class,
+                modelMasterEntity.id,
+                modelMasterEntity.responsePath,
+                modelMasterEntity.uuid))
         .from(modelMasterEntity)
         .where(
             modelMasterEntity.step2EndDttm.isNotNull(),

src/main/java/com/kamco/cd/training/postgres/repository/train/ModelTrainMetricsJobRepositoryImpl.java

@@ -29,7 +29,10 @@ public class ModelTrainMetricsJobRepositoryImpl extends QuerydslRepositorySuppor
     return queryFactory
         .select(
             Projections.constructor(
-                ResponsePathDto.class, modelMasterEntity.id, modelMasterEntity.responsePath))
+                ResponsePathDto.class,
+                modelMasterEntity.id,
+                modelMasterEntity.responsePath,
+                modelMasterEntity.uuid))
         .from(modelMasterEntity)
         .where(
             modelMasterEntity.step1EndDttm.isNotNull(),

src/main/java/com/kamco/cd/training/train/dto/ModelTrainMetricsDto.java

@@ -1,6 +1,7 @@
 package com.kamco.cd.training.train.dto;
 
 import io.swagger.v3.oas.annotations.media.Schema;
+import java.util.UUID;
 import lombok.AllArgsConstructor;
 import lombok.Getter;
 import lombok.NoArgsConstructor;
@@ -17,5 +18,6 @@ public class ModelTrainMetricsDto {
 
     private Long modelId;
     private String responsePath;
+    private UUID uuid;
   }
 }

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

@@ -169,7 +169,7 @@ public class DockerTrainService {
 
     // 컨테이너 이름 지정
     c.add("--name");
-    c.add(containerName + "-" + req.getUuid().substring(0, 8));
+    c.add(containerName);
 
     // 실행 종료 시 자동 삭제
     c.add("--rm");
@@ -206,7 +206,7 @@ public class DockerTrainService {
 
     // 요청/결과 디렉토리 볼륨 마운트
     c.add("-v");
-    c.add(requestDir + ":/data");
+    c.add(requestDir + "/tmp:/data");
     c.add("-v");
     c.add(responseDir + ":/checkpoints");
 
@@ -377,7 +377,7 @@ public class DockerTrainService {
     c.add("docker");
     c.add("run");
     c.add("--name");
-    c.add(containerName + "=" + req.getUuid().substring(0, 8));
+    c.add(containerName);
     c.add("--rm");
 
     c.add("--gpus");

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

@@ -27,6 +27,10 @@ public class ModelTestMetricsJobService {
   @Value("${spring.profiles.active}")
   private String profile;
 
+  // 학습 결과가 저장될 호스트 디렉토리
+  @Value("${train.docker.responseDir}")
+  private String responseDir;
+
   /**
    * 실행중인 profile
    *
@@ -51,7 +55,7 @@ public class ModelTestMetricsJobService {
 
     for (ResponsePathDto modelInfo : modelIds) {
 
-      String testPath = modelInfo.getResponsePath() + "/metrics/test.csv";
+      String testPath = responseDir + "/" + modelInfo.getUuid() + "/metrics/test.csv";
       try (BufferedReader reader =
           Files.newBufferedReader(Paths.get(testPath), StandardCharsets.UTF_8); ) {
 

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

@@ -55,7 +55,7 @@ public class ModelTrainMetricsJobService {
 
     for (ResponsePathDto modelInfo : modelIds) {
 
-      String trainPath = responseDir + "{uuid}/metrics/train.csv"; // TODO
+      String trainPath = responseDir + "/" + modelInfo.getUuid() + "/metrics/train.csv";
       try (BufferedReader reader =
           Files.newBufferedReader(Paths.get(trainPath), StandardCharsets.UTF_8); ) {
 
@@ -80,7 +80,7 @@ public class ModelTrainMetricsJobService {
         throw new RuntimeException(e);
       }
 
-      String validationPath = modelInfo.getResponsePath() + "/metrics/val.csv";
+      String validationPath = responseDir + "/" + modelInfo.getUuid() + "/metrics/val.csv";
       try (BufferedReader reader =
           Files.newBufferedReader(Paths.get(validationPath), StandardCharsets.UTF_8); ) {