HttpClient工具类

本文介绍了一个用于构建和配置HTTP客户端的工具类,该工具类能够帮助开发者设置各种HTTP请求参数,如超时时间、连接池大小等,并提供了对HTTPS的支持及信任所有证书的能力。此外,还提供了一个用于监控和管理空闲连接的线程。

工具类代码:

package com.zzkx.udp.framework.utils;

import java.io.IOException;
import java.security.cert.CertificateException;
import java.security.cert.X509Certificate;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicBoolean;

import javax.net.ssl.SSLContext;

import org.apache.http.client.HttpRequestRetryHandler;
import org.apache.http.client.config.RequestConfig;
import org.apache.http.config.Registry;
import org.apache.http.config.RegistryBuilder;
import org.apache.http.config.SocketConfig;
import org.apache.http.conn.HttpClientConnectionManager;
import org.apache.http.conn.socket.ConnectionSocketFactory;
import org.apache.http.conn.socket.PlainConnectionSocketFactory;
import org.apache.http.conn.ssl.SSLConnectionSocketFactory;
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.protocol.HttpContext;
import org.apache.http.ssl.SSLContexts;
import org.apache.http.ssl.TrustStrategy;

public class HttpClientBuilder {

	  private final AtomicBoolean prepared = new AtomicBoolean(false);
	  private int connectionRequestTimeout = 3000;
	  private int connectionTimeout = 5000;
	  private int soTimeout = 5000;
	  private int idleConnTimeout = 60000;
	  private int checkWaitTime = 60000;
	  private int maxConnPerHost = 10;
	  private int maxTotalConn = 50;
	  private HttpRequestRetryHandler httpRequestRetryHandler = new HttpRequestRetryHandler() {
	    @Override
	    public boolean retryRequest(IOException exception, int executionCount, HttpContext context) {
	      return false;
	    }
	  };
	  private SSLConnectionSocketFactory sslConnectionSocketFactory = SSLConnectionSocketFactory.getSocketFactory();
	  private PlainConnectionSocketFactory plainConnectionSocketFactory = PlainConnectionSocketFactory.getSocketFactory();
	  /**
	   * 闲置连接监控线程
	   */
	  private IdleConnectionMonitorThread idleConnectionMonitorThread;
	  /**
	   * 持有client对象,仅初始化一次,避免多service实例的时候造成重复初始化的问题
	   */
	  private CloseableHttpClient closeableHttpClient;

	  private HttpClientBuilder() {
	  }

	  public static HttpClientBuilder get() {
	    return HttpClientBuilder.SingletonHolder.INSTANCE;
	  }


	  /**
	   * 获取链接的超时时间设置,默认3000ms
	   * <p>
	   * 设置为零时不超时,一直等待.
	   * 设置为负数是使用系统默认设置(非上述的3000ms的默认值,而是httpclient的默认设置).
	   * </p>
	   *
	   * @param connectionRequestTimeout 获取链接的超时时间设置(单位毫秒),默认3000ms
	   */
	  public void setConnectionRequestTimeout(int connectionRequestTimeout) {
	    this.connectionRequestTimeout = connectionRequestTimeout;
	  }

	  /**
	   * 建立链接的超时时间,默认为5000ms.由于是在链接池获取链接,此设置应该并不起什么作用
	   * <p>
	   * 设置为零时不超时,一直等待.
	   * 设置为负数是使用系统默认设置(非上述的5000ms的默认值,而是httpclient的默认设置).
	   * </p>
	   *
	   * @param connectionTimeout 建立链接的超时时间设置(单位毫秒),默认5000ms
	   */
	  public void setConnectionTimeout(int connectionTimeout) {
	    this.connectionTimeout = connectionTimeout;
	  }

	  /**
	   * 默认NIO的socket超时设置,默认5000ms.
	   *
	   * @param soTimeout 默认NIO的socket超时设置,默认5000ms.
	   * @see java.net.SocketOptions#SO_TIMEOUT
	   */
	  public void setSoTimeout(int soTimeout) {
	    this.soTimeout = soTimeout;
	  }

	  /**
	   * 空闲链接的超时时间,默认60000ms.
	   * <p>
	   * 超时的链接将在下一次空闲链接检查是被销毁
	   * </p>
	   *
	   * @param idleConnTimeout 空闲链接的超时时间,默认60000ms.
	   */
	  public void setIdleConnTimeout(int idleConnTimeout) {
	    this.idleConnTimeout = idleConnTimeout;
	  }

	  /**
	   * 检查空间链接的间隔周期,默认60000ms.
	   *
	   * @param checkWaitTime 检查空间链接的间隔周期,默认60000ms.
	   */
	  public void setCheckWaitTime(int checkWaitTime) {
	    this.checkWaitTime = checkWaitTime;
	  }

	  /**
	   * 每路的最大链接数,默认10
	   *
	   * @param maxConnPerHost 每路的最大链接数,默认10
	   */
	  public void setMaxConnPerHost(int maxConnPerHost) {
	    this.maxConnPerHost = maxConnPerHost;
	  }

	  /**
	   * 最大总连接数,默认50
	   *
	   * @param maxTotalConn 最大总连接数,默认50
	   */
	  public void setMaxTotalConn(int maxTotalConn) {
	    this.maxTotalConn = maxTotalConn;
	  }


