From f1f8753fbbb4af3eb7201fd056c63aee894e176a Mon Sep 17 00:00:00 2001 From: teddy Date: Mon, 26 Jan 2026 15:26:09 +0900 Subject: [PATCH] =?UTF-8?q?batchId=20list=20=EC=83=9D=EC=84=B1=20=EC=88=98?= =?UTF-8?q?=EC=A0=95?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../core/InferenceResultCoreService.java | 17 +++++++++-------- 1 file changed, 9 insertions(+), 8 deletions(-) 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 edcdaf2c..9dda7e4c 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 @@ -38,8 +38,10 @@ import jakarta.validation.constraints.NotNull; import java.time.ZonedDateTime; import java.util.ArrayList; import java.util.List; +import java.util.Objects; import java.util.UUID; import java.util.function.Consumer; +import java.util.stream.Stream; import lombok.RequiredArgsConstructor; import lombok.extern.log4j.Log4j2; import org.springframework.dao.DataAccessException; @@ -263,10 +265,12 @@ public class InferenceResultCoreService { applyModelUpdate(entity, request); } - List batchIds = new ArrayList<>(); - batchIds.add(entity.getM1ModelBatchId()); - batchIds.add(entity.getM2ModelBatchId()); - batchIds.add(entity.getM3ModelBatchId()); + List batchIds = + Stream.of( + entity.getM1ModelBatchId(), entity.getM2ModelBatchId(), entity.getM3ModelBatchId()) + .filter(Objects::nonNull) + .distinct() // 중복 방지 (선택) + .toList(); // testing 추론결과 테이블 조회하여 탐지 개수 업데이트 Long testing = getInferenceResultCnt(batchIds); @@ -509,10 +513,7 @@ public class InferenceResultCoreService { if (entity == null) { throw new CustomApiException("NOT_FOUND_DATA", HttpStatus.NOT_FOUND); } - // inferenceResultRepository - // .getInferenceUid(uuid) - // .orElseThrow(() -> new CustomApiException("NOT_FOUND_DATA", - // HttpStatus.NOT_FOUND)); + InferenceLearnDto dto = new InferenceLearnDto(); dto.setUid(entity.getUid()); dto.setM1ModelBatchId(entity.getM1ModelBatchId());