HttpClientUtil

来源：互联网发布：sql语句自动优化工具编辑：程序博客网时间：2024/06/05 02:32
package com.uniubi.management.util;import java.io.IOException;import java.io.InterruptedIOException;import java.io.UnsupportedEncodingException;import java.net.UnknownHostException;import java.util.ArrayList;import java.util.Iterator;import java.util.List;import java.util.Map;import java.util.Map.Entry;import java.util.Random;import java.util.concurrent.CountDownLatch;import java.util.concurrent.ExecutorService;import java.util.concurrent.Executors;import javax.net.ssl.SSLException;import javax.net.ssl.SSLHandshakeException;import org.apache.http.HttpEntity;import org.apache.http.HttpEntityEnclosingRequest;import org.apache.http.HttpHost;import org.apache.http.HttpRequest;import org.apache.http.HttpStatus;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.config.Registry;import org.apache.http.config.RegistryBuilder;import org.apache.http.conn.ConnectTimeoutException;import org.apache.http.conn.routing.HttpRoute;import org.apache.http.conn.socket.ConnectionSocketFactory;import org.apache.http.conn.socket.LayeredConnectionSocketFactory;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.message.BasicNameValuePair;import org.apache.http.protocol.HttpContext;import org.apache.http.util.EntityUtils;import com.alibaba.fastjson.JSON;import com.uniubi.management.model.ResultWrapper;public class HttpClientUtil {        static class HttpResult{        public String text = "";        public HttpResult(String text) {            super();            this.text = text;        }    }    private static final int THREAD_COUNT = 10;    private static CloseableHttpClient httpClient;    private static ExecutorService executors;    private static PoolingHttpClientConnectionManager cm;    static{        executors = Executors.newFixedThreadPool(THREAD_COUNT);                ConnectionSocketFactory plainsf = PlainConnectionSocketFactory.getSocketFactory();        LayeredConnectionSocketFactory sslsf = SSLConnectionSocketFactory.getSocketFactory();        Registry<ConnectionSocketFactory> registry = RegistryBuilder.<ConnectionSocketFactory>create()                .register("http", plainsf)                .register("https", sslsf)                .build();        cm = new PoolingHttpClientConnectionManager(registry);        // 将最大连接数增加到200        cm.setMaxTotal(200);        // 将每个路由基础的连接增加到20        cm.setDefaultMaxPerRoute(20);        // 将目标主机的最大连接数增加到50        HttpHost localhost = new HttpHost(WebService.WS_ADDRESS);        cm.setMaxPerRoute(new HttpRoute(localhost), 50);              //请求重试处理        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 = HttpClients.custom()                .setConnectionManager(cm)                .setRetryHandler(httpRequestRetryHandler)                .build();    }        public static void release() {          if (executors != null) {              executors.shutdown();            executors = null;        }          if (cm != null) {              cm.shutdown();            cm = null;        }      }        private static void config(HttpRequestBase httpRequestBase) {        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(3000)                .setConnectTimeout(3000)                .setSocketTimeout(3000)                .build();        httpRequestBase.setConfig(requestConfig);            }        private static String executeMethod(HttpRequestBase httpget) {//, CountDownLatch countDownLatch        CloseableHttpResponse response = null;        String result = "";        try {            response = httpClient.execute(httpget,HttpClientContext.create());            HttpEntity entity = response.getEntity();            result = EntityUtils.toString(entity, "utf-8");            EntityUtils.consume(entity);        } catch (IOException e) {            e.printStackTrace();        } finally {//            countDownLatch.countDown();            try {                if(response != null)                    response.close();            } catch (IOException e) {                e.printStackTrace();            }        }        return result;    }        public static String doGet(String url){        return doGet(url,null);    }    public static String doGet(String url,String param) {        HttpResult result = new HttpResult("");        try {            CountDownLatch countDownLatch = new CountDownLatch(1);            url = url + "?apiKey="+WebService.WS_APIKEY;            if(!StringUtil.isEmpty(param)){                url = url + "&" + param;            }            HttpGet request = new HttpGet(url);            config(request);            //启动线程抓取            executors.execute(new GetRunnable(result,request,countDownLatch));            countDownLatch.await();        } catch (InterruptedException e) {            e.printStackTrace();        } finally {//            System.out.println("线程" + Thread.currentThread().getName() + "," + System.currentTimeMillis() + ", 所有线程已完成，开始进入下一步！");        }        return result.text;    }        public static String doPost(String url) {        return doPost(url,null);    }    public static String doPost(String url,Map<String,String> param) {//        HttpResult result = new HttpResult("");        String responeText = "";        try {//            CountDownLatch countDownLatch = new CountDownLatch(1);                     HttpPost request = new HttpPost(url);            config(request);            List<NameValuePair> nvps = new ArrayList<NameValuePair>();            //增加apikey            nvps.add(new BasicNameValuePair("apiKey", WebService.WS_APIKEY));                          if(!param.isEmpty()){                Iterator<Entry<String, String>> iter = param.entrySet().iterator();                if(iter.hasNext()){                    Entry<String, String> next = iter.next();                    nvps.add(new BasicNameValuePair(next.getKey(), next.getValue()));                  }            }            request.setEntity(new UrlEncodedFormEntity(nvps));              //启动线程抓取//            executors.execute(new GetRunnable(result,request,countDownLatch));            responeText = executeMethod(request);//            countDownLatch.await();        } catch (UnsupportedEncodingException e) {            e.printStackTrace();        } finally {//            System.out.println("线程" + Thread.currentThread().getName() + "," + System.currentTimeMillis() + ", 所有线程已完成，开始进入下一步！");        }//        return result.text;        return responeText;    }    private static class GetRunnable implements Runnable {        private CountDownLatch countDownLatch;        private final HttpRequestBase request;        private HttpResult result;                 public GetRunnable(HttpResult result, HttpRequestBase request, CountDownLatch countDownLatch){            this.result = result;            this.request = request;                        this.countDownLatch = countDownLatch;        }        @Override        public void run() {            CloseableHttpResponse response = null;            try {                response = httpClient.execute(request,HttpClientContext.create());                if(response.getStatusLine().getStatusCode() == HttpStatus.SC_OK){                    HttpEntity entity = response.getEntity();                    EntityUtils.consume(entity);                    result.text = EntityUtils.toString(entity, "utf-8");                }                else{                    result.text = JSON.toJSONString(ResultWrapper.wrapFailure());                }            } catch (IOException e) {                e.printStackTrace();            } finally {                countDownLatch.countDown();                                 try {                    if(response != null)                        response.close();                } catch (IOException e) {                    e.printStackTrace();                }            }        }    }        public static void main(String[] args) {//        String url = WebService.USER_GET_INFO;//        String result = doGet(url,"id=2");//        System.out.println(result);                Random r = new Random();        for (int i = 0; i < 10; i++) {            long l1 = System.currentTimeMillis();            String resultText = doGet("http://www.baidu.com/s?wd=" + r.nextInt(5000));            System.out.println("查询耗时" + (System.currentTimeMillis() - l1));        }    }}
0 0