중단일 때 stop 추가

2026-04-09 17:53:17 +09:00
parent edff3b0ef8
commit afb3f57ef6
1 changed files with 5 additions and 0 deletions
--- a/src/main/java/com/kamco/cd/training/postgres/core/ModelTrainMngCoreService.java
+++ b/src/main/java/com/kamco/cd/training/postgres/core/ModelTrainMngCoreService.java
@@ -419,6 +419,11 @@ public class ModelTrainMngCoreService {
            .orElseThrow(() -> new IllegalArgumentException("Model not found: " + modelId));

    master.setStatusCd(TrainStatusType.STOPPED.getId());
+    if (master.getStep2StrtDttm() != null) {
+      master.setStep2State(TrainStatusType.STOPPED.getId());
+    } else {
+      master.setStep1State(TrainStatusType.STOPPED.getId());
+    }
  }

  /** 완료 처리(옵션) - Worker가 성공 시 호출 */