모델학습 2단계 패키징 시작,종료일시,상태 로직 추가

src/main/java/com/kamco/cd/training/train/service/ModelTestMetricsJobService.java

@@ -2,6 +2,7 @@ package com.kamco.cd.training.train.service;
 
 import com.fasterxml.jackson.databind.ObjectMapper;
 import com.fasterxml.jackson.databind.SerializationFeature;
+import com.kamco.cd.training.common.enums.TrainStatusType;
 import com.kamco.cd.training.postgres.core.ModelTestMetricsJobCoreService;
 import com.kamco.cd.training.train.dto.ModelTrainMetricsDto.ModelMetricJsonDto;
 import com.kamco.cd.training.train.dto.ModelTrainMetricsDto.ModelTestFileName;
@@ -14,6 +15,7 @@ import java.nio.file.Files;
 import java.nio.file.Path;
 import java.nio.file.Paths;
 import java.nio.file.StandardOpenOption;
+import java.time.ZonedDateTime;
 import java.util.ArrayList;
 import java.util.List;
 import java.util.Set;
@@ -111,11 +113,18 @@ public class ModelTestMetricsJobService {
 
         modelTestMetricsJobCoreService.insertModelMetricsTest(batchArgs);
 
+        // test.csv 파일 읽어서 저장한 여부로만 사용하기
+        modelTestMetricsJobCoreService.updateModelMetricsTrainSaveYn(
+            modelInfo.getModelId(), "step2");
+
       } catch (IOException e) {
         throw new RuntimeException(e);
       }
 
       // 패키징할 파일 만들기
+      modelTestMetricsJobCoreService.updatePackingStart(
+          modelInfo.getModelId(), ZonedDateTime.now());
+
       ModelMetricJsonDto jsonDto =
           modelTestMetricsJobCoreService.getTestMetricPackingInfo(modelInfo.getModelId());
       try {
@@ -165,8 +174,16 @@ public class ModelTestMetricsJobService {
         throw new RuntimeException(e);
       }
 
-      zipFiles(files, zipPath);
-      modelTestMetricsJobCoreService.updateModelMetricsTrainSaveYn(modelInfo.getModelId(), "step2");
+      try {
+        zipFiles(files, zipPath);
+
+        modelTestMetricsJobCoreService.updatePackingEnd(
+            modelInfo.getModelId(), ZonedDateTime.now(), TrainStatusType.COMPLETED.getId());
+      } catch (IOException e) {
+        modelTestMetricsJobCoreService.updatePackingEnd(
+            modelInfo.getModelId(), ZonedDateTime.now(), TrainStatusType.ERROR.getId());
+        throw new RuntimeException(e);
+      }
     }
   }