httpclient pool帮助类
摘自爬虫类 用于频繁请求减少网络消耗
import java.io.IOException;
import java.io.InterruptedIOException;
import java.io.UnsupportedEncodingException;
import java.net.UnknownHostException;
import java.util.*;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.TimeUnit;
import javax.net.ssl.SSLException;
import javax.net.ssl.SSLHandshakeException;
import com.alibaba.fastjson.JSONObject;
import org.apache.commons.lang3.ClassUtils;
import org.apache.http.HttpEntity;
import org.apache.http.HttpEntityEnclosingRequest;
import org.apache.http.HttpHost;
import org.apache.http.HttpRequest;
import org.apache.http.NameValuePair;
import org.apache.http.NoHttpResponseException;
import org.apache.http.client.HttpRequestRetryHandler;
import org.apache.http.client.config.RequestConfig;
import org.apache.http.client.entity.UrlEncodedFormEntity;
import org.apache.http.client.methods.CloseableHttpResponse;
import org.apache.http.client.methods.HttpGet;
import org.apache.http.client.methods.HttpPost;
import org.apache.http.client.methods.HttpRequestBase;
import org.apache.http.client.protocol.HttpClientContext;
import org.apache.http.conn.ConnectTimeoutException;
import org.apache.http.conn.routing.HttpRoute;
import org.apache.http.entity.StringEntity;
import org.apache.http.impl.client.CloseableHttpClient;
import org.apache.http.impl.client.DefaultConnectionKeepAliveStrategy;
import org.apache.http.impl.client.HttpClients;
import org.apache.http.impl.conn.PoolingHttpClientConnectionManager;
import org.apache.http.protocol.HttpContext;
import org.apache.http.util.EntityUtils;
/**
* http请求池
*/
public class HttpClientUtil {
//超时时间
static final int timeOut = 6 * 1000;
//httpclient
private static CloseableHttpClient httpClient = null;
//锁校验
private final static Object syncLock = new Object();
private static void config(HttpRequestBase httpRequestBase) {
// // 设置Header等
// httpRequestBase.setHeader("User-Agent", "Mozilla/5.0");
// httpRequestBase
// .setHeader("Accept",
// "text/html,application/xhtml+xml,application/xml;q=0.9,*/*;q=0.8");
// httpRequestBase.setHeader("Accept-Language",
// "zh-CN,zh;q=0.8,en-US;q=0.5,en;q=0.3");// "en-US,en;q=0.5");
// httpRequestBase.setHeader("Accept-Charset",
// "ISO-8859-1,utf-8,gbk,gb2312;q=0.7,*;q=0.7");
// 配置请求的超时设置
RequestConfig requestConfig = RequestConfig.custom()
.setConnectionRequestTimeout(timeOut)
.setConnectTimeout(timeOut).setSocketTimeout(timeOut).build();
httpRequestBase.setConfig(requestConfig);
}
/**
* 获取HttpClient对象
*
* @return
* @author zyz
* @create
*/
public static CloseableHttpClient getHttpClient(String url) {
String hostname = url.split("/")[2];
int port = 80;
if (hostname.contains(":")) {
String[] arr = hostname.split(":");
hostname = arr[0];
port = Integer.parseInt(arr[1]);
}
if (httpClient == null) {
synchronized (syncLock) {
if (httpClient == null) {
System.out.println("创建httpclient"+System.currentTimeMillis());
httpClient = createHttpClient(40, 40, 100, hostname, port);
}
}
}
return httpClient;
}
/**
* 创建httpclient对象
* @param maxTotal 最大连接
* @param maxPerRoute 每个路由最大连接数
* @param maxRoute
* @param hostname
* @param port
* @return
*/
public static CloseableHttpClient createHttpClient(int maxTotal,
int maxPerRoute, int maxRoute, String hostname, int port) {
// ConnectionSocketFactory plainsf = PlainConnectionSocketFactory
// .getSocketFactory();
// LayeredConnectionSocketFactory sslsf = SSLConnectionSocketFactory
// .getSocketFactory();
// Registry<ConnectionSocketFactory> registry = RegistryBuilder
// .<ConnectionSocketFactory> create().register("http", plainsf)
// .register("https", sslsf).build();
//长连接保持30秒
PoolingHttpClientConnectionManager cm = new PoolingHttpClientConnectionManager(30, TimeUnit.SECONDS);
// 将最大连接数增加
cm.setMaxTotal(maxTotal);
// 将每个路由基础的连接增加
cm.setDefaultMaxPerRoute(maxPerRoute);
HttpHost httpHost = new HttpHost(hostname, port);
// 将目标主机的最大连接数增加
cm.setMaxPerRoute(new HttpRoute(httpHost), maxRoute);
//关闭无效链接
cm.closeExpiredConnections();
// 请求重试处理
HttpRequestRetryHandler httpRequestRetryHandler = new HttpRequestRetryHandler() {
public boolean retryRequest(IOException exception,
int executionCount, HttpContext context) {
if (executionCount >= 5) {// 如果已经重试了5次，就放弃
return false;
}
if (exception instanceof NoHttpResponseException) {// 如果服务器丢掉了连接，那么就重试
return true;
}
if (exception instanceof SSLHandshakeException) {// 不要重试SSL握手异常
return false;
}
if (exception instanceof InterruptedIOException) {// 超时
return false;
}
if (exception instanceof UnknownHostException) {// 目标服务器不可达
return false;
}
if (exception instanceof ConnectTimeoutException) {// 连接被拒绝
return false;
}
if (exception instanceof SSLException) {// SSL握手异常
return false;
}
HttpClientContext clientContext = HttpClientContext
.adapt(context);
HttpRequest request = clientContext.getRequest();
// 如果请求是幂等的，就再次尝试
if (!(request instanceof HttpEntityEnclosingRequest)) {
return true;
}
return false;
}
};
//构建httpclient
CloseableHttpClient httpClient = HttpClients.custom()
.setConnectionManager(cm)
.setRetryHandler(httpRequestRetryHandler)
//保持长连接
.setKeepAliveStrategy(new DefaultConnectionKeepAliveStrategy()).build();
return httpClient;
}
/**
* 设置post请求参数
* @param httpost
* @param params 参数
*/
// private static void setPostParams(HttpPost httpost,
// Map<String, Object> params) {
// List<NameValuePair> nvps = new ArrayList<NameValuePair>();
// NameValuePair pair1 = new BasicNameValuePair("Content_type","application/json");
// nvps.add(pair1);
// Set<String> keySet = params.keySet();
// for (String key : keySet) {
// nvps.add(new BasicNameValuePair(key, params.get(key).toString()));
// }
// try {
// httpost.setEntity(new UrlEncodedFormEntity(nvps, "UTF-8"));
// } catch (UnsupportedEncodingException e) {
// e.printStackTrace();
// }
// }
private static void setPostParams(HttpPost httpost,Map<String, Object> params,String contentType) {
try {
HttpEntity httpEntity= new StringEntity(JSONObject.toJSONString(params),contentType,"UTF-8");
httpost.setEntity(httpEntity);
} catch (Exception e) {
}
}
/**
* post 获取内容
* @param url
* @param params
* @return
* @throws Exception
*/
public static String post(String url, Map<String, Object> params) throws Exception {
HttpPost httppost = new HttpPost(url);
config(httppost);
httppost.setHeader("Content_type","application/json");
setPostParams(httppost, params,"application/json");
CloseableHttpResponse closeableHttpResponse = null;
try {
closeableHttpResponse = getHttpClient(url).execute(httppost,
HttpClientContext.create());
// int statusCode = closeableHttpResponse.getStatusLine().getStatusCode();
HttpEntity entity = closeableHttpResponse.getEntity();
String result = EntityUtils.toString(entity, "utf-8");
EntityUtils.consume(entity);
return result;
} catch (Exception e) {
throw e;
} finally {
try {
if (closeableHttpResponse != null)
closeableHttpResponse.close();
httppost.releaseConnection();
} catch (IOException e) {
e.printStackTrace();
}
}
}
/**
* get 请求url 获取数据
* @param url
* @return
*/
public static String get(String url) {
HttpGet httpget = new HttpGet(url);
config(httpget);
CloseableHttpResponse response = null;
try {
response = getHttpClient(url).execute(httpget,
HttpClientContext.create());
HttpEntity entity = response.getEntity();
String result = EntityUtils.toString(entity, "utf-8");
EntityUtils.consume(entity);
return result;
} catch (IOException e) {
e.printStackTrace();
} finally {
try {
if (response != null)
response.close();
httpget.releaseConnection();
} catch (IOException e) {
e.printStackTrace();
}
}
return null;
}
public static void main(String[] args) throws Exception{
Map<String,Object> map = new HashMap<String,Object>();
map.put("__query_select_id","fam.CarpayMapper10.33.selectCarPayNotBalancedDetail");
map.put("d1","2017-12-01 23:59:59");
map.put("d2","2017-12-02 23:59:59");
map.put("page",1);
map.put("pageSize",20);
String url ="http://bug-dus.kxtx.cn/kxtx-dus/dus/controller/queryjson";
// // URL列表数组
// for (int i = 0; i <1000 ; i++) {
// String string =post("http://bug-dus.kxtx.cn/kxtx-dus/dus/controller/queryjson",map);
// System.out.println(string);
// }
long start = System.currentTimeMillis();
try {
ExecutorService executors = Executors.newFixedThreadPool(1000);
CountDownLatch countDownLatch = new CountDownLatch(1000);
for (int i = 0; i <1000 ; i++) {
executors.execute(new GetRunnable(url,countDownLatch,map));
}
countDownLatch.await();
executors.shutdown();
} catch (InterruptedException e) {
e.printStackTrace();
} finally {
System.out.println("线程" + Thread.currentThread().getName() + ","
+ System.currentTimeMillis() + ", 所有线程已完成，开始进入下一步！");
}
long end = System.currentTimeMillis();
System.out.println("consume -> " + (end - start));
}
static class GetRunnable implements Runnable {
private CountDownLatch countDownLatch;
private String url;
private Map<String,Object> map;
public GetRunnable(String url, CountDownLatch countDownLatch,Map<String,Object> map) {
this.url = url;
this.countDownLatch = countDownLatch;
this.map = map;
}
public void run() {
try {
String result = HttpClientUtil.post(url,map);
JSONObject obj = JSONObject.parseObject(result);
String data = obj.getString("data");
System.out.println("result="+result);
System.out.println("data="+data);
}catch (Exception e){
}finally {
countDownLatch.countDown();
}
}
}
}
版权声明：本文为zyzcj原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。