MVPTeam/test
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[KC-103] 추론 실행 배치 수정

Browse Source
This commit is contained in:
teddy
2026-01-13 15:12:26 +09:00
parent a983223562
commit 65bac54db5
2 changed files with 55 additions and 49 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
src/main/java/com/kamco/cd/kamcoback/postgres/core/InferenceResultCoreService.java
View File

@@ -233,7 +233,9 @@ public class InferenceResultCoreService {
.getInferenceResultByUuid(request.getUuid()) .getInferenceResultByUuid(request.getUuid())
.orElseThrow(() -> new EntityNotFoundException()); .orElseThrow(() -> new EntityNotFoundException());
if (request.getType().equals("M1")) { if (request.getType() != null) {
switch (request.getType()) {
case "M1" -> {
if (request.getBatchId() != null) { if (request.getBatchId() != null) {
entity.setM1ModelBatchId(request.getBatchId()); entity.setM1ModelBatchId(request.getBatchId());
} }
@@ -245,8 +247,8 @@ public class InferenceResultCoreService {
if (request.getModelEndDttm() != null) { if (request.getModelEndDttm() != null) {
entity.setM1ModelEndDttm(request.getModelEndDttm()); entity.setM1ModelEndDttm(request.getModelEndDttm());
} }
}
} else if (request.getType().equals("M2")) { case "M2" -> {
if (request.getBatchId() != null) { if (request.getBatchId() != null) {
entity.setM2ModelBatchId(request.getBatchId()); entity.setM2ModelBatchId(request.getBatchId());
} }
@@ -258,8 +260,8 @@ public class InferenceResultCoreService {
if (request.getModelEndDttm() != null) { if (request.getModelEndDttm() != null) {
entity.setM2ModelEndDttm(request.getModelEndDttm()); entity.setM2ModelEndDttm(request.getModelEndDttm());
} }
}
} else if (request.getType().equals("M3")) { case "M3" -> {
if (request.getBatchId() != null) { if (request.getBatchId() != null) {
entity.setM3ModelBatchId(request.getBatchId()); entity.setM3ModelBatchId(request.getBatchId());
} }
@@ -272,6 +274,12 @@ public class InferenceResultCoreService {
entity.setM3ModelEndDttm(request.getModelEndDttm()); entity.setM3ModelEndDttm(request.getModelEndDttm());
} }
} }
}
}
if (request.getRunningModelType() != null) {
entity.setRunningModelType(request.getRunningModelType());
}
if (request.getInferStartDttm() != null) { if (request.getInferStartDttm() != null) {
entity.setInferStartDttm(request.getInferStartDttm()); entity.setInferStartDttm(request.getInferStartDttm());
@@ -293,10 +301,6 @@ public class InferenceResultCoreService {
entity.setDetectEndCnt(request.getDetectEndCnt()); entity.setDetectEndCnt(request.getDetectEndCnt());
} }
if (request.getType() != null) {
entity.setRunningModelType(request.getType());
}
if (request.getStatus() != null) { if (request.getStatus() != null) {
entity.setStatus(request.getStatus()); entity.setStatus(request.getStatus());
} }

12
src/main/java/com/kamco/cd/kamcoback/scheduler/service/MapSheetInferenceJobService.java
View File

@@ -48,7 +48,9 @@ public class MapSheetInferenceJobService {
@Value("${inference.url}") @Value("${inference.url}")
private String inferenceUrl; private String inferenceUrl;
/** 추론 진행 배치 1분 */ /**
* 추론 진행 배치 1분
*/
@Scheduled(fixedDelay = 60_000) @Scheduled(fixedDelay = 60_000)
@Transactional @Transactional
public void runBatch() { public void runBatch() {
@@ -134,7 +136,6 @@ public class MapSheetInferenceJobService {
saveInferenceAiDto.setUuid(batchSheet.getUuid()); saveInferenceAiDto.setUuid(batchSheet.getUuid());
saveInferenceAiDto.setStatus(Status.IN_PROGRESS.getId()); saveInferenceAiDto.setStatus(Status.IN_PROGRESS.getId());
saveInferenceAiDto.setDetectEndCnt((long) (completedJobs + failedJobs)); saveInferenceAiDto.setDetectEndCnt((long) (completedJobs + failedJobs));
saveInferenceAiDto.setType(batchSheet.getRunningModelType());
inferenceResultCoreService.update(saveInferenceAiDto); inferenceResultCoreService.update(saveInferenceAiDto);
} }
@@ -176,7 +177,7 @@ public class MapSheetInferenceJobService {
m.setCls_model_path( m.setCls_model_path(
progressDto.getCdModelClsPath() + "/" + progressDto.getCdModelClsFileName()); progressDto.getCdModelClsPath() + "/" + progressDto.getCdModelClsFileName());
m.setCls_model_version(progressDto.getClsModelVersion()); m.setCls_model_version(progressDto.getClsModelVersion());
m.setCd_model_type(type); m.setCd_model_type(inferenceType);
m.setPriority(progressDto.getPriority()); m.setPriority(progressDto.getPriority());
// 추론 다음모델 실행 // 추론 다음모델 실행
@@ -188,7 +189,7 @@ public class MapSheetInferenceJobService {
saveInferenceAiDto.setStatus(Status.IN_PROGRESS.getId()); saveInferenceAiDto.setStatus(Status.IN_PROGRESS.getId());
saveInferenceAiDto.setType(type); saveInferenceAiDto.setType(type);
saveInferenceAiDto.setModelStartDttm(ZonedDateTime.now()); saveInferenceAiDto.setModelStartDttm(ZonedDateTime.now());
saveInferenceAiDto.setRunningModelType(inferenceType); saveInferenceAiDto.setRunningModelType(type);
inferenceResultCoreService.update(saveInferenceAiDto); inferenceResultCoreService.update(saveInferenceAiDto);
} }
@@ -229,7 +230,8 @@ public class MapSheetInferenceJobService {
try { try {
List<Map<String, Object>> list = List<Map<String, Object>> list =
om.readValue(body, new TypeReference<List<Map<String, Object>>>() {}); om.readValue(body, new TypeReference<List<Map<String, Object>>>() {
});
Integer batchIdInt = (Integer) list.get(0).get("batch_id"); Integer batchIdInt = (Integer) list.get(0).get("batch_id");
batchId = batchIdInt.longValue(); batchId = batchIdInt.longValue();