Search in sources :

Example 1 with DefaultConnectionKeepAliveStrategy

use of org.apache.http.impl.client.DefaultConnectionKeepAliveStrategy in project dropwizard by dropwizard.

the class HttpClientBuilderTest method usesDefaultForNonPersistentConnections.

@Test
void usesDefaultForNonPersistentConnections() throws Exception {
    configuration.setKeepAlive(Duration.seconds(1));
    assertThat(builder.using(configuration).createClient(apacheBuilder, connectionManager, "test")).isNotNull();
    final Field field = getInaccessibleField(httpClientBuilderClass, "keepAliveStrategy");
    final DefaultConnectionKeepAliveStrategy strategy = (DefaultConnectionKeepAliveStrategy) field.get(apacheBuilder);
    final HttpContext context = mock(HttpContext.class);
    final HttpResponse response = mock(HttpResponse.class);
    final HeaderIterator iterator = new BasicListHeaderIterator(Collections.singletonList(new BasicHeader(HttpHeaders.CONNECTION, "timeout=50")), HttpHeaders.CONNECTION);
    when(response.headerIterator(HTTP.CONN_KEEP_ALIVE)).thenReturn(iterator);
    assertThat(strategy.getKeepAliveDuration(response, context)).isEqualTo(50_000);
}
Also used : Field(java.lang.reflect.Field) DefaultConnectionKeepAliveStrategy(org.apache.http.impl.client.DefaultConnectionKeepAliveStrategy) BasicHttpContext(org.apache.http.protocol.BasicHttpContext) HttpContext(org.apache.http.protocol.HttpContext) BasicListHeaderIterator(org.apache.http.message.BasicListHeaderIterator) HttpResponse(org.apache.http.HttpResponse) HeaderIterator(org.apache.http.HeaderIterator) BasicListHeaderIterator(org.apache.http.message.BasicListHeaderIterator) BasicHeader(org.apache.http.message.BasicHeader) Test(org.junit.jupiter.api.Test)

Example 2 with DefaultConnectionKeepAliveStrategy

use of org.apache.http.impl.client.DefaultConnectionKeepAliveStrategy in project dropwizard by dropwizard.

the class HttpClientBuilder method createClient.

/**
 * Map the parameters in {@link HttpClientConfiguration} to configuration on a
 * {@link org.apache.http.impl.client.HttpClientBuilder} instance
 *
 * @param builder
 * @param manager
 * @param name
 * @return the configured {@link CloseableHttpClient}
 */
