Java HttpClientUtil 请求工具类

Jokes-T
已于 2024-09-01 11:43:42 修改
阅读量52
点赞数
文章标签： java spring
于 2023-02-28 11:23:25 首次发布
本文链接：https://blog.csdn.net/qq_44785726/article/details/129258104
版权

import org.apache.http.HttpEntity;
import org.apache.http.HttpEntityEnclosingRequest;
import org.apache.http.HttpRequest;
import org.apache.http.NameValuePair;
import org.apache.http.client.ClientProtocolException;
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.*;
import org.apache.http.client.protocol.HttpClientContext;
import org.apache.http.client.utils.URIBuilder;
import org.apache.http.conn.ConnectTimeoutException;
import org.apache.http.entity.ByteArrayEntity;
import org.apache.http.entity.ContentType;
import org.apache.http.entity.mime.MultipartEntityBuilder;
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 org.springframework.web.multipart.MultipartFile;

import javax.net.ssl.SSLException;
import java.io.IOException;
import java.io.InterruptedIOException;
import java.io.UnsupportedEncodingException;
import java.net.URISyntaxException;
import java.net.UnknownHostException;
import java.util.ArrayList;
import java.util.Map;

public class HttpClientUtil {

    private static PoolingHttpClientConnectionManager cm;
    @SuppressWarnings("unused")
    private static String EMPTY_STR = "";
    private static final int DEFAULT_TIME_OUT = 300000; // 5分钟
    public static final String UTF_8 = "UTF-8";

    private static void init() {
        if (cm == null) {
            cm = new PoolingHttpClientConnectionManager();
            cm.setMaxTotal(50);// 整个连接池最大连接数
            cm.setDefaultMaxPerRoute(30);// 每路由最大连接数，默认值是2
        }
    }

    /**
     * 通过连接池获取HttpClient
     *
     * @return
     */
    public static CloseableHttpClient getHttpClient() {
        init();
        // 自定义请求重试HANDLER
        HttpRequestRetryHandler customRetryHandler = new HttpRequestRetryHandler() {
            @Override
            public boolean retryRequest(IOException exception,
                                        int executionCount, HttpContext context) {
                if (executionCount >= 5) {
                    // Do not retry if over max retry count
                    return false;
                }
                if (exception instanceof InterruptedIOException) {
                    // Timeout
                    return false;
                }
                if (exception instanceof UnknownHostException) {
                    // Unknown host
                    return false;
                }
                if (exception instanceof ConnectTimeoutException) {
                    // Connection refused
                    return false;
                }
                if (exception instanceof SSLException) {
                    // SSL handshake exception
                    return false;
                }
                HttpClientContext clientContext = HttpClientContext
                        .adapt(context);
                HttpRequest request = clientContext.getRequest();
                boolean idempotent = !(request instanceof HttpEntityEnclosingRequest);
                if (idempotent) {
                    // Retry if the request is considered idempotent
                    return true;
                }
                return false;
            }
        };
        return HttpClients.custom().setConnectionManager(cm)
                .setRetryHandler(customRetryHandler).build();
    }

    /**
     * @param url
     * @return
     */
    public static String httpGetRequest(String url) {
        HttpGet httpGet = new HttpGet(url);
        return getResult(httpGet);
    }

    public static String httpGetRequest(String url, Map<String, Object> params)
            throws URISyntaxException {
        URIBuilder ub = new URIBuilder();
        ub.setPath(url);

        ArrayList<NameValuePair> pairs = covertParams2NVPS(params);
        ub.setParameters(pairs);

        HttpGet httpGet = new HttpGet(ub.build());
        return getResult(httpGet);
    }

    public static byte[] httpGetRequestToByteArray(String url,
                                                   Map<String, Object> params) throws URISyntaxException {
        URIBuilder ub = new URIBuilder();
        ub.setPath(url);
        ArrayList<NameValuePair> pairs = covertParams2NVPS(params);
        ub.setParameters(pairs);

        HttpGet httpGet = new HttpGet(ub.build());
        return getResultToByteArray(httpGet);
    }

