diff --git a/src/main/java/com/kamco/cd/kamcoback/postgres/repository/mapsheet/MapSheetMngRepositoryImpl.java b/src/main/java/com/kamco/cd/kamcoback/postgres/repository/mapsheet/MapSheetMngRepositoryImpl.java index 74873c8b..3405e344 100644 --- a/src/main/java/com/kamco/cd/kamcoback/postgres/repository/mapsheet/MapSheetMngRepositoryImpl.java +++ b/src/main/java/com/kamco/cd/kamcoback/postgres/repository/mapsheet/MapSheetMngRepositoryImpl.java @@ -504,7 +504,7 @@ public class MapSheetMngRepositoryImpl extends QuerydslRepositorySupport whereBuilder.and(mapSheetMngHstEntity.mngYyyy.eq(mngYyyy)); whereBuilder.and(mapSheetMngHstEntity.mapSheetNum.in(mapIds)); whereBuilder.and(mapSheetMngHstEntity.dataState.eq("DONE")); - whereBuilder.and(mapSheetMngHstEntity.syncState.eq("DONE")); + whereBuilder.and(mapSheetMngHstEntity.syncState.eq("DONE")); // TODO 싱크체크 or조건 추가 whereBuilder.and(mapSheetMngHstEntity.useInference.eq("USE")); return queryFactory 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 c59e1703..824ab33c 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 @@ -113,10 +113,18 @@ public class MapSheetInferenceJobService { saveInferenceAiDto.setUuid(batchSheet.getUuid()); saveInferenceAiDto.setStatus(Status.END.getId()); saveInferenceAiDto.setInferEndDttm(ZonedDateTime.now()); + saveInferenceAiDto.setType(type); inferenceResultCoreService.update(saveInferenceAiDto); // 종료시간 this.updateProcessingEndTimeByModel(batchSheet.getUuid(), ZonedDateTime.now(), "M3"); } + } else if ("COMPLETED_WITH_FAILURES".equals(dto.getStatus())) { + SaveInferenceAiDto saveInferenceAiDto = new SaveInferenceAiDto(); + saveInferenceAiDto.setUuid(batchSheet.getUuid()); + saveInferenceAiDto.setStatus(Status.END.getId()); + saveInferenceAiDto.setInferEndDttm(ZonedDateTime.now()); + saveInferenceAiDto.setType(batchSheet.getRunningModelType()); + inferenceResultCoreService.update(saveInferenceAiDto); } else { SaveInferenceAiDto saveInferenceAiDto = new SaveInferenceAiDto(); saveInferenceAiDto.setUuid(batchSheet.getUuid());