protected ConfiguredCloseableHttpClient createClient(final org.apache.http.impl.client.HttpClientBuilder builder, final InstrumentedHttpClientConnectionManager manager, final String name) {
    final String cookiePolicy = configuration.isCookiesEnabled() ? CookieSpecs.DEFAULT : CookieSpecs.IGNORE_COOKIES;
    final Integer timeout = (int) configuration.getTimeout().toMilliseconds();
    final Integer connectionTimeout = (int) configuration.getConnectionTimeout().toMilliseconds();
    final Integer connectionRequestTimeout = (int) configuration.getConnectionRequestTimeout().toMilliseconds();
    final boolean normalizeUri = configuration.isNormalizeUriEnabled();
    final long keepAlive = configuration.getKeepAlive().toMilliseconds();
    final ConnectionReuseStrategy reuseStrategy = keepAlive == 0 ? new NoConnectionReuseStrategy() : new DefaultConnectionReuseStrategy();
    final HttpRequestRetryHandler retryHandler = configuration.getRetries() == 0 ? NO_RETRIES : (httpRequestRetryHandler == null ? new DefaultHttpRequestRetryHandler(configuration.getRetries(), false) : httpRequestRetryHandler);
    final RequestConfig requestConfig = RequestConfig.custom().setCookieSpec(cookiePolicy).setSocketTimeout(timeout).setConnectTimeout(connectionTimeout).setConnectionRequestTimeout(connectionRequestTimeout).setNormalizeUri(normalizeUri).build();
    final SocketConfig socketConfig = SocketConfig.custom().setTcpNoDelay(true).setSoTimeout(timeout).build();
    builder.setRequestExecutor(createRequestExecutor(name)).setConnectionManager(manager).setDefaultRequestConfig(requestConfig).setDefaultSocketConfig(socketConfig).setConnectionReuseStrategy(reuseStrategy).setRetryHandler(retryHandler).setUserAgent(createUserAgent(name));
    if (keepAlive != 0) {
        // either keep alive based on response header Keep-Alive,
        // or if the server can keep a persistent connection (-1), then override based on client's configuration
        builder.setKeepAliveStrategy(new DefaultConnectionKeepAliveStrategy() {

            @Override
            public long getKeepAliveDuration(HttpResponse response, HttpContext context) {
                final long duration = super.getKeepAliveDuration(response, context);
                return (duration == -1) ? keepAlive : duration;
            }
        });
    }
    // create a tunnel through a proxy host if it's specified in the config
    final ProxyConfiguration proxy = configuration.getProxyConfiguration();
    if (proxy != null) {
        final HttpHost httpHost = new HttpHost(proxy.getHost(), proxy.getPort(), proxy.getScheme());
        builder.setRoutePlanner(new NonProxyListProxyRoutePlanner(httpHost, proxy.getNonProxyHosts()));
        // if the proxy host requires authentication then add the host credentials to the credentials provider
        final AuthConfiguration auth = proxy.getAuth();
        if (auth != null) {
            if (credentialsProvider == null) {
                credentialsProvider = new BasicCredentialsProvider();
            }
            // set the AuthScope
            AuthScope authScope = new AuthScope(httpHost, auth.getRealm(), auth.getAuthScheme());
            // set the credentials type
            Credentials credentials = configureCredentials(auth);
            credentialsProvider.setCredentials(authScope, credentials);
        }
    }
    if (credentialsProvider != null) {
        builder.setDefaultCredentialsProvider(credentialsProvider);
    }
    if (routePlanner != null) {
        builder.setRoutePlanner(routePlanner);
    }
    if (disableContentCompression) {
        builder.disableContentCompression();
    }
    if (redirectStrategy != null) {
        builder.setRedirectStrategy(redirectStrategy);
    }
    if (defaultHeaders != null) {
        builder.setDefaultHeaders(defaultHeaders);
    }
    if (verifier != null) {
        builder.setSSLHostnameVerifier(verifier);
    }
    if (httpProcessor != null) {
        builder.setHttpProcessor(httpProcessor);
    }
    if (serviceUnavailableRetryStrategy != null) {
        builder.setServiceUnavailableRetryStrategy(serviceUnavailableRetryStrategy);
    }
    customizeBuilder(builder);
    return new ConfiguredCloseableHttpClient(builder.build(), requestConfig);
}
Also used : RequestConfig(org.apache.http.client.config.RequestConfig) BasicCredentialsProvider(org.apache.http.impl.client.BasicCredentialsProvider) SocketConfig(org.apache.http.config.SocketConfig) NonProxyListProxyRoutePlanner(io.dropwizard.client.proxy.NonProxyListProxyRoutePlanner) DefaultConnectionReuseStrategy(org.apache.http.impl.DefaultConnectionReuseStrategy) DefaultHttpRequestRetryHandler(org.apache.http.impl.client.DefaultHttpRequestRetryHandler) HttpContext(org.apache.http.protocol.HttpContext) HttpResponse(org.apache.http.HttpResponse) AuthConfiguration(io.dropwizard.client.proxy.AuthConfiguration) NoConnectionReuseStrategy(org.apache.http.impl.NoConnectionReuseStrategy) ProxyConfiguration(io.dropwizard.client.proxy.ProxyConfiguration) DefaultConnectionKeepAliveStrategy(org.apache.http.impl.client.DefaultConnectionKeepAliveStrategy) HttpHost(org.apache.http.HttpHost) AuthScope(org.apache.http.auth.AuthScope) NoConnectionReuseStrategy(org.apache.http.impl.NoConnectionReuseStrategy) DefaultConnectionReuseStrategy(org.apache.http.impl.DefaultConnectionReuseStrategy) ConnectionReuseStrategy(org.apache.http.ConnectionReuseStrategy) DefaultHttpRequestRetryHandler(org.apache.http.impl.client.DefaultHttpRequestRetryHandler) HttpRequestRetryHandler(org.apache.http.client.HttpRequestRetryHandler) NTCredentials(org.apache.http.auth.NTCredentials) Credentials(org.apache.http.auth.Credentials) UsernamePasswordCredentials(org.apache.http.auth.UsernamePasswordCredentials)

Example 3 with DefaultConnectionKeepAliveStrategy

use of org.apache.http.impl.client.DefaultConnectionKeepAliveStrategy in project dropwizard by dropwizard.

the class HttpClientBuilderTest method usesKeepAliveForPersistentConnections.

