diff --git a/src/main/java/com/kamco/cd/training/postgres/core/ModelTrainMngCoreService.java b/src/main/java/com/kamco/cd/training/postgres/core/ModelTrainMngCoreService.java index 2df7c31..0508206 100644 --- a/src/main/java/com/kamco/cd/training/postgres/core/ModelTrainMngCoreService.java +++ b/src/main/java/com/kamco/cd/training/postgres/core/ModelTrainMngCoreService.java @@ -120,6 +120,15 @@ public class ModelTrainMngCoreService { entity.setTrainType(addReq.getTrainType()); // 일반, 전이 entity.setBeforeModelId(addReq.getBeforeModelId()); + // 전이학습일 때 beforeModelUuid 추가하기 + if (addReq.getBeforeModelId() != null) { + ModelMasterEntity beforeModel = + modelMngRepository + .findById(addReq.getBeforeModelId()) + .orElseThrow(() -> new CustomApiException("NOT_FOUND_DATA", HttpStatus.NOT_FOUND)); + entity.setBeforeModelUuid(beforeModel.getUuid()); + } + entity.setStatusCd(TrainStatusType.READY.getId()); entity.setStep1State(TrainStatusType.READY.getId());