패키징 zip파일 만들기 커밋

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

@@ -1,6 +1,8 @@
 package com.kamco.cd.training.postgres.core;
 
 import com.kamco.cd.training.postgres.repository.train.ModelTestMetricsJobRepository;
+import com.kamco.cd.training.train.dto.ModelTrainMetricsDto.ModelMetricJsonDto;
+import com.kamco.cd.training.train.dto.ModelTrainMetricsDto.ModelTestFileName;
 import com.kamco.cd.training.train.dto.ModelTrainMetricsDto.ResponsePathDto;
 import java.util.List;
 import lombok.RequiredArgsConstructor;
@@ -26,4 +28,12 @@ public class ModelTestMetricsJobCoreService {
   public void insertModelMetricsTest(List<Object[]> batchArgs) {
     modelTestMetricsJobRepository.insertModelMetricsTest(batchArgs);
   }
+
+  public ModelMetricJsonDto getTestMetricPackingInfo(Long modelId) {
+    return modelTestMetricsJobRepository.getTestMetricPackingInfo(modelId);
+  }
+
+  public ModelTestFileName findModelTestFileNames(Long modelId) {
+    return modelTestMetricsJobRepository.findModelTestFileNames(modelId);
+  }
 }

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

@@ -1,5 +1,7 @@
 package com.kamco.cd.training.postgres.repository.train;
 
+import com.kamco.cd.training.train.dto.ModelTrainMetricsDto.ModelMetricJsonDto;
+import com.kamco.cd.training.train.dto.ModelTrainMetricsDto.ModelTestFileName;
 import com.kamco.cd.training.train.dto.ModelTrainMetricsDto.ResponsePathDto;
 import java.util.List;
 
@@ -10,4 +12,8 @@ public interface ModelTestMetricsJobRepositoryCustom {
   List<ResponsePathDto> getTestMetricSaveNotYetModelIds();
 
   void insertModelMetricsTest(List<Object[]> batchArgs);
+
+  ModelMetricJsonDto getTestMetricPackingInfo(Long modelId);
+
+  ModelTestFileName findModelTestFileNames(Long modelId);
 }

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

@@ -1,9 +1,14 @@
 package com.kamco.cd.training.postgres.repository.train;
 
 import static com.kamco.cd.training.postgres.entity.QModelMasterEntity.modelMasterEntity;
+import static com.kamco.cd.training.postgres.entity.QModelMetricsTestEntity.modelMetricsTestEntity;
+import static com.kamco.cd.training.postgres.entity.QModelMetricsTrainEntity.modelMetricsTrainEntity;
 
 import com.kamco.cd.training.common.enums.TrainStatusType;
 import com.kamco.cd.training.postgres.entity.ModelMetricsTestEntity;
+import com.kamco.cd.training.train.dto.ModelTrainMetricsDto.ModelMetricJsonDto;
+import com.kamco.cd.training.train.dto.ModelTrainMetricsDto.ModelTestFileName;
+import com.kamco.cd.training.train.dto.ModelTrainMetricsDto.Properties;
 import com.kamco.cd.training.train.dto.ModelTrainMetricsDto.ResponsePathDto;
 import com.querydsl.core.types.Projections;
 import com.querydsl.jpa.impl.JPAQueryFactory;
@@ -70,4 +75,43 @@ public class ModelTestMetricsJobRepositoryImpl extends QuerydslRepositorySupport
 
     jdbcTemplate.batchUpdate(sql, batchArgs);
   }
+
+  @Override
+  public ModelMetricJsonDto getTestMetricPackingInfo(Long modelId) {
+    return queryFactory
+        .select(
+            Projections.constructor(
+                ModelMetricJsonDto.class,
+                modelMasterEntity.modelNo,
+                modelMasterEntity.modelVer,
+                Projections.constructor(
+                    Properties.class,
+                    modelMetricsTestEntity.f1Score,
+                    modelMetricsTestEntity.precisions,
+                    modelMetricsTestEntity.recall,
+                    modelMetricsTestEntity.iou,
+                    modelMetricsTrainEntity.loss)))
+        .from(modelMetricsTestEntity)
+        .innerJoin(modelMasterEntity)
+        .on(modelMetricsTestEntity.model.id.eq(modelMasterEntity.id))
+        .innerJoin(modelMetricsTrainEntity)
+        .on(
+            modelMetricsTestEntity.model.eq(modelMetricsTrainEntity.model),
+            modelMasterEntity.bestEpoch.eq(modelMetricsTrainEntity.epoch))
+        .where(modelMetricsTestEntity.model.id.eq(modelId))
+        .fetchOne();
+  }
+
+  @Override
+  public ModelTestFileName findModelTestFileNames(Long modelId) {
+    return queryFactory
+        .select(
+            Projections.constructor(
+                ModelTestFileName.class, modelMetricsTestEntity.model1, modelMasterEntity.modelVer))
+        .from(modelMetricsTestEntity)
+        .innerJoin(modelMasterEntity)
+        .on(modelMetricsTestEntity.model.id.eq(modelMasterEntity.id))
+        .where(modelMetricsTestEntity.model.id.eq(modelId))
+        .fetchOne();
+  }
 }