    public static byte[] httpGetRequestToByteArray(String url,
                                                   Map<String, Object> headers, Map<String, Object> params)
            throws URISyntaxException {
        URIBuilder ub = new URIBuilder();
        ub.setPath(url);
        if (params != null) {
            ArrayList<NameValuePair> pairs = covertParams2NVPS(params);
            ub.setParameters(pairs);
        }
        HttpGet httpGet = new HttpGet(ub.build());
        if (headers != null) {
            for (Map.Entry<String, Object> param : headers.entrySet()) {
                httpGet.addHeader(param.getKey(),
                        String.valueOf(param.getValue()));
            }
        }
        return getResultToByteArray(httpGet);
    }

    public static String httpGetRequest(String url,
                                        Map<String, Object> headers, Map<String, Object> params)
            throws URISyntaxException {
        URIBuilder ub = new URIBuilder();
        ub.setPath(url);
        if (params != null) {
            ArrayList<NameValuePair> pairs = covertParams2NVPS(params);
            ub.setParameters(pairs);
        }
        HttpGet httpGet = new HttpGet(ub.build());
        if (headers != null) {
            for (Map.Entry<String, Object> param : headers.entrySet()) {
                httpGet.addHeader(param.getKey(),
                        String.valueOf(param.getValue()));
            }
        }

        return getResult(httpGet);
    }

    public static String httpPostRequest(String url) {
        HttpPost httpPost = new HttpPost(url);
        return getResult(httpPost);
    }

    public static String httpPostRequest(String url, Map<String, Object> params)
            throws UnsupportedEncodingException {
        HttpPost httpPost = new HttpPost(url);
        ArrayList<NameValuePair> pairs = covertParams2NVPS(params);
        httpPost.setEntity(new UrlEncodedFormEntity(pairs, UTF_8));
        return getResult(httpPost);
    }

    public static String httpPostRequest(String url,
                                         Map<String, Object> headers, Map<String, Object> params)
            throws UnsupportedEncodingException {
        HttpPost httpPost = new HttpPost(url);

        for (Map.Entry<String, Object> param : headers.entrySet()) {
            httpPost.addHeader(param.getKey(), String.valueOf(param.getValue()));
        }

        ArrayList<NameValuePair> pairs = covertParams2NVPS(params);
        httpPost.setEntity(new UrlEncodedFormEntity(pairs, UTF_8));
        return getResult(httpPost);
    }

    /**
     * request-body方式是在request-body中提供参数，此方式只能用于进行POST请求。
     * 在HttpClient程序包中有两个类可以完成此项工作
     * ，它们分别是UrlEncodedFormEntity类与MultipartEntity,ByteArrayEntity类。这
     * 两个类均实现了HttpEntity接口
     *
     * @param url
     * @param headers
     * @param requestBody
     * @return
     * @throws UnsupportedEncodingException
     */
    public static String httpPostRequest(String url,
                                         Map<String, Object> headers, String requestBody)
            throws UnsupportedEncodingException {
        HttpPost httpPost = new HttpPost(url);
        for (Map.Entry<String, Object> param : headers.entrySet()) {
            httpPost.addHeader(param.getKey(), String.valueOf(param.getValue()));
        }
        HttpEntity entity = new ByteArrayEntity(requestBody.getBytes(UTF_8));
        httpPost.setEntity(entity);
        return getResult(httpPost);
    }

    public static String httpPutRequest(String url,
                                         Map<String, Object> headers, String requestBody)
            throws UnsupportedEncodingException {
        HttpPut httpPut = new HttpPut(url);
        for (Map.Entry<String, Object> param : headers.entrySet()) {
            httpPut.addHeader(param.getKey(), String.valueOf(param.getValue()));
        }
        HttpEntity entity = new ByteArrayEntity(requestBody.getBytes(UTF_8));
        httpPut.setEntity(entity);
        return getResult(httpPut);
    }