@Test
void usesKeepAliveForPersistentConnections() throws Exception {
    configuration.setKeepAlive(Duration.seconds(1));
    assertThat(builder.using(configuration).createClient(apacheBuilder, connectionManager, "test")).isNotNull();
    final DefaultConnectionKeepAliveStrategy strategy = (DefaultConnectionKeepAliveStrategy) spyHttpClientBuilderField("keepAliveStrategy", apacheBuilder);
    final HttpContext context = mock(HttpContext.class);
    final HttpResponse response = mock(HttpResponse.class);
    when(response.headerIterator(HTTP.CONN_KEEP_ALIVE)).thenReturn(mock(HeaderIterator.class));
    assertThat(strategy.getKeepAliveDuration(response, context)).isEqualTo(1000);
}
Also used : DefaultConnectionKeepAliveStrategy(org.apache.http.impl.client.DefaultConnectionKeepAliveStrategy) BasicHttpContext(org.apache.http.protocol.BasicHttpContext) HttpContext(org.apache.http.protocol.HttpContext) HttpResponse(org.apache.http.HttpResponse) HeaderIterator(org.apache.http.HeaderIterator) BasicListHeaderIterator(org.apache.http.message.BasicListHeaderIterator) Test(org.junit.jupiter.api.Test)

Example 4 with DefaultConnectionKeepAliveStrategy

use of org.apache.http.impl.client.DefaultConnectionKeepAliveStrategy in project scout.rt by eclipse.

the class ApacheHttpTransportFactory method setConnectionKeepAliveAndRetrySettings.

/**
 * @param builder
 */
protected void setConnectionKeepAliveAndRetrySettings(HttpClientBuilder builder) {
    final boolean keepAliveProp = CONFIG.getPropertyValue(ApacheHttpTransportKeepAliveProperty.class);
    builder.setConnectionReuseStrategy(new ConnectionReuseStrategy() {

        @Override
        public boolean keepAlive(HttpResponse response, HttpContext context) {
            return keepAliveProp;
        }
    });
    // Connections should not be kept open forever, there are numerous reasons a connection could get invalid. Also it does not make much sense to keep a connection open forever
    builder.setKeepAliveStrategy(new DefaultConnectionKeepAliveStrategy() {

        @Override
        public long getKeepAliveDuration(HttpResponse response, HttpContext context) {
            long headerKeepAliveDuration = super.getKeepAliveDuration(response, context);
            return headerKeepAliveDuration < 0 ? 5000 : headerKeepAliveDuration;
        }
    });
    final boolean retryPostProp = CONFIG.getPropertyValue(ApacheHttpTransportRetryPostProperty.class);
    builder.setRetryHandler(new DefaultHttpRequestRetryHandler(1, true) {

        @Override
        protected boolean handleAsIdempotent(HttpRequest request) {
            return retryPostProp || super.handleAsIdempotent(request);
        }
    });
}
Also used : HttpRequest(org.apache.http.HttpRequest) DefaultConnectionKeepAliveStrategy(org.apache.http.impl.client.DefaultConnectionKeepAliveStrategy) HttpContext(org.apache.http.protocol.HttpContext) DefaultHttpRequestRetryHandler(org.apache.http.impl.client.DefaultHttpRequestRetryHandler) HttpResponse(org.apache.http.HttpResponse) ConnectionReuseStrategy(org.apache.http.ConnectionReuseStrategy)

Aggregations

HttpResponse (org.apache.http.HttpResponse)4 DefaultConnectionKeepAliveStrategy (org.apache.http.impl.client.DefaultConnectionKeepAliveStrategy)4 HttpContext (org.apache.http.protocol.HttpContext)4 ConnectionReuseStrategy (org.apache.http.ConnectionReuseStrategy)2 HeaderIterator (org.apache.http.HeaderIterator)2 DefaultHttpRequestRetryHandler (org.apache.http.impl.client.DefaultHttpRequestRetryHandler)2 BasicListHeaderIterator (org.apache.http.message.BasicListHeaderIterator)2 BasicHttpContext (org.apache.http.protocol.BasicHttpContext)2 Test (org.junit.jupiter.api.Test)2 AuthConfiguration (io.dropwizard.client.proxy.AuthConfiguration)1 NonProxyListProxyRoutePlanner (io.dropwizard.client.proxy.NonProxyListProxyRoutePlanner)1 ProxyConfiguration (io.dropwizard.client.proxy.ProxyConfiguration)1 Field (java.lang.reflect.Field)1 HttpHost (org.apache.http.HttpHost)1 HttpRequest (org.apache.http.HttpRequest)1 AuthScope (org.apache.http.auth.AuthScope)1 Credentials (org.apache.http.auth.Credentials)1 NTCredentials (org.apache.http.auth.NTCredentials)1 UsernamePasswordCredentials (org.apache.http.auth.UsernamePasswordCredentials)1 HttpRequestRetryHandler (org.apache.http.client.HttpRequestRetryHandler)1