Skip to content

Commit db1699a

Browse files
committed
Set time to live #142
1 parent eca774b commit db1699a

File tree

1 file changed

+22
-1
lines changed

1 file changed

+22
-1
lines changed

src/main/java/aa/aggregators/AbstractAttributeAggregator.java

Lines changed: 22 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,11 @@
55
import aa.model.RequiredInputAttribute;
66
import aa.model.UserAttribute;
77
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;
813
import org.slf4j.Logger;
914
import org.slf4j.LoggerFactory;
1015
import org.springframework.boot.web.client.RestTemplateBuilder;
@@ -22,8 +27,10 @@
2227
import java.util.List;
2328
import java.util.Optional;
2429
import java.util.Set;
30+
import java.util.concurrent.TimeUnit;
2531
import java.util.regex.Pattern;
2632

33+
import static io.restassured.path.json.JsonPath.config;
2734
import static java.lang.String.format;
2835
import static java.util.Collections.emptyList;
2936
import static java.util.Collections.singletonList;
@@ -119,7 +126,21 @@ protected List<UserAttribute> mapValuesToUserAttribute(String attributeName, Lis
119126

120127
private ClientHttpRequestFactory getRequestFactory(AttributeAuthorityConfiguration
121128
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);
123144
// Set the connectionRequestTimeout value to 10 seconds
124145
requestFactory.setConnectionRequestTimeout(10000);
125146
int timeOut = attributeAuthorityConfiguration.getTimeOut();

0 commit comments

Comments
 (0)