Merge branch 'develop' of https://kamco.git.gs.dabeeo.com/MVPTeam/kamco-train-api into develop

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

@@ -5,6 +5,7 @@ import com.kamco.cd.training.postgres.repository.train.ModelTrainJobRepository;
 import com.kamco.cd.training.train.dto.ModelTrainJobDto;
 import java.time.ZonedDateTime;
 import java.util.Map;
+import java.util.Objects;
 import java.util.Optional;
 import lombok.RequiredArgsConstructor;
 import org.springframework.stereotype.Service;
@@ -104,4 +105,16 @@ public class ModelTrainJobCoreService {
     job.setStatusCd("STOPPED");
     job.setFinishedDttm(ZonedDateTime.now());
   }
+
+  @Transactional
+  public void updateEpoch(String containerName, Integer epoch) {
+    ModelTrainJobEntity job =
+        modelTrainJobRepository
+            .findByContainerName(containerName)
+            .orElseThrow(() -> new IllegalArgumentException("Job not found: " + containerName));
+
+    job.setCurrentEpoch(epoch);
+
+    if (Objects.equals(job.getTotalEpoch(), epoch)) {}
+  }
 }

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

@@ -7,4 +7,6 @@ public interface ModelTrainJobRepositoryCustom {
   int findMaxAttemptNo(Long modelId);
 
   Optional<ModelTrainJobEntity> findLatestByModelId(Long modelId);
+
+  Optional<ModelTrainJobEntity> findByContainerName(String containerName);
 }

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

@@ -40,4 +40,18 @@ public class ModelTrainJobRepositoryImpl implements ModelTrainJobRepositoryCusto
 
     return Optional.ofNullable(job);
   }
+
+  @Override
+  public Optional<ModelTrainJobEntity> findByContainerName(String containerName) {
+    QModelTrainJobEntity j = QModelTrainJobEntity.modelTrainJobEntity;
+
+    ModelTrainJobEntity job =
+        queryFactory
+            .selectFrom(j)
+            .where(j.containerName.eq(containerName))
+            .orderBy(j.id.desc())
+            .fetchFirst();
+
+    return Optional.ofNullable(job);
+  }
 }