diff --git a/src/main/java/org/gridsuite/voltageinit/server/RestTemplateConfig.java b/src/main/java/org/gridsuite/voltageinit/server/RestTemplateConfig.java index 28cb198..a140b12 100644 --- a/src/main/java/org/gridsuite/voltageinit/server/RestTemplateConfig.java +++ b/src/main/java/org/gridsuite/voltageinit/server/RestTemplateConfig.java @@ -13,6 +13,7 @@ import com.powsybl.commons.report.ReportNodeDeserializer; import com.powsybl.commons.report.ReportNodeJsonModule; import com.powsybl.openreac.parameters.input.json.OpenReactJsonModule; +import org.springframework.boot.web.client.RestTemplateBuilder; import org.springframework.context.annotation.Bean; import org.springframework.context.annotation.Configuration; import org.springframework.http.converter.HttpMessageConverter; @@ -27,8 +28,8 @@ public class RestTemplateConfig { @Bean - public RestTemplate restTemplate() { - final RestTemplate restTemplate = new RestTemplate(); + public RestTemplate restTemplate(RestTemplateBuilder builder) { + final RestTemplate restTemplate = builder.build(); //find and replace Jackson message converter with our own for (int i = 0; i < restTemplate.getMessageConverters().size(); i++) { diff --git a/src/main/java/org/gridsuite/voltageinit/server/service/NetworkModificationService.java b/src/main/java/org/gridsuite/voltageinit/server/service/NetworkModificationService.java index 0b843a8..b7cf1b4 100644 --- a/src/main/java/org/gridsuite/voltageinit/server/service/NetworkModificationService.java +++ b/src/main/java/org/gridsuite/voltageinit/server/service/NetworkModificationService.java @@ -55,17 +55,19 @@ public class NetworkModificationService { private String networkModificationServerBaseUri; - private final RestTemplate restTemplate = new RestTemplate(); + private final RestTemplate restTemplate; private final ObjectMapper objectMapper; private final UuidGeneratorService uuidGeneratorService; NetworkModificationService(@Value("${gridsuite.services.network-modification-server.base-uri:http://network-modification-server/}") String networkModificationServerBaseUri, - ObjectMapper objectMapper, UuidGeneratorService uuidGeneratorService) { + ObjectMapper objectMapper, UuidGeneratorService uuidGeneratorService, + RestTemplate restTemplate) { this.networkModificationServerBaseUri = networkModificationServerBaseUri; this.objectMapper = objectMapper; this.uuidGeneratorService = uuidGeneratorService; + this.restTemplate = restTemplate; } public void setNetworkModificationServerBaseUri(String networkModificationServerBaseUri) { diff --git a/src/main/java/org/gridsuite/voltageinit/server/service/parameters/FilterService.java b/src/main/java/org/gridsuite/voltageinit/server/service/parameters/FilterService.java index e12af5e..5410abf 100644 --- a/src/main/java/org/gridsuite/voltageinit/server/service/parameters/FilterService.java +++ b/src/main/java/org/gridsuite/voltageinit/server/service/parameters/FilterService.java @@ -56,16 +56,18 @@ public class FilterService implements FilterLoader { private static String filterServerBaseUri; - private final RestTemplate restTemplate = new RestTemplate(); + private final RestTemplate restTemplate; private final NetworkStoreService networkStoreService; public static final String FILTERS_NOT_FOUND = "Filters not found"; public FilterService(NetworkStoreService networkStoreService, - @Value("${gridsuite.services.filter-server.base-uri:http://filter-server/}") String filterServerBaseUri) { + @Value("${gridsuite.services.filter-server.base-uri:http://filter-server/}") String filterServerBaseUri, + RestTemplate restTemplate) { this.networkStoreService = networkStoreService; setFilterServerBaseUri(filterServerBaseUri); + this.restTemplate = restTemplate; } public static void setFilterServerBaseUri(String filterServerBaseUri) { diff --git a/src/test/java/org/gridsuite/voltageinit/server/service/parameters/FilterServiceTest.java b/src/test/java/org/gridsuite/voltageinit/server/service/parameters/FilterServiceTest.java index 9cef230..ed76dc4 100644 --- a/src/test/java/org/gridsuite/voltageinit/server/service/parameters/FilterServiceTest.java +++ b/src/test/java/org/gridsuite/voltageinit/server/service/parameters/FilterServiceTest.java @@ -33,7 +33,7 @@ class FilterServiceTest { @BeforeEach void setUp() { - filterService = spy(new FilterService(Mockito.mock(NetworkStoreService.class), "http://filter-server/")); + filterService = spy(new FilterService(Mockito.mock(NetworkStoreService.class), "http://filter-server/", null)); } @Test