摘自爬虫类 用于频繁请求减少网络消耗
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(); } } } }