containerName 생성 변경

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

@@ -47,7 +47,8 @@ public class ModelTrainJobCoreService {
 
   /** 실행 시작 처리 */
   @Transactional
-  public void markRunning(Long jobId, String containerName, String logPath, String lockedBy) {
+  public void markRunning(
+      Long jobId, String containerName, String logPath, String lockedBy, Integer totalEpoch) {
     ModelTrainJobEntity job =
         modelTrainJobRepository
             .findById(jobId)
@@ -59,6 +60,10 @@ public class ModelTrainJobCoreService {
     job.setStartedDttm(ZonedDateTime.now());
     job.setLockedDttm(ZonedDateTime.now());
     job.setLockedBy(lockedBy);
+
+    if (totalEpoch != null) {
+      job.setTotalEpoch(totalEpoch);
+    }
   }
 
   /** 성공 처리 */

src/main/java/com/kamco/cd/training/postgres/entity/ModelTrainJobEntity.java

@@ -78,6 +78,12 @@ public class ModelTrainJobEntity {
   @Column(name = "locked_by", length = 100)
   private String lockedBy;
 
+  @Column(name = "total_epoch")
+  private Integer totalEpoch;
+
+  @Column(name = "current_epoch")
+  private Integer currentEpoch;
+
   public ModelTrainJobDto toDto() {
     return new ModelTrainJobDto(
         this.id,
@@ -90,6 +96,8 @@ public class ModelTrainJobEntity {
         this.paramsJson,
         this.queuedDttm,
         this.startedDttm,
-        this.finishedDttm);
+        this.finishedDttm,
+        this.totalEpoch,
+        this.currentEpoch);
   }
 }