From 8e4bea53daf6e100afaea4d0408c9e9b1e5436b6 Mon Sep 17 00:00:00 2001
From: teddy <teddy.kim.kim@tf.dabeeo.com>
Date: Thu, 12 Feb 2026 15:43:52 +0900
Subject: [PATCH] =?UTF-8?q?=EB=AA=A8=EB=8D=B8=EB=93=B1=EB=A1=9D=20?=
 =?UTF-8?q?=EC=88=98=EC=A0=95?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../training/model/ModelTrainMngApiController.java   |  5 ++---
 .../training/model/service/ModelTrainMngService.java | 12 ++++++------
 .../postgres/core/ModelTrainMngCoreService.java      |  9 +++++++--
 3 files changed, 15 insertions(+), 11 deletions(-)

diff --git a/src/main/java/com/kamco/cd/training/model/ModelTrainMngApiController.java b/src/main/java/com/kamco/cd/training/model/ModelTrainMngApiController.java
index fce7428..6251ae0 100644
--- a/src/main/java/com/kamco/cd/training/model/ModelTrainMngApiController.java
+++ b/src/main/java/com/kamco/cd/training/model/ModelTrainMngApiController.java
@@ -92,9 +92,8 @@ public class ModelTrainMngApiController {
         @ApiResponse(responseCode = "500", description = "서버 오류", content = @Content)
       })
   @PostMapping
-  public ApiResponseDto<String> createModelTrain(@Valid @RequestBody ModelTrainMngDto.AddReq req) {
-    modelTrainMngService.createModelTrain(req);
-    return ApiResponseDto.ok("ok");
+  public ApiResponseDto<UUID> createModelTrain(@Valid @RequestBody ModelTrainMngDto.AddReq req) {
+    return ApiResponseDto.ok(modelTrainMngService.createModelTrain(req));
   }
 
   @Operation(summary = "모델학습 config 정보 조회", description = "모델학습 config 정보 조회 API")
diff --git a/src/main/java/com/kamco/cd/training/model/service/ModelTrainMngService.java b/src/main/java/com/kamco/cd/training/model/service/ModelTrainMngService.java
index 8b4c832..3a82566 100644
--- a/src/main/java/com/kamco/cd/training/model/service/ModelTrainMngService.java
+++ b/src/main/java/com/kamco/cd/training/model/service/ModelTrainMngService.java
@@ -62,7 +62,7 @@ public class ModelTrainMngService {
    * @return
    */
   @Transactional
-  public void createModelTrain(ModelTrainMngDto.AddReq req) {
+  public UUID createModelTrain(ModelTrainMngDto.AddReq req) {
     HyperParam hyperParam = req.getHyperParam();
     HyperParamDto.Basic hyper = new HyperParamDto.Basic();
 
@@ -81,7 +81,10 @@ public class ModelTrainMngService {
     }
 
     // 모델학습 테이블 저장
-    Long modelId = modelTrainMngCoreService.saveModel(req);
+    ModelTrainMngDto.Basic modelDto = modelTrainMngCoreService.saveModel(req);
+
+    Long modelId = modelDto.getId();
+    UUID modelUuid = modelDto.getUuid();
 
     // 모델학습 데이터셋 저장
     modelTrainMngCoreService.saveModelDataset(modelId, req);
@@ -109,10 +112,7 @@ public class ModelTrainMngService {
       throw new RuntimeException(e);
     }
 
-    // 저장 다 끝난 뒤에 job enqueue
-    if (Boolean.TRUE.equals(req.getIsStart())) {
-      trainJobService.enqueue(modelId); // job 저장 + 이벤트 발행(실행은 AFTER_COMMIT)
-    }
+    return modelUuid;
   }
 
   /**
diff --git a/src/main/java/com/kamco/cd/training/postgres/core/ModelTrainMngCoreService.java b/src/main/java/com/kamco/cd/training/postgres/core/ModelTrainMngCoreService.java
index 692f475..f28c1ef 100644
--- a/src/main/java/com/kamco/cd/training/postgres/core/ModelTrainMngCoreService.java
+++ b/src/main/java/com/kamco/cd/training/postgres/core/ModelTrainMngCoreService.java
@@ -78,7 +78,7 @@ public class ModelTrainMngCoreService {
    * @param addReq
    * @return
    */
-  public Long saveModel(ModelTrainMngDto.AddReq addReq) {
+  public ModelTrainMngDto.Basic saveModel(ModelTrainMngDto.AddReq addReq) {
     ModelMasterEntity entity = new ModelMasterEntity();
     ModelHyperParamEntity hyperParamEntity = new ModelHyperParamEntity();
 
@@ -117,7 +117,12 @@ public class ModelTrainMngCoreService {
 
     entity.setCreatedUid(userUtil.getId());
     ModelMasterEntity resultEntity = modelMngRepository.save(entity);
-    return resultEntity.getId();
+
+    ModelTrainMngDto.Basic result = new ModelTrainMngDto.Basic();
+    result.setId(resultEntity.getId());
+    result.setUuid(resultEntity.getUuid());
+
+    return result;
   }
 
   /**