From 8c19c996f7280156f9df27d51f598c44962e913e Mon Sep 17 00:00:00 2001
From: teddy <teddy.kim.kim@tf.dabeeo.com>
Date: Fri, 13 Feb 2026 08:33:36 +0900
Subject: [PATCH] =?UTF-8?q?tmp=20=ED=8C=8C=EC=9D=BC=20=EB=A7=81=ED=81=AC?=
 =?UTF-8?q?=20=EC=88=98=EC=A0=95?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../cd/training/model/service/TmpDatasetService.java  |  7 +++++--
 src/main/resources/application-dev.yml                | 11 ++++++-----
 src/main/resources/application-prod.yml               | 11 ++++++-----
 3 files changed, 17 insertions(+), 12 deletions(-)

diff --git a/src/main/java/com/kamco/cd/training/model/service/TmpDatasetService.java b/src/main/java/com/kamco/cd/training/model/service/TmpDatasetService.java
index 1edfe52..428ac23 100644
--- a/src/main/java/com/kamco/cd/training/model/service/TmpDatasetService.java
+++ b/src/main/java/com/kamco/cd/training/model/service/TmpDatasetService.java
@@ -16,6 +16,9 @@ public class TmpDatasetService {
   @Value("${train.docker.requestDir}")
   private String requestDir;
 
+  @Value("${train.docker.basePath}")
+  private String trainBaseDir;
+
   public String buildTmpDatasetSymlink(String uid, List<String> datasetUids) throws IOException {
 
     log.info("========== buildTmpDatasetHardlink START ==========");
@@ -24,7 +27,7 @@ public class TmpDatasetService {
     log.info("requestDir(raw)={}", requestDir);
 
     Path BASE = toPath(requestDir);
-    Path tmp = Path.of("/home/kcomu/data", "tmp", uid);
+    Path tmp = Path.of(trainBaseDir, "tmp", uid);
 
     log.info("BASE={}", BASE);
     log.info("BASE exists? {}", Files.isDirectory(BASE));
@@ -33,7 +36,7 @@ public class TmpDatasetService {
     long noDir = 0, scannedDirs = 0, regularFiles = 0, hardlinksMade = 0;
 
     // tmp 디렉토리 준비
-    for (String type : List.of("train", "val")) {
+    for (String type : List.of("train", "val", "test")) {
       for (String part : List.of("input1", "input2", "label")) {
         Path dir = tmp.resolve(type).resolve(part);
         Files.createDirectories(dir);
diff --git a/src/main/resources/application-dev.yml b/src/main/resources/application-dev.yml
index d418625..f678018 100644
--- a/src/main/resources/application-dev.yml
+++ b/src/main/resources/application-dev.yml
@@ -63,9 +63,10 @@ file:
 
 train:
   docker:
-    image: "kamco-cd-train:latest"
-    requestDir: "/home/kcomu/data/request"
-    responseDir: "/home/kcomu/data/response"
-    containerPrefix: "kamco-cd-train"
-    shmSize: "16g"
+    image: kamco-cd-train:latest
+    requestDir: /home/kcomu/data/request
+    responseDir: /home/kcomu/data/response
+    basePath: /home/kcomu/data
+    containerPrefix: kamco-cd-train
+    shmSize: 16g
     ipcHost: true
diff --git a/src/main/resources/application-prod.yml b/src/main/resources/application-prod.yml
index 9d8b40d..754c90f 100644
--- a/src/main/resources/application-prod.yml
+++ b/src/main/resources/application-prod.yml
@@ -62,10 +62,11 @@ file:
 
 train:
   docker:
-    image: "kamco-cd-train:latest"
-    requestDir: "/home/kcomu/data/request"
-    responseDir: "/home/kcomu/data/response"
-    containerPrefix: "kamco-cd-train"
-    shmSize: "16g"
+    image: kamco-cd-train:latest
+    requestDir: /home/kcomu/data/request
+    responseDir: /home/kcomu/data/response
+    basePath: /home/kcomu/data
+    containerPrefix: kamco-cd-train
+    shmSize: 16g
     ipcHost: true