利用Apache的HttpClient專案建立的HttpUtils工具類
阿新 • • 發佈：2020-09-15
網上這個工具類也有一些，但是很多都是基於HttpClient4.5之前寫的，看著SSL連線那一塊程式碼那麼多劃線，雖然不算報錯，但是看著也是挺不舒服的
所以經過一番查詢，找到一個修改的比較好的例子，同時將裡面的httpclient客戶端弄成了連線池型別，增加了超時限制，裡面使用了單例模式，可以在併發情況下使用，直接貼程式碼了，然後將幾個參考的連線加上去
import java.io.IOException; import java.io.UnsupportedEncodingException; import java.net.URLEncoder; import java.security.KeyManagementException; importjava.security.NoSuchAlgorithmException; import java.security.cert.X509Certificate; import java.util.ArrayList; import java.util.Arrays; import java.util.List; import java.util.Map; import javax.net.ssl.SSLContext; import javax.net.ssl.TrustManager; import javax.net.ssl.X509TrustManager; importorg.apache.commons.lang.StringUtils; import org.apache.http.HttpEntity; import org.apache.http.HttpResponse; import org.apache.http.NameValuePair; import org.apache.http.client.HttpClient; import org.apache.http.client.config.AuthSchemes; import org.apache.http.client.config.CookieSpecs; importorg.apache.http.client.config.RequestConfig; import org.apache.http.client.entity.UrlEncodedFormEntity; import org.apache.http.client.methods.HttpDelete; import org.apache.http.client.methods.HttpGet; import org.apache.http.client.methods.HttpPost; import org.apache.http.client.methods.HttpPut; 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.entity.ByteArrayEntity; import org.apache.http.entity.StringEntity; import org.apache.http.impl.client.CloseableHttpClient; import org.apache.http.impl.client.HttpClientBuilder; import org.apache.http.impl.client.HttpClients; import org.apache.http.impl.conn.PoolingHttpClientConnectionManager; import org.apache.http.message.BasicNameValuePair; import org.apache.http.util.EntityUtils; import com.alibaba.fastjson.JSON; import com.alibaba.fastjson.JSONObject; /** * http請求工具類 * github地址：https://github.com/zhuzhegithub/utils * @author zhuzhe * @date 2018/5/3 11:46 */ public class HttpUtils { /** * 普通http連線的連線建立類 */ private static HttpClientBuilder httpClientBuilder; /** * ssl連線的連線建立類 */ private static HttpClientBuilder sslHttpClientBuilder; /** * get * * @param host * @param path * * @param headers * @param querys * @return * @throws Exception */ public static HttpResponse doGet(String host, String path, Map<String, String> headers, Map<String, String> querys) throws Exception { HttpClient httpClient = wrapClient(host, path); HttpGet request = new HttpGet(buildUrl(host, path, querys)); for (Map.Entry<String, String> e : headers.entrySet()) { request.addHeader(e.getKey(), e.getValue()); } return httpClient.execute(request); } /** * post form * * @param host * @param path * * @param headers * @param querys * @param bodys * @return * @throws Exception */ public static HttpResponse doPost(String host, String path, Map<String, String> headers, Map<String, String> querys, Map<String, String> bodys) throws Exception { HttpClient httpClient = wrapClient(host, path); HttpPost request = new HttpPost(buildUrl(host, path, querys)); for (Map.Entry<String, String> e : headers.entrySet()) { request.addHeader(e.getKey(), e.getValue()); } if (bodys != null) { List<NameValuePair> nameValuePairList = new ArrayList<NameValuePair>(); for (String key : bodys.keySet()) { nameValuePairList.add(new BasicNameValuePair(key, bodys.get(key))); } UrlEncodedFormEntity formEntity = new UrlEncodedFormEntity(nameValuePairList, "utf-8"); formEntity.setContentType("application/x-www-form-urlencoded; charset=UTF-8"); request.setEntity(formEntity); } return httpClient.execute(request); } /** * Post String * * @param host * @param path * * @param headers * @param querys * @param body * @return * @throws Exception */ public static HttpResponse doPost(String host, String path, Map<String, String> headers, Map<String, String> querys, String body) throws Exception { HttpClient httpClient = wrapClient(host, path); HttpPost request = new HttpPost(buildUrl(host, path, querys)); for (Map.Entry<String, String> e : headers.entrySet()) { request.addHeader(e.getKey(), e.getValue()); } if (StringUtils.isNotBlank(body)) { request.setEntity(new StringEntity(body, "utf-8")); } return httpClient.execute(request); } /** * Post stream * * @param host * @param path * * @param headers * @param querys * @param body * @return * @throws Exception */ public static HttpResponse doPost(String host, String path, Map<String, String> headers, Map<String, String> querys, byte[] body) throws Exception { HttpClient httpClient = wrapClient(host, path); HttpPost request = new HttpPost(buildUrl(host, path, querys)); for (Map.Entry<String, String> e : headers.entrySet()) { request.addHeader(e.getKey(), e.getValue()); } if (body != null) { request.setEntity(new ByteArrayEntity(body)); } return httpClient.execute(request); } /** * Put String * * @param host * @param path * * @param headers * @param querys * @param body * @return * @throws Exception */ public static HttpResponse doPut(String host, String path, Map<String, String> headers, Map<String, String> querys, String body) throws Exception { HttpClient httpClient = wrapClient(host, path); HttpPut request = new HttpPut(buildUrl(host, path, querys)); for (Map.Entry<String, String> e : headers.entrySet()) { request.addHeader(e.getKey(), e.getValue()); } if (StringUtils.isNotBlank(body)) { request.setEntity(new StringEntity(body, "utf-8")); } return httpClient.execute(request); } /** * Put stream * * @param host * @param path * * @param headers * @param querys * @param body * @return * @throws Exception */ public static HttpResponse doPut(String host, String path, Map<String, String> headers, Map<String, String> querys, byte[] body) throws Exception { HttpClient httpClient = wrapClient(host, path); HttpPut request = new HttpPut(buildUrl(host, path, querys)); for (Map.Entry<String, String> e : headers.entrySet()) { request.addHeader(e.getKey(), e.getValue()); } if (body != null) { request.setEntity(new ByteArrayEntity(body)); } return httpClient.execute(request); } /** * Delete * * @param host * @param path * * @param headers * @param querys * @return * @throws Exception */ public static HttpResponse doDelete(String host, String path, Map<String, String> headers, Map<String, String> querys) throws Exception { HttpClient httpClient = wrapClient(host, path); HttpDelete request = new HttpDelete(buildUrl(host, path, querys)); for (Map.Entry<String, String> e : headers.entrySet()) { request.addHeader(e.getKey(), e.getValue()); } return httpClient.execute(request); } /** * 構建請求的 url * * @param host * @param path * @param querys * @return * @throws UnsupportedEncodingException */ private static String buildUrl(String host, String path, Map<String, String> querys) throws UnsupportedEncodingException { StringBuilder sbUrl = new StringBuilder(); if (!StringUtils.isBlank(host)) { sbUrl.append(host); } if (!StringUtils.isBlank(path)) { sbUrl.append(path); } if (null != querys) { StringBuilder sbQuery = new StringBuilder(); for (Map.Entry<String, String> query : querys.entrySet()) { if (0 < sbQuery.length()) { sbQuery.append("&"); } if (StringUtils.isBlank(query.getKey()) && !StringUtils.isBlank(query.getValue())) { sbQuery.append(query.getValue()); } if (!StringUtils.isBlank(query.getKey())) { sbQuery.append(query.getKey()); if (!StringUtils.isBlank(query.getValue())) { sbQuery.append("="); sbQuery.append(URLEncoder.encode(query.getValue(), "utf-8")); } } } if (0 < sbQuery.length()) { sbUrl.append("?").append(sbQuery); } } return sbUrl.toString(); } /** * 獲取 HttpClient * * @param host * @param path * @return */ private static HttpClient wrapClient(String host, String path) { if (host != null && host.startsWith("https://")) { return sslClient(); } else if (StringUtils.isBlank(host) && path != null && path.startsWith("https://")) { return sslClient(); } else {//非ssl連線 if(null == httpClientBuilder) {//單例模式 synchronized (HttpUtils.class) { if(null == httpClientBuilder) { //增加httpclient的超時時間 RequestConfig requestConfig = RequestConfig.custom().setConnectTimeout(500)// 連線超時時間 .setSocketTimeout(5000)// 請求獲取資料的超時時間 .build(); httpClientBuilder = HttpClientBuilder.create().setMaxConnTotal(HttpConfig.maxConnTotal) // 連線池中最大連線數 /** * 分配給同一個route(路由)最大的併發連線數。 route：執行環境機器 到 目標機器的一條線路。 * 舉例來說，我們使用HttpClient的實現來分別請求 www.baidu.com 的資源和 www.bing.com * 的資源那麼他就會產生兩個route。 */ // .setMaxConnPerRoute(config.maxConnPerRoute) .setDefaultRequestConfig(requestConfig); } } } return httpClientBuilder.build(); } } /** * 在呼叫SSL之前需要重寫驗證方法，取消檢測SSL 建立ConnectionManager，新增Connection配置資訊 * * @return HttpClient 支援https */ private static HttpClient sslClient() { try { if(null == sslHttpClientBuilder) { synchronized (HttpUtils.class) { if(null == sslHttpClientBuilder) { // 在呼叫SSL之前需要重寫驗證方法，取消檢測SSL X509TrustManager trustManager = new X509TrustManager() { @Override public X509Certificate[] getAcceptedIssuers() { return null; } @Override public void checkClientTrusted(X509Certificate[] xcs, String str) { } @Override public void checkServerTrusted(X509Certificate[] xcs, String str) { } }; SSLContext ctx = SSLContext.getInstance(SSLConnectionSocketFactory.TLS); ctx.init(null, new TrustManager[] { trustManager }, null); SSLConnectionSocketFactory socketFactory = new SSLConnectionSocketFactory(ctx, NoopHostnameVerifier.INSTANCE); // 建立Registry RequestConfig requestConfig = RequestConfig.custom().setCookieSpec(CookieSpecs.STANDARD_STRICT) .setConnectTimeout(500)// 連線超時時間 .setSocketTimeout(5000)// 請求獲取資料的超時時間 .setExpectContinueEnabled(Boolean.TRUE) .setTargetPreferredAuthSchemes(Arrays.asList(AuthSchemes.NTLM, AuthSchemes.DIGEST)) .setProxyPreferredAuthSchemes(Arrays.asList(AuthSchemes.BASIC)).build(); Registry<ConnectionSocketFactory> socketFactoryRegistry = RegistryBuilder.<ConnectionSocketFactory>create() .register("http", PlainConnectionSocketFactory.INSTANCE).register("https", socketFactory).build(); // 建立ConnectionManager，新增Connection配置資訊 PoolingHttpClientConnectionManager connectionManager = new PoolingHttpClientConnectionManager( socketFactoryRegistry); connectionManager.setMaxTotal(HttpConfig.maxConnTotal); sslHttpClientBuilder = HttpClientBuilder.create().setConnectionManager(connectionManager) .setDefaultRequestConfig(requestConfig); } } } CloseableHttpClient closeableHttpClient = sslHttpClientBuilder.build(); return closeableHttpClient; } catch (KeyManagementException ex) { throw new RuntimeException(ex); } catch (NoSuchAlgorithmException ex) { throw new RuntimeException(ex); } } /** * 將結果轉換成JSONObject * * @param httpResponse * @return * @throws IOException */ public static JSONObject getJson(HttpResponse httpResponse) throws IOException { HttpEntity entity = httpResponse.getEntity(); String resp = EntityUtils.toString(entity, "UTF-8"); EntityUtils.consume(entity); return JSON.parseObject(resp); } }