Add endpoint to return icc of voltage level for a specific result (#171)

Signed-off-by: LE SAULNIER Kevin <[email protected]>

Author: klesaulnier

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

@@ -29,6 +29,7 @@
 import java.net.URLDecoder;
 import java.nio.charset.StandardCharsets;
 import java.util.List;
+import java.util.Map;
 import java.util.Optional;
 import java.util.UUID;
 
@@ -78,6 +79,14 @@ public ResponseEntity<ShortCircuitAnalysisResult> getResult(@Parameter(descripti
                 : ResponseEntity.notFound().build();
     }
 
+    @GetMapping(value = "/results/{resultUuid}/fault_results/icc", produces = APPLICATION_JSON_VALUE)
+    @Operation(summary = "Get a map from fault results for a given short circuit analysis result and a specific voltage level")
+    @ApiResponses(value = {@ApiResponse(responseCode = "200", description = "The map busId -> ICC is returned")})
+    public ResponseEntity<Map<String, Double>> getResult(@Parameter(description = "Result UUID") @PathVariable("resultUuid") UUID resultUuid,
+                                                        @Parameter(description = "Voltage level ID to filter on") @RequestParam(value = "voltageLevelId")String voltageLevelId) {
+        return ResponseEntity.ok().contentType(MediaType.APPLICATION_JSON).body(shortCircuitService.getBasicResultForSpecificEquipment(resultUuid, voltageLevelId));
+    }
+
     @PostMapping(value = "/results/{resultUuid}/csv", produces = APPLICATION_OCTET_STREAM_VALUE)
     @Operation(summary = "Get a short circuit analysis csv result from the database")
     @ApiResponses(value = {@ApiResponse(responseCode = "200", description = "The short circuit analysis csv export"),

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

@@ -36,6 +36,8 @@ public interface FaultResultRepository extends JpaRepository<FaultResultEntity,
     @Query(value = "SELECT faultResultUuid FROM FaultResultEntity WHERE result.resultUuid = ?1")
     Set<UUID> findAllFaultResultUuidsByShortCircuitResultUuid(UUID resultUuid);
 
+    List<FaultResultEntity> findAllByResultResultUuidAndFaultVoltageLevelId(UUID resultUuid, String voltageLevelId);
+
     // From: https://www.baeldung.com/spring-data-jpa-deleteby
     // "The @Query method creates a single SQL query against the database. By comparison, the deleteBy methods execute a read query, then delete each of the items one by one."
     // As we need here to delete thousands of fault results, using native SQL query was required for having decent performance.

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

@@ -446,6 +446,10 @@ private Pageable addDefaultSort(Pageable pageable, String defaultSortColumn) {
         return pageable;
     }
 
+    public List<FaultResultEntity> getFaultResultByVoltageLevelId(UUID resultUuid, String voltageLevelId) {
+        return faultResultRepository.findAllByResultResultUuidAndFaultVoltageLevelId(resultUuid, voltageLevelId);
+    }
+
     @Override
     @Transactional
     public void saveDebugFileLocation(UUID resultUuid, String debugFilePath) {

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

@@ -340,6 +340,10 @@ public ShortCircuitAnalysisResult getResult(UUID resultUuid, FaultResultsMode mo
         return null;
     }
 
+    public Map<String, Double> getBasicResultForSpecificEquipment(UUID resultUuid, String voltageLevelId) {
+        return resultService.getFaultResultByVoltageLevelId(resultUuid, voltageLevelId).stream().collect(Collectors.toMap(fr -> fr.getFault().getElementId(), FaultResultEntity::getCurrent));
+    }
+
     @Transactional(readOnly = true)
     public Page<FaultResult> getFaultResultsPage(UUID rootNetworkUuid,
                                                  String variantId,