org.zalando.riptide.spring.HttpClientFactoryBean Maven / Gradle / Ivy

Go to download
package org.zalando.riptide.spring;

import org.apache.http.HttpRequestInterceptor;
import org.apache.http.HttpResponseInterceptor;
import org.apache.http.client.config.RequestConfig;
import org.apache.http.conn.ssl.SSLConnectionSocketFactory;
import org.apache.http.impl.client.CloseableHttpClient;
import org.apache.http.impl.client.HttpClientBuilder;
import org.apache.http.ssl.SSLContextBuilder;
import org.apache.http.ssl.SSLContexts;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.config.AbstractFactoryBean;
import org.zalando.riptide.spring.RiptideSettings.Client.Keystore;

import java.io.FileNotFoundException;
import java.net.URL;
import java.util.List;
import java.util.concurrent.TimeUnit;

import static java.lang.String.format;
import static org.apache.http.conn.ssl.SSLConnectionSocketFactory.getDefaultHostnameVerifier;

class HttpClientFactoryBean extends AbstractFactoryBean {

    private static final Logger LOG = LoggerFactory.getLogger(HttpClientFactoryBean.class);

    private final HttpClientBuilder builder = HttpClientBuilder.create();
    private final RequestConfig.Builder config = RequestConfig.custom();
    private HttpClientCustomizer customizer = $ -> {
    };

    public void setFirstRequestInterceptors(final List interceptors) {
        interceptors.forEach(builder::addInterceptorFirst);
    }

    public void setLastRequestInterceptors(final List interceptors) {
        interceptors.forEach(builder::addInterceptorLast);
    }

    public void setLastResponseInterceptors(final List interceptors) {
        interceptors.forEach(builder::addInterceptorLast);
    }

    public void setConnectTimeout(final TimeSpan connectTimeout) {
        config.setConnectTimeout((int) connectTimeout.to(TimeUnit.MILLISECONDS));
    }

    public void setSocketTimeout(final TimeSpan socketTimeout) {
        config.setSocketTimeout((int) socketTimeout.to(TimeUnit.MILLISECONDS));
    }

    public void setConnectionTimeToLive(final TimeSpan timeToLive) {
        builder.setConnectionTimeToLive(timeToLive.getAmount(), timeToLive.getUnit());
    }

    public void setMaxConnectionsPerRoute(final int maxConnectionsPerRoute) {
        builder.setMaxConnPerRoute(maxConnectionsPerRoute);
    }

    public void setMaxConnectionsTotal(final int maxConnectionsTotal) {
        builder.setMaxConnTotal(maxConnectionsTotal);
    }

    public void setTrustedKeystore(final Keystore keystore) throws Exception {
        final SSLContextBuilder ssl = SSLContexts.custom();

        final String path = keystore.getPath();
        final String password = keystore.getPassword();

        final URL resource = HttpClientFactoryBean.class.getClassLoader().getResource(path);

        if (resource == null) {
            throw new FileNotFoundException(format("Keystore [%s] not found.", path));
        }

        try {
            ssl.loadTrustMaterial(resource, password == null ? null : password.toCharArray());
            builder.setSSLSocketFactory(new SSLConnectionSocketFactory(ssl.build(), getDefaultHostnameVerifier()));
        } catch (final Exception e) {
            LOG.error("Error loading keystore [{}]:", path, e); // log full exception, bean initialization code swallows it
            throw e;
        }
    }

    public void setCustomizer(final HttpClientCustomizer customizer) {
        this.customizer = customizer;
    }

    @Override
    public Class getObjectType() {
        return CloseableHttpClient.class;
    }

    @Override
    protected CloseableHttpClient createInstance() throws Exception {
        builder.setDefaultRequestConfig(config.build());
        customizer.customize(builder);
        return builder.build();
    }

    @Override
    protected void destroyInstance(final CloseableHttpClient client) throws Exception {
        client.close();
    }

}