diff --git a/src/main/java/com/kamco/cd/kamcoback/inference/dto/InferenceResultDto.java b/src/main/java/com/kamco/cd/kamcoback/inference/dto/InferenceResultDto.java index c97942d3..6fcc9612 100644 --- a/src/main/java/com/kamco/cd/kamcoback/inference/dto/InferenceResultDto.java +++ b/src/main/java/com/kamco/cd/kamcoback/inference/dto/InferenceResultDto.java @@ -233,42 +233,34 @@ public class InferenceResultDto { public String getCpuStatus() { String enumId = "SAFETY"; - if (this.cpu_user + this.cpu_system >= 80) { - enumId = "CAUTION"; - } + if (this.cpu_user + this.cpu_system >= 80) enumId = "CAUTION"; return enumId; } public String getGpuStatus() { String enumId = "SAFETY"; - if (this.gpuUtil >= 80) { - enumId = "CAUTION"; - } + if (this.gpuUtil >= 80) enumId = "CAUTION"; return enumId; } public String getMemStatus() { String enumId = "SAFETY"; - if (this.memused >= 80) { - enumId = "CAUTION"; - } + if (this.memused >= 80) enumId = "CAUTION"; return enumId; } public String getCpuStatusName() { - if (this.cpu_user + this.cpu_system >= 80) { - return ServerStatus.CAUTION.getText(); - } + if (this.cpu_user + this.cpu_system >= 80) return ServerStatus.CAUTION.getText(); return ServerStatus.SAFETY.getText(); } public String getGpuStatusName() { - if (this.gpuUtil >= 80) { - return ServerStatus.CAUTION.getText(); - } + if (this.gpuUtil >= 80) return ServerStatus.CAUTION.getText(); return ServerStatus.SAFETY.getText(); } + public String getMemStatusName() { + if (this.memused >= 80) return ServerStatus.CAUTION.getText(); public String getMemStatusName() { if (this.memused >= 80) { return ServerStatus.CAUTION.getText(); 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 34419e5f..be21bef3 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 @@ -238,5 +238,7 @@ public class InferenceResultCoreService { .getInferenceResultByStatus(status) .orElseThrow(() -> new EntityNotFoundException(status)); return entity.getBatchId(); + public List getInferenceServerStatusList() { + return mapSheetLearnRepository.getInferenceServerStatusList(); } }