리커버리 추가

src/main/java/com/kamco/cd/training/postgres/core/ModelTrainJobCoreService.java

@@ -5,6 +5,8 @@ import com.kamco.cd.training.postgres.entity.ModelTrainJobEntity;
 import com.kamco.cd.training.postgres.repository.train.ModelTrainJobRepository;
 import com.kamco.cd.training.train.dto.ModelTrainJobDto;
 import java.time.ZonedDateTime;
+import java.util.Collections;
+import java.util.List;
 import java.util.Map;
 import java.util.Objects;
 import java.util.Optional;
@@ -118,6 +120,28 @@ public class ModelTrainJobCoreService {
     log.info("[TRAIN JOB FAIL] jobId={}, modelId={}", jobId, errorMessage);
   }
 
+  /**
+   * 중단됨 처리
+   *
+   * @param jobId
+   * @param exitCode
+   * @param errorMessage
+   */
+  @Transactional
+  public void markPaused(Long jobId, Integer exitCode, String errorMessage) {
+    ModelTrainJobEntity job =
+        modelTrainJobRepository
+            .findById(jobId)
+            .orElseThrow(() -> new CustomApiException("NOT_FOUND_DATA", HttpStatus.NOT_FOUND));
+
+    job.setStatusCd("STOPPED");
+    job.setExitCode(exitCode);
+    job.setErrorMessage(errorMessage);
+    job.setFinishedDttm(ZonedDateTime.now());
+
+    log.info("[TRAIN JOB FAIL] jobId={}, modelId={}", jobId, errorMessage);
+  }
+
   /** 취소 처리 */
   @Transactional
   public void markCanceled(Long jobId) {
@@ -151,11 +175,13 @@ public class ModelTrainJobCoreService {
    *
    * @return
    */
-  public ModelTrainJobDto findRunningJobs() {
-    ModelTrainJobEntity entity = modelTrainJobRepository.findRunningJobs().orElse(null);
-    if (entity == null) {
-      return null;
+  public List<ModelTrainJobDto> findRunningJobs() {
+    List<ModelTrainJobEntity> entity = modelTrainJobRepository.findRunningJobs();
+
+    if (entity == null || entity.isEmpty()) {
+      return Collections.emptyList();
     }
-    return entity.toDto();
+
+    return entity.stream().map(ModelTrainJobEntity::toDto).toList();
   }
 }

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

@@ -1,6 +1,7 @@
 package com.kamco.cd.training.postgres.repository.train;
 
 import com.kamco.cd.training.postgres.entity.ModelTrainJobEntity;
+import java.util.List;
 import java.util.Optional;
 
 public interface ModelTrainJobRepositoryCustom {
@@ -12,5 +13,5 @@ public interface ModelTrainJobRepositoryCustom {
 
   void insertModelTestTrainingRun(Long modelId, Long jobId, int epoch);
 
-  Optional<ModelTrainJobEntity> findRunningJobs();
+  List<ModelTrainJobEntity> findRunningJobs();
 }

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

@@ -9,6 +9,7 @@ import com.kamco.cd.training.postgres.entity.ModelTrainJobEntity;
 import com.kamco.cd.training.postgres.entity.QModelTrainJobEntity;
 import com.querydsl.jpa.impl.JPAQueryFactory;
 import jakarta.persistence.EntityManager;
+import java.util.List;
 import java.util.Optional;
 import org.springframework.stereotype.Repository;
 
@@ -83,18 +84,16 @@ public class ModelTrainJobRepositoryImpl implements ModelTrainJobRepositoryCusto
   }
 
   @Override
-  public Optional<ModelTrainJobEntity> findRunningJobs() {
-    return Optional.ofNullable(
-        queryFactory
-            .select(modelTrainJobEntity)
-            .from(modelTrainJobEntity)
-            .where(
-                modelTrainJobEntity
-                    .statusCd
-                    .eq(JobStatusType.RUNNING.getId())
-                    .and(modelTrainJobEntity.jobType.eq(JobType.TRAIN.getId())))
-            .orderBy(modelTrainJobEntity.id.desc())
-            .limit(1)
-            .fetchOne());
+  public List<ModelTrainJobEntity> findRunningJobs() {
+    return queryFactory
+        .select(modelTrainJobEntity)
+        .from(modelTrainJobEntity)
+        .where(
+            modelTrainJobEntity
+                .statusCd
+                .eq(JobStatusType.RUNNING.getId())
+                .and(modelTrainJobEntity.jobType.eq(JobType.TRAIN.getId())))
+        .orderBy(modelTrainJobEntity.id.desc())
+        .fetch();
   }
 }