com.alibaba.csb.sdk.internel.HttpClientFactory Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of http-client Show documentation
There is a newer version: 1.1.5.11
package com.alibaba.csb.sdk.internel;

import javax.net.ssl.HostnameVerifier;
import javax.net.ssl.SSLContext;
import javax.net.ssl.SSLSocketFactory;

import org.apache.http.HeaderElement;
import org.apache.http.HeaderElementIterator;
import org.apache.http.HttpResponse;
import org.apache.http.config.Registry;
import org.apache.http.config.RegistryBuilder;
import org.apache.http.conn.ConnectionKeepAliveStrategy;
import org.apache.http.conn.socket.ConnectionSocketFactory;
import org.apache.http.conn.socket.PlainConnectionSocketFactory;
import org.apache.http.conn.ssl.NoopHostnameVerifier;
import org.apache.http.conn.ssl.SSLConnectionSocketFactory;
import org.apache.http.conn.ssl.TrustStrategy;
import org.apache.http.impl.client.CloseableHttpClient;
import org.apache.http.impl.client.HttpClients;
import org.apache.http.impl.conn.PoolingHttpClientConnectionManager;
import org.apache.http.message.BasicHeaderElementIterator;
import org.apache.http.protocol.HTTP;
import org.apache.http.protocol.HttpContext;
import org.apache.http.ssl.SSLContextBuilder;

import com.alibaba.csb.sdk.HttpCallerException;

public class HttpClientFactory {
	private static ConnectionKeepAliveStrategy createKeepAliveStrategy() {
		return new ConnectionKeepAliveStrategy() {
			@Override
			public long getKeepAliveDuration(HttpResponse response, HttpContext context) {
				HeaderElementIterator it = new BasicHeaderElementIterator(
						response.headerIterator(HTTP.CONN_KEEP_ALIVE));
				while (it.hasNext()) {
					HeaderElement he = it.nextElement();
					String param = he.getName();
					String value = he.getValue();
					if (value != null && param.equalsIgnoreCase("timeout")) {
						return Long.parseLong(value) * 1000;
					}
				}
				return 5 * 1000;
			}
		};
	}

	public static CloseableHttpClient createCloseableHttpClient(PoolingHttpClientConnectionManager connManager)
			throws HttpCallerException {
		ConnectionKeepAliveStrategy myStrategy = createKeepAliveStrategy();
		CloseableHttpClient client;
		try {
			client = HttpClients.custom()
					.setConnectionManager(connManager)
					.setKeepAliveStrategy(myStrategy)
					.build();
		} catch (Exception e) {
			throw new HttpCallerException("Failed to create httpclient: " + e.getMessage(), e);
		}

		return client;
	}

	/**
	 * Create a connection pool which supports http and https socket
	 * @return
	 * @throws HttpCallerException
	 */
	public static PoolingHttpClientConnectionManager createConnManager() throws HttpCallerException {
		try {
			// ignore SSL certificate info with the below two setting:

			// 1. trust https server certificate always.
			SSLContext sslContext = new SSLContextBuilder().loadTrustMaterial(null, new TrustStrategy() {
				@Override
				public boolean isTrusted(java.security.cert.X509Certificate[] chain, String authType)
						throws java.security.cert.CertificateException {
					return true;
				}
			}).build();

			// 2. hostname verifier pass
			HostnameVerifier hostnameVerifier = NoopHostnameVerifier.INSTANCE;

			SSLConnectionSocketFactory sslSocketFactory = new SSLConnectionSocketFactory(sslContext, hostnameVerifier);

			Registry socketFactoryRegistry = RegistryBuilder.create()
					.register("http", PlainConnectionSocketFactory.getSocketFactory())
					.register("https", sslSocketFactory).build();

			return new PoolingHttpClientConnectionManager(socketFactoryRegistry);
		} catch (Exception e) {
			throw new HttpCallerException("Failed to create httpclient: " + e.getMessage(), e);
		}
	}
}