基于SpringCloud的httpclient线程池

本文介绍了如何在SpringCloud中配置并利用httpclient线程池进行HTTP请求,包括设置连接池以定时关闭空闲连接,以及支持GET和POST请求的方式。

连接池:通过连接池创建httpclien,定时关闭空闲的连接,有get和post(未测试)请求方式。

package com.base.config;

import java.io.IOException;
import java.security.KeyManagementException;
import java.security.KeyStoreException;
import java.security.NoSuchAlgorithmException;
import java.security.cert.CertificateException;
import java.security.cert.X509Certificate;
import java.util.Collections;

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

import org.apache.http.client.HttpClient;
import org.apache.http.config.Registry;
import org.apache.http.config.RegistryBuilder;
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.ssl.SSLContextBuilder;
import org.apache.log4j.Logger;
import org.apache.http.conn.ssl.TrustStrategy;
import org.apache.http.impl.client.DefaultHttpRequestRetryHandler;
import org.apache.http.impl.client.HttpClientBuilder;
import org.apache.http.impl.conn.PoolingHttpClientConnectionManager;
import org.springframework.cloud.client.loadbalancer.LoadBalanced;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Primary;
import org.springframework.http.client.ClientHttpRequestFactory;
import org.springframework.http.client.ClientHttpResponse;
import org.springframework.http.client.HttpComponentsClientHttpRequestFactory;
import org.springframework.web.client.ResponseErrorHandler;
import org.springframework.web.client.RestTemplate;

import com.base.interceptor.RestTemplateInterceptor;

@Configuration
public class RestTemplateConfigBean {
	private static final int DEFAULT_MAX_TOTAL_CONNECTIONS = 100;
	private static final int DEFAULT_MAX_CONECTIONS_PRE_ROUTE = 5;
	private static final int DEFAULT_READ_TIMEOUT_MILLISECONDS = (15 * 1000);
	private static final int HTTP_CONNECT_TIMEOUT = (15 * 1000);
	private static final int CONNECT_REQUEST_TIMEOUT = (15 * 1000);
	private static final int RETRY_COUNT = 3;
	private final static Logger logger = Logger.getLogger(RestTemplateConfigBean.class);

private ClientHttpRequestFactory getClientHttpRequestFactory() throws KeyManagementException, NoSuchAlgorithmException, KeyStoreException {
	HttpClientBuilder httpClientBuilder = HttpClientBuilder.create();
    SSLContext sslContext = new SSLContextBuilder().loadTrustMaterial(null, new TrustStrategy() {
        public boolean isTrusted(X509Certificate[] arg0, String arg1) throws CertificateException {
            return true;
        }
    }).build();
    httpClientBuilder.setSSLContext(sslContext);
    HostnameVerifier hostnameVerifier = NoopHostnameVerifier.INSTANCE;
    SSLConnectionSocketFactory sslConnectionSocketFactory = new SSLConnectionSocketFactory(sslContext,
            hostnameVerifier);
    Registry<ConnectionSocketFactory> socketFactoryRegistry = RegistryBuilder.<ConnectionSocketFactory>create()
            .register("http", PlainConnectionSocketFactory.getSocketFactory())
            .register("https", sslConnectionSocketFactory).build();// 注册http和https请求
    // 开始设置连接池
	PoolingHttpClientConnectionManager poolingHttpClientConnectionManager = new PoolingHttpClientConnectionManager(
            socketFactoryRegistry);
	poolingHttpClientConnectionManager.setMaxTotal(DEFAULT_MAX_TOTAL_CONNECTIONS);
	poolingHttpClientConnectionManager.setDefaultMaxPerRoute(DEFAULT_MAX_CONECTIONS_PRE_ROUTE);
	httpClientBuilder.setConnectionManager(poolingHttpClientConnectionManager);
	httpClientBuilder.setRetryHandler(new DefaultHttpRequestRetryHandler(RETRY_COUNT, true)); // 重试次数
	HttpClient client = httpClientBuilder.build();
    HttpComponentsClientHttpRequestFactory clientHttpRequestFactory = new HttpComponentsClientHttpRequestFactory(
    		client);
    clientHttpRequestFactory.setConnectTimeout(HTTP_CONNECT_TIMEOUT);
    clientHttpRequestFactory.setReadTimeout(DEFAULT_READ_TIMEOUT_MILLISECONDS);
    clientHttpRequestFactory.setConnectionRequestTimeout(CONNECT_REQUEST_TIMEOUT);
    return clientHttpRequestFactory;
}

@Bean
@LoadBalanced
public RestTemplate getRestTemplate() {
	RestTemplate restTemplate = null;
	try {
		restTemplate = new RestTemplate(getClientHttpRequestFactory());
	} catch (KeyManagementException | NoSuchAlgorithmException | KeyStoreException e) {
		e.printStackTrace();
	}
	ResponseErrorHandler responseErrorHandler = new ResponseErrorHandler() {
        @Override
        public boolean hasError(ClientHttpResponse clientHttpResponse) throws IOException {
            return true;
        }
        @Override
        public void handleError(ClientHttpResponse clientHttpResponse) throws IOException {
        	logger.info("httpCode:" + clientHttpResponse.getStatusCode().value());
        }
    };
    restTemplate.setErrorHandler(responseErrorHandler);
    restTemplate.setInterceptors(Collections.singletonList(new RestTemplateInterceptor()));
	return restTemplate;
}

@Bean
@Primary
public RestTemplate localRestTemplate() {
	RestTemplate restTemplate = null;
	try {
		restTemplate = new RestTemplate(getClientHttpRequestFactory());
	} catch (KeyManagementException | NoSuchAlgorithmException | KeyStoreException e) {
		e.printStackTrace();
	}
	ResponseErrorHandler responseErrorHandler = new ResponseErrorHandler() {
        @Override
        public boolean hasError(ClientHttpResponse clientHttpResponse) throws IOException {
            return true;
        }
        @Override
        public void handleError(ClientHttpResponse clientHttpResponse) throws IOException {
        	logger.info(clientHttpResponse.getHeaders());
        }
    };
    restTemplate.setErrorHandler(responseErrorHandler);
    restTemplate.setInterceptors(Collections.singletonList(new RestTemplateInterceptor()));
	return restTemplate;
}

}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值