|
5 | 5 | import aa.model.RequiredInputAttribute; |
6 | 6 | import aa.model.UserAttribute; |
7 | 7 | import lombok.Getter; |
| 8 | +import org.apache.hc.client5.http.config.ConnectionConfig; |
| 9 | +import org.apache.hc.client5.http.impl.classic.CloseableHttpClient; |
| 10 | +import org.apache.hc.client5.http.impl.classic.HttpClients; |
| 11 | +import org.apache.hc.client5.http.impl.io.PoolingHttpClientConnectionManager; |
| 12 | +import org.apache.hc.client5.http.impl.io.PoolingHttpClientConnectionManagerBuilder; |
8 | 13 | import org.slf4j.Logger; |
9 | 14 | import org.slf4j.LoggerFactory; |
10 | 15 | import org.springframework.boot.web.client.RestTemplateBuilder; |
|
22 | 27 | import java.util.List; |
23 | 28 | import java.util.Optional; |
24 | 29 | import java.util.Set; |
| 30 | +import java.util.concurrent.TimeUnit; |
25 | 31 | import java.util.regex.Pattern; |
26 | 32 |
|
| 33 | +import static io.restassured.path.json.JsonPath.config; |
27 | 34 | import static java.lang.String.format; |
28 | 35 | import static java.util.Collections.emptyList; |
29 | 36 | import static java.util.Collections.singletonList; |
@@ -119,7 +126,21 @@ protected List<UserAttribute> mapValuesToUserAttribute(String attributeName, Lis |
119 | 126 |
|
120 | 127 | private ClientHttpRequestFactory getRequestFactory(AttributeAuthorityConfiguration |
121 | 128 | attributeAuthorityConfiguration) { |
122 | | - HttpComponentsClientHttpRequestFactory requestFactory = new HttpComponentsClientHttpRequestFactory(); |
| 129 | + ConnectionConfig connectionConfig = ConnectionConfig |
| 130 | + .custom() |
| 131 | + .setTimeToLive(60, TimeUnit.SECONDS) |
| 132 | + .build(); |
| 133 | + |
| 134 | + PoolingHttpClientConnectionManager connManager = |
| 135 | + PoolingHttpClientConnectionManagerBuilder.create() |
| 136 | + .setDefaultConnectionConfig(connectionConfig) |
| 137 | + .build(); |
| 138 | + |
| 139 | + CloseableHttpClient httpClient = HttpClients.custom() |
| 140 | + .setConnectionManager(connManager) |
| 141 | + .build(); |
| 142 | + |
| 143 | + HttpComponentsClientHttpRequestFactory requestFactory = new HttpComponentsClientHttpRequestFactory(httpClient); |
123 | 144 | // Set the connectionRequestTimeout value to 10 seconds |
124 | 145 | requestFactory.setConnectionRequestTimeout(10000); |
125 | 146 | int timeOut = attributeAuthorityConfiguration.getTimeOut(); |
|
0 commit comments