diff --git a/src/main/java/com/kamco/cd/kamcoback/inference/service/InferenceResultService.java b/src/main/java/com/kamco/cd/kamcoback/inference/service/InferenceResultService.java index 7e21b5dc..770adb70 100644 --- a/src/main/java/com/kamco/cd/kamcoback/inference/service/InferenceResultService.java +++ b/src/main/java/com/kamco/cd/kamcoback/inference/service/InferenceResultService.java @@ -26,6 +26,7 @@ import com.kamco.cd.kamcoback.model.dto.ModelMngDto.Basic; import com.kamco.cd.kamcoback.postgres.core.InferenceResultCoreService; import com.kamco.cd.kamcoback.postgres.core.MapSheetMngCoreService; import com.kamco.cd.kamcoback.postgres.core.ModelMngCoreService; +import jakarta.persistence.EntityNotFoundException; import jakarta.validation.constraints.NotNull; import java.time.ZonedDateTime; import java.util.ArrayList; @@ -100,6 +101,10 @@ public class InferenceResultService { req.setMapSheetNum(this.createdMngDto(req, targetList)); } + if (req.getMapSheetNum().isEmpty()) { + throw new EntityNotFoundException("분석 대상 정보가 부족합니다."); + } + // 추론 테이블 저장 UUID uuid = inferenceResultCoreService.saveInferenceInfo(req); this.startInference(req, uuid); diff --git a/src/main/java/com/kamco/cd/kamcoback/postgres/core/InferenceResultCoreService.java b/src/main/java/com/kamco/cd/kamcoback/postgres/core/InferenceResultCoreService.java index 4b017b63..7232e378 100644 --- a/src/main/java/com/kamco/cd/kamcoback/postgres/core/InferenceResultCoreService.java +++ b/src/main/java/com/kamco/cd/kamcoback/postgres/core/InferenceResultCoreService.java @@ -64,8 +64,12 @@ public class InferenceResultCoreService { * @param req */ public UUID saveInferenceInfo(InferenceResultDto.RegReq req) { + String mapSheetName = - req.getMapSheetNum().get(0).getMapSheetName() + " 외 " + req.getMapSheetNum().size() + "건"; + req.getMapSheetNum().get(0).getMapSheetName() + + " 외 " + + (req.getMapSheetNum().size() - 1) + + "건"; if (req.getMapSheetNum().size() == 1) { mapSheetName = diff --git a/src/main/java/com/kamco/cd/kamcoback/scheduler/service/MapSheetInferenceJobService.java b/src/main/java/com/kamco/cd/kamcoback/scheduler/service/MapSheetInferenceJobService.java index 56f4dfa9..c59e1703 100644 --- a/src/main/java/com/kamco/cd/kamcoback/scheduler/service/MapSheetInferenceJobService.java +++ b/src/main/java/com/kamco/cd/kamcoback/scheduler/service/MapSheetInferenceJobService.java @@ -102,7 +102,7 @@ public class MapSheetInferenceJobService { // 종료시간 this.updateProcessingEndTimeByModel(batchSheet.getUuid(), ZonedDateTime.now(), "M1"); } else if (type.equals("M2")) { - // M1 완료되었으면 M3 실행 + // M2 완료되었으면 M3 실행 this.startInference( batchSheet.getId(), batchSheet.getUuid(), "M3", batchSheet.getM3ModelUuid()); // 종료시간