    public static String httpPostRequest(String url,
                                         Map<String, Object> headers, MultipartFile file)
            throws UnsupportedEncodingException {
        HttpPost httpPost = new HttpPost(url);
        for (Map.Entry<String, Object> param : headers.entrySet()) {
            httpPost.addHeader(param.getKey(), String.valueOf(param.getValue()));
        }
        MultipartEntityBuilder entity = MultipartEntityBuilder.create();
        try {
            entity.addBinaryBody("file", file.getInputStream(),
                    ContentType.MULTIPART_FORM_DATA, file.getOriginalFilename());
        } catch (IOException e) {
            e.printStackTrace();
        }
        httpPost.setEntity(entity.build());
        return getResult(httpPost);
    }

    public static byte[] httpPostRequestToByteArray(String url,
                                                    Map<String, Object> headers, String requestBody)
            throws UnsupportedEncodingException {
        HttpPost httpPost = new HttpPost(url);
        for (Map.Entry<String, Object> param : headers.entrySet()) {
            httpPost.addHeader(param.getKey(), String.valueOf(param.getValue()));
        }
        HttpEntity entity = new ByteArrayEntity(requestBody.getBytes(UTF_8));
        httpPost.setEntity(entity);
        return getResultToByteArray(httpPost);
    }

    /**
     * request-body方式是在request-body中提供参数，此方式只能用于进行POST请求。
     * 在HttpClient程序包中有两个类可以完成此项工作
     * ，它们分别是UrlEncodedFormEntity类与MultipartEntity,ByteArrayEntity类。这
     * 两个类均实现了HttpEntity接口
     *
     * @param url
     * @param requestBody
     * @return
     * @throws UnsupportedEncodingException
     */
    public static String httpPostRequest(String url, String requestBody)
            throws UnsupportedEncodingException {
        HttpPost httpPost = new HttpPost(url);
        HttpEntity entity = new ByteArrayEntity(requestBody.getBytes(UTF_8));
        httpPost.setEntity(entity);
        return getResult(httpPost);
    }

    private static ArrayList<NameValuePair> covertParams2NVPS(
            Map<String, Object> params) {
        ArrayList<NameValuePair> pairs = new ArrayList<NameValuePair>();
        for (Map.Entry<String, Object> param : params.entrySet()) {
            pairs.add(new BasicNameValuePair(param.getKey(), String
                    .valueOf(param.getValue())));
        }
        return pairs;
    }

    /**
     * 处理Http请求
     *
     * @param request
     * @return
     */
    public static String getResult(HttpRequestBase request) {
        // CloseableHttpClient httpClient = HttpClients.createDefault();
        CloseableHttpClient httpClient = getHttpClient();
        CloseableHttpResponse response = null;
        try {
            request.setConfig(RequestConfig.custom()
                    .setConnectTimeout(DEFAULT_TIME_OUT)
                    .setConnectionRequestTimeout(DEFAULT_TIME_OUT)
                    .setSocketTimeout(DEFAULT_TIME_OUT).build()); // 设置超时时间
            response = httpClient.execute(request);
            // response.getStatusLine().getStatusCode();
            HttpEntity entity = response.getEntity();
            String result = "";
            if (entity != null) {
                result = EntityUtils.toString(entity);
            }
            /*
             * if (response == null || response.getStatusLine().getStatusCode() != 200) {
			 * throw new RuntimeException("请求失败,原因:" + result); }
			 */
            return result;
        } catch (Exception e) {
            throw new RuntimeException(e);
        } finally {
            if (response != null) {
                try {
                    response.close();
                } catch (IOException e) {
                }
            }
            // if(httpClient != null){
            // try{
            // httpClient.close();
            // }catch(IOException e){
            // }
            // }
        }
    }

    /**
     * 处理Http请求
     *
     * @param request
     * @return
     */
    public static byte[] getResultToByteArray(HttpRequestBase request) {
        // CloseableHttpClient httpClient = HttpClients.createDefault();
        CloseableHttpClient httpClient = getHttpClient();
        try {
            CloseableHttpResponse response = httpClient.execute(request);
            // response.getStatusLine().getStatusCode();
            HttpEntity entity = response.getEntity();
            if (entity != null) {
                // long len = entity.getContentLength();// -1 表示长度未知
                byte[] result = EntityUtils.toByteArray(entity);
                response.close();
                // httpClient.close();
                return result;
            }
        } catch (ClientProtocolException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        } finally {
        }
        return null;
    }

}