	  public IdleConnectionMonitorThread getIdleConnectionMonitorThread() {
	    return this.idleConnectionMonitorThread;
	  }

	  private synchronized void prepare() {
	    if (prepared.get()) {
	      return;
	    }
	    Registry<ConnectionSocketFactory> registry = RegistryBuilder.<ConnectionSocketFactory>create()
	      .register("http", this.plainConnectionSocketFactory)
	      .register("https", this.buildSSLConnectionSocketFactory())
	      .build();

	    @SuppressWarnings("resource")
	    PoolingHttpClientConnectionManager connectionManager = new PoolingHttpClientConnectionManager(registry);
	    connectionManager.setMaxTotal(this.maxTotalConn);
	    connectionManager.setDefaultMaxPerRoute(this.maxConnPerHost);
	    connectionManager.setDefaultSocketConfig(
	      SocketConfig.copy(SocketConfig.DEFAULT)
	        .setSoTimeout(this.soTimeout)
	        .build()
	    );

	    this.idleConnectionMonitorThread = new IdleConnectionMonitorThread(
	      connectionManager, this.idleConnTimeout, this.checkWaitTime);
	    this.idleConnectionMonitorThread.setDaemon(true);
	    this.idleConnectionMonitorThread.start();

	    org.apache.http.impl.client.HttpClientBuilder httpClientBuilder = HttpClients.custom()
	      .setConnectionManager(connectionManager)
	      .setConnectionManagerShared(true)
	      .setSSLSocketFactory(this.buildSSLConnectionSocketFactory())
	      .setDefaultRequestConfig(
	        RequestConfig.custom()
	          .setSocketTimeout(this.soTimeout)
	          .setConnectTimeout(this.connectionTimeout)
	          .setConnectionRequestTimeout(this.connectionRequestTimeout)
	          .build()
	      )
	      .setRetryHandler(this.httpRequestRetryHandler);



	   this.closeableHttpClient = httpClientBuilder.build();
	    prepared.set(true);
	  }

	  private SSLConnectionSocketFactory buildSSLConnectionSocketFactory() {
	    try {
	      SSLContext sslcontext = SSLContexts.custom()
	        //忽略掉对服务器端证书的校验
	        .loadTrustMaterial(new TrustStrategy() {
	          @Override
	          public boolean isTrusted(X509Certificate[] chain, String authType) throws CertificateException {
	            return true;
	          }
	        }).build();

	      return new SSLConnectionSocketFactory(
	        sslcontext,
	        new String[]{"TLSv1"},
	        null,
	        SSLConnectionSocketFactory.getDefaultHostnameVerifier());
	    } catch (Exception e) {
	    	e.printStackTrace();
	    }

	    return null;
	  }

	  public CloseableHttpClient build() {
	    if (!prepared.get()) {
	      prepare();
	    }
	    return this.closeableHttpClient;
	  }

	  /**
	   * HttpClientBuilder 改为单例模式,并持有唯一的CloseableHttpClient(仅首次调用创建)
	   */
	  private static class SingletonHolder {
	    private static final HttpClientBuilder INSTANCE = new HttpClientBuilder();
	  }

	  public static class IdleConnectionMonitorThread extends Thread {
	    private final HttpClientConnectionManager connMgr;
	    private final int idleConnTimeout;
	    private final int checkWaitTime;
	    private volatile boolean shutdown;

	    public IdleConnectionMonitorThread(HttpClientConnectionManager connMgr, int idleConnTimeout, int checkWaitTime) {
	      super("IdleConnectionMonitorThread");
	      this.connMgr = connMgr;
	      this.idleConnTimeout = idleConnTimeout;
	      this.checkWaitTime = checkWaitTime;
	    }

	    @Override
	    public void run() {
	      try {
	        while (!this.shutdown) {
	          synchronized (this) {
	            wait(this.checkWaitTime);
	            this.connMgr.closeExpiredConnections();
	            this.connMgr.closeIdleConnections(this.idleConnTimeout,
	              TimeUnit.MILLISECONDS);
	          }
	        }
	      } catch (InterruptedException ignore) {
	      }
	    }

	    public void trigger() {
	      synchronized (this) {
	        notifyAll();
	      }
	    }

	    public void shutdown() {
	      this.shutdown = true;
	      synchronized (this) {
	        notifyAll();
	      }
	    }
	  }
}


使用:

		CloseableHttpClient build = HttpClientBuilder.get().build();
		HttpPost post = new HttpPost(uri);

		StringEntity entity = new StringEntity(requestParams.toString(), Consts.UTF_8);
		post.setEntity(entity);
		CloseableHttpResponse response = build.execute(post);
		String result = handleResponse(response);

响应处理:handleRsponse方法


	/**
	 * 将响应流转成字符串
	 * @param response
	 * @return
	 * @throws IOException
	 */
	  public String handleResponse(final HttpResponse response) throws IOException {
	    final StatusLine statusLine = response.getStatusLine();
	    final HttpEntity entity = response.getEntity();
	    if (statusLine.getStatusCode() >= 300) {
	      EntityUtils.consume(entity);
	      throw new HttpResponseException(statusLine.getStatusCode(), statusLine.getReasonPhrase());
	    }
	    return entity == null ? null : EntityUtils.toString(entity, Consts.UTF_8);
	  }

源码出处:https://github.com/Wechat-Group/weixin-java-tools/wiki
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值