securitytools.common.http.HttpClientFactory.java Source code

Introduction

Here is the source code for securitytools.common.http.HttpClientFactory.java

Source

/*
 * The MIT License
 *
 * Copyright 2014 Security Tools SDK for Java.
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 * THE SOFTWARE.
 */

package securitytools.common.http;

import java.security.NoSuchAlgorithmException;
import javax.net.ssl.SSLContext;
import org.apache.http.client.config.RequestConfig;
import org.apache.http.conn.ssl.SSLConnectionSocketFactory;
import org.apache.http.conn.ssl.SSLContexts;
import org.apache.http.impl.client.CloseableHttpClient;
import org.apache.http.impl.client.HttpClientBuilder;
import org.apache.http.impl.client.HttpClients;
import org.apache.http.impl.nio.client.CloseableHttpAsyncClient;
import org.apache.http.impl.nio.client.HttpAsyncClientBuilder;
import org.apache.http.impl.nio.client.HttpAsyncClients;
import org.apache.http.nio.conn.ssl.SSLIOSessionStrategy;
import securitytools.common.ClientConfiguration;

/**
 * Creates and configures new instances of Apache HttpClient according to the
 * ClientConfiguration.
 *
 * @author Adam Parsons
 * @version 0.0.1, 05/25/14
 * @since 1.0.0
 */
public class HttpClientFactory {

    public static CloseableHttpClient build(ClientConfiguration clientConfiguration)
            throws NoSuchAlgorithmException {
        HttpClientBuilder builder = HttpClients.custom();

        // Certificate Validation
        if (clientConfiguration.isCertificateValidationEnabled()) {
            builder.setSSLSocketFactory(new SSLConnectionSocketFactory(SSLContext.getDefault(),
                    SSLConnectionSocketFactory.STRICT_HOSTNAME_VERIFIER));
        } else {
            // Disable
            builder.setSSLSocketFactory(new TrustingSSLConnectionSocketFactory());
        }

        // Timeouts
        RequestConfig.Builder requestConfigBuilder = RequestConfig.custom();
        requestConfigBuilder.setConnectTimeout(clientConfiguration.getConnectionTimeout());
        requestConfigBuilder.setConnectionRequestTimeout(clientConfiguration.getConnectionTimeout());
        requestConfigBuilder.setSocketTimeout(clientConfiguration.getSocketTimeout());
        builder.setDefaultRequestConfig(requestConfigBuilder.build());

        // User Agent
        builder.setUserAgent(clientConfiguration.getUserAgent());

        // Proxy
        if (clientConfiguration.getProxyHost() != null) {
            builder.setProxy(clientConfiguration.getProxyHost());
        }

        return builder.build();
    }

    public static CloseableHttpAsyncClient buildAsync(ClientConfiguration clientConfiguration)
            throws NoSuchAlgorithmException {
        HttpAsyncClientBuilder builder = HttpAsyncClients.custom();

        // Certificate Validation
        // TODO
        if (clientConfiguration.isCertificateValidationEnabled()) {
            builder.setSSLStrategy(new SSLIOSessionStrategy(SSLContext.getDefault(),
                    SSLConnectionSocketFactory.STRICT_HOSTNAME_VERIFIER));
        } else {
            // Disable
            SSLIOSessionStrategy sslStrategy = new SSLIOSessionStrategy(SSLContext.getDefault(),
                    SSLConnectionSocketFactory.ALLOW_ALL_HOSTNAME_VERIFIER);
            builder.setSSLStrategy(sslStrategy);
        }

        // Timeouts
        RequestConfig.Builder requestConfigBuilder = RequestConfig.custom();
        requestConfigBuilder.setConnectTimeout(clientConfiguration.getConnectionTimeout());
        requestConfigBuilder.setConnectionRequestTimeout(clientConfiguration.getConnectionTimeout());
        requestConfigBuilder.setSocketTimeout(clientConfiguration.getSocketTimeout());
        builder.setDefaultRequestConfig(requestConfigBuilder.build());

        // User Agent
        builder.setUserAgent(clientConfiguration.getUserAgent());

        // Proxy
        if (clientConfiguration.getProxyHost() != null) {
            builder.setProxy(clientConfiguration.getProxyHost());
        }

        return builder.build();
    }

}