get filter enum values (#86)

Signed-off-by: Rehili Ghazwa <[email protected]>

Author: ghazwarhili

src/main/java/org/gridsuite/shortcircuit/server/ShortCircuitController.java

@@ -165,18 +165,18 @@ public ResponseEntity<Void> stop(@Parameter(description = "Result UUID") @PathVa
         return ResponseEntity.ok().build();
     }
 
-    @GetMapping(value = "/fault-types", produces = APPLICATION_JSON_VALUE)
+    @GetMapping(value = "/results/{resultUuid}/fault-types", produces = APPLICATION_JSON_VALUE)
     @Operation(summary = "Get list of fault types")
     @ApiResponses(value = {@ApiResponse(responseCode = "200", description = "The list of fault types")})
-    public ResponseEntity<FaultType[]> getFaultTypes() {
-        return ResponseEntity.ok().body(FaultType.values());
+    public ResponseEntity<List<FaultType>> getFaultTypes(@Parameter(description = "Result UUID") @PathVariable("resultUuid") UUID resultUuid) {
+        return ResponseEntity.ok().contentType(MediaType.APPLICATION_JSON).body(shortCircuitService.getFaultTypes(resultUuid));
     }
 
-    @GetMapping(value = "/limit-violation-types", produces = APPLICATION_JSON_VALUE)
+    @GetMapping(value = "/results/{resultUuid}/limit-violation-types", produces = APPLICATION_JSON_VALUE)
     @Operation(summary = "Get list of limit violation types")
-    @ApiResponses(value = {@ApiResponse(responseCode = "200", description = "The list of limit violation types")})
-    public ResponseEntity<List<LimitViolationType>> getLimitTypes() {
-        return ResponseEntity.ok().body(List.of(LimitViolationType.LOW_SHORT_CIRCUIT_CURRENT, LimitViolationType.HIGH_SHORT_CIRCUIT_CURRENT));
+    @ApiResponses(@ApiResponse(responseCode = "200", description = "The list of limit violation types"))
+    public ResponseEntity<List<LimitViolationType>> getLimitTypes(@Parameter(description = "Result UUID") @PathVariable("resultUuid") UUID resultUuid) {
+        return ResponseEntity.ok().contentType(MediaType.APPLICATION_JSON).body(shortCircuitService.getLimitTypes(resultUuid));
     }
 
 }

src/main/java/org/gridsuite/shortcircuit/server/repositories/FaultResultRepository.java

@@ -6,6 +6,8 @@
  */
 package org.gridsuite.shortcircuit.server.repositories;
 
+import com.powsybl.security.LimitViolationType;
+import com.powsybl.shortcircuit.Fault;
 import org.gridsuite.shortcircuit.server.entities.FaultResultEntity;
 import org.springframework.data.jpa.domain.Specification;
 import org.springframework.data.jpa.repository.*;
@@ -55,4 +57,14 @@ public interface FaultResultRepository extends JpaRepository<FaultResultEntity,
     interface EntityId {
         UUID getFaultResultUuid();
     }
+
+    @Query(value = " SELECT DISTINCT limit_Type FROM fault_result_entity " +
+            " where result_result_uuid = :resultUuid AND limit_Type not like ''" +
+            "order by limit_Type", nativeQuery = true)
+    List<LimitViolationType> findLimitTypes(UUID resultUuid);
+
+    @Query(value = " SELECT DISTINCT fault_Type FROM fault_result_entity " +
+            " where result_result_uuid = :resultUuid AND fault_Type not like ''" +
+            "order by fault_Type", nativeQuery = true)
+    List<Fault.FaultType> findFaultTypes(UUID resultUuid);
 }

src/main/java/org/gridsuite/shortcircuit/server/repositories/ShortCircuitAnalysisResultRepository.java

@@ -6,6 +6,7 @@
  */
 package org.gridsuite.shortcircuit.server.repositories;
 
+import com.powsybl.security.LimitViolationType;
 import com.powsybl.shortcircuit.*;
 import lombok.extern.slf4j.Slf4j;
 import org.gridsuite.shortcircuit.server.dto.FaultResultsMode;
@@ -166,6 +167,18 @@ public void insertStatus(List<UUID> resultUuids, String status) {
             .map(uuid -> toStatusEntity(uuid, status)).toList());
     }
 
+    @Transactional(readOnly = true)
+    public List<LimitViolationType> findLimitTypes(UUID resultUuid) {
+        Objects.requireNonNull(resultUuid);
+        return faultResultRepository.findLimitTypes(resultUuid);
+    }
+
+    @Transactional(readOnly = true)
+    public List<Fault.FaultType> findFaultTypes(UUID resultUuid) {
+        Objects.requireNonNull(resultUuid);
+        return faultResultRepository.findFaultTypes(resultUuid);
+    }
+
     @Transactional
     public void insert(UUID resultUuid, ShortCircuitAnalysisResult result, ShortCircuitRunContext runContext, String status) {
         Objects.requireNonNull(resultUuid);

src/main/java/org/gridsuite/shortcircuit/server/service/ShortCircuitService.java

@@ -7,6 +7,7 @@
 package org.gridsuite.shortcircuit.server.service;
 
 import com.fasterxml.jackson.databind.ObjectMapper;
+import com.powsybl.security.LimitViolationType;
 import com.powsybl.ws.commons.LogUtils;
 import com.univocity.parsers.csv.CsvWriter;
 import com.univocity.parsers.csv.CsvWriterSettings;
@@ -312,4 +313,12 @@ public void setStatus(List<UUID> resultUuids, String status) {
     public void stop(UUID resultUuid, String receiver) {
         notificationService.sendCancelMessage(new ShortCircuitCancelContext(resultUuid, receiver).toMessage());
     }
+
+    public List<LimitViolationType> getLimitTypes(UUID resultUuid) {
+        return resultRepository.findLimitTypes(resultUuid);
+    }
+
+    public List<com.powsybl.shortcircuit.Fault.FaultType> getFaultTypes(UUID resultUuid) {
+        return resultRepository.findFaultTypes(resultUuid);
+    }
 }

src/test/java/org/gridsuite/shortcircuit/server/ShortCircuitAnalysisControllerTest.java

@@ -738,6 +738,26 @@ public void testStatus() {
         assertEquals(ShortCircuitAnalysisStatus.NOT_DONE.name(), result.getResponse().getContentAsString());
     }
 
+    @SneakyThrows
+    @Test
+    public void testGetFaultTypes() {
+        MvcResult result = mockMvc.perform(get(
+                        "/" + VERSION + "/results/{resultUuid}/fault-types", RESULT_UUID))
+                .andExpect(status().isOk())
+                .andReturn();
+        assertEquals("[]", result.getResponse().getContentAsString());
+    }
+
+    @SneakyThrows
+    @Test
+    public void testGetLimitTypes() {
+        MvcResult result = mockMvc.perform(get(
+                        "/" + VERSION + "/results/{resultUuid}/limit-violation-types", RESULT_UUID))
+                .andExpect(status().isOk())
+                .andReturn();
+        assertEquals("[]", result.getResponse().getContentAsString());
+    }
+
     @SneakyThrows
     @Test
     public void runWithReportTest() {