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 f916b6f2..60fe3de2 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 @@ -46,6 +46,7 @@ import java.util.HashMap; import java.util.List; import java.util.Map; import java.util.Objects; +import java.util.Optional; import java.util.Set; import java.util.UUID; import java.util.stream.Collectors; @@ -81,9 +82,14 @@ public class InferenceResultService { @Value("${inference.batch-url}") private String batchUrl; + @Value("${inference.inference-server-name}") + private String inferenceServerName; + @Value("${file.dataset-dir}") private String datasetDir; + + @Value("${spring.profiles.active}") private String profile; @@ -516,44 +522,46 @@ public class InferenceResultService { public List getInferenceServerStatusList() { + String[] serverNames = inferenceServerName.split(","); + int serveCnt = serverNames.length; + List dtoList = inferenceResultCoreService.getInferenceServerStatusList(); int size = dtoList.size(); + List serverStatusList = new ArrayList<>(); + System.out.println("size =" + size); if (size == 0) { - for (int k = 1; k <= 4; k++) { + for (int k = 0; k < serveCnt; k++) { InferenceServerStatusDto dto = new InferenceServerStatusDto(); - dto.setServerName("server" + k); + dto.setServerName(serverNames[k]); dto.setCpu_user(0); dto.setCpu_system(0); dto.setMemused(0); dto.setKbmemused(0L); dto.setGpuUtil(0); - dtoList.add(dto); + serverStatusList.add(dto); } } else { - for (int k = 1; k <= 4; k++) { - String srvNm = "server" + k; + for (int k = 0; k < serveCnt; k++) { + String srvNm = serverNames[k]; String srvNmChk = "N"; - int serverCnt = - (int) - dtoList.stream().filter(dto -> dto.getServerName().toString().equals(srvNm)).count(); + Optional matchingDto = dtoList.stream() + .filter(dto -> srvNm.equals(dto.getServerName())) + .findFirst(); /* - for (InferenceServerStatusDto dto : dtoList) { - if (srvNm.equals(dto.getServerName())) { - srvNmChk = "Y"; - break; - } - } + int serverChekcCnt = + (int) + dtoList.stream().filter(dto -> dto.getServerName().toString().equals(srvNm)).count(); */ - if (serverCnt == 0) { + if (matchingDto.isEmpty()) { InferenceServerStatusDto dto = new InferenceServerStatusDto(); dto.setServerName(srvNm); dto.setCpu_user(0); @@ -561,16 +569,20 @@ public class InferenceResultService { dto.setMemused(0); dto.setKbmemused(0L); dto.setGpuUtil(0); - dtoList.add(dto); + serverStatusList.add(dto); + } + else + { + serverStatusList.add(matchingDto.get()); } } } - dtoList.sort(Comparator.comparing(InferenceServerStatusDto::getServerName)); + serverStatusList.sort(Comparator.comparing(InferenceServerStatusDto::getServerName)); - return dtoList; + return serverStatusList; } /** diff --git a/src/main/java/com/kamco/cd/kamcoback/postgres/repository/Inference/MapSheetLearnRepositoryImpl.java b/src/main/java/com/kamco/cd/kamcoback/postgres/repository/Inference/MapSheetLearnRepositoryImpl.java index 919cc91c..1e099d78 100644 --- a/src/main/java/com/kamco/cd/kamcoback/postgres/repository/Inference/MapSheetLearnRepositoryImpl.java +++ b/src/main/java/com/kamco/cd/kamcoback/postgres/repository/Inference/MapSheetLearnRepositoryImpl.java @@ -153,8 +153,6 @@ public class MapSheetLearnRepositoryImpl implements MapSheetLearnRepositoryCusto systemMetricEntity.memused, systemMetricEntity.kbmemused, gpuMetricEntity.gpuUtil - // gpuMetricEntity.gpuMemUsed, - // gpuMetricEntity.gpuMemTotal )) .from(systemMetricEntity) .leftJoin(gpuMetricEntity) @@ -163,7 +161,9 @@ public class MapSheetLearnRepositoryImpl implements MapSheetLearnRepositoryCusto .timestamp .in(latestGpuIds) .and(gpuMetricEntity.serverName.eq(systemMetricEntity.serverName))) - .where(systemMetricEntity.timestamp.in(latestIds)) // In 절 사용 + .where(systemMetricEntity.timestamp.in(latestIds) + //.and(systemMetricEntity.serverName.ne("server2")) + ) // In 절 사용 .orderBy(systemMetricEntity.serverName.asc()) .limit(4) .fetch(); diff --git a/src/main/resources/application-dev.yml b/src/main/resources/application-dev.yml index c22952d8..554f7de9 100644 --- a/src/main/resources/application-dev.yml +++ b/src/main/resources/application-dev.yml @@ -92,7 +92,7 @@ file: sync-root-dir: /kamco-nfs/images/ sync-tmp-dir: ${file.sync-root-dir}/tmp sync-file-extention: tfw,tif - sync-auto-exception-start-year: 2025 + sync-auto-exception-start-year: 2024 sync-auto-exception-before-year-cnt: 3 #dataset-dir: D:/kamco-nfs/dataset/ @@ -109,6 +109,7 @@ inference: batch-url: http://10.100.0.11:8000/batches geojson-dir: /kamco-nfs/requests/ jar-path: /kamco-nfs/dataset/shp_exporter-1.0.0.jar + inference-server-name: server1,server2,server3,server4 gukyuin: #url: http://localhost:8080