1. 程式人生 > 實用技巧 >HTTP請求 工具類

HTTP請求 工具類

繞開SSL的http請求工具類:

package com.brave.solid.utils;

import org.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; import org.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 org.springframework.stereotype.Component; import javax.net.ssl.SSLContext; import javax.net.ssl.TrustManager; import javax.net.ssl.X509TrustManager; import java.io.IOException; import java.io.UnsupportedEncodingException; import java.net.URLEncoder; import java.security.KeyManagementException; import java.security.NoSuchAlgorithmException; import java.security.cert.X509Certificate; import java.util.ArrayList; import java.util.Arrays; import java.util.List; import java.util.Map; /** * http請求工具類 * * @author admin */ @Component public class HttpUtils { /** * get * * @param host * @param path * @param headers * @param queryMap * @return * @throws Exception */ public static HttpResponse doGet(String host, String path, Map<String, String> headers, Map<String, String> queryMap) throws Exception { HttpClient httpClient = wrapClient(host, path); HttpGet request = new HttpGet(buildUrl(host, path, queryMap)); for (Map.Entry<String, String> e : headers.entrySet()) { request.addHeader(e.getKey(), e.getValue()); } request.setConfig(setTimeOutConfig(request.getConfig())); HttpResponse httpResponse = httpClient.execute(request); return httpClient.execute(request); } /** * post form * * @param host * @param path * @param headers * @param queryMap * @param bodyMap * @return * @throws Exception */ public static HttpResponse doPost(String host, String path, Map<String, String> headers, Map<String, String> queryMap, Map<String, String> bodyMap) throws Exception { HttpClient httpClient = wrapClient(host, path); HttpPost request = new HttpPost(buildUrl(host, path, queryMap)); for (Map.Entry<String, String> e : headers.entrySet()) { request.addHeader(e.getKey(), e.getValue()); } request.setConfig(setTimeOutConfig(request.getConfig())); if (bodyMap != null && bodyMap.size() > 0) { List<NameValuePair> nameValuePairList = new ArrayList<NameValuePair>(); for (String key : bodyMap.keySet()) { nameValuePairList.add(new BasicNameValuePair(key, bodyMap.get(key))); } UrlEncodedFormEntity formEntity = new UrlEncodedFormEntity(nameValuePairList, "utf-8"); formEntity.setContentType("application/json"); request.setEntity(formEntity); } return httpClient.execute(request); } /** * Post String * * @param host * @param path * @param headers * @param queryMap * @param body * @return * @throws Exception */ public static HttpResponse doPost(String host, String path, Map<String, String> headers, Map<String, String> queryMap, String body) throws Exception { HttpClient httpClient = wrapClient(host, path); HttpPost request = new HttpPost(buildUrl(host, path, queryMap)); 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")); } request.setConfig(setTimeOutConfig(request.getConfig())); return httpClient.execute(request); } /** * Post stream * * @param host * @param path * @param headers * @param queryMap * @param body * @return * @throws Exception */ public static HttpResponse doPost(String host, String path, Map<String, String> headers, Map<String, String> queryMap, byte[] body) throws Exception { HttpClient httpClient = wrapClient(host, path); HttpPost request = new HttpPost(buildUrl(host, path, queryMap)); for (Map.Entry<String, String> e : headers.entrySet()) { request.addHeader(e.getKey(), e.getValue()); } if (body != null) { request.setEntity(new ByteArrayEntity(body)); } request.setConfig(setTimeOutConfig(request.getConfig())); return httpClient.execute(request); } /** * Put String * * @param host * @param path * @param headers * @param queryMap * @param body * @return * @throws Exception */ public static HttpResponse doPut(String host, String path, Map<String, String> headers, Map<String, String> queryMap, String body) throws Exception { HttpClient httpClient = wrapClient(host, path); HttpPut request = new HttpPut(buildUrl(host, path, queryMap)); 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")); } request.setConfig(setTimeOutConfig(request.getConfig())); return httpClient.execute(request); } /** * Put stream * * @param host * @param path * @param headers * @param queryMap * @param body * @return * @throws Exception */ public static HttpResponse doPut(String host, String path, Map<String, String> headers, Map<String, String> queryMap, byte[] body) throws Exception { HttpClient httpClient = wrapClient(host, path); HttpPut request = new HttpPut(buildUrl(host, path, queryMap)); for (Map.Entry<String, String> e : headers.entrySet()) { request.addHeader(e.getKey(), e.getValue()); } if (body != null) { request.setEntity(new ByteArrayEntity(body)); } request.setConfig(setTimeOutConfig(request.getConfig())); return httpClient.execute(request); } /** * Delete * * @param host * @param path * @param headers * @param queryMap * @return * @throws Exception */ public static HttpResponse doDelete(String host, String path, Map<String, String> headers, Map<String, String> queryMap) throws Exception { HttpClient httpClient = wrapClient(host, path); HttpDelete request = new HttpDelete(buildUrl(host, path, queryMap)); for (Map.Entry<String, String> e : headers.entrySet()) { request.addHeader(e.getKey(), e.getValue()); } request.setConfig(setTimeOutConfig(request.getConfig())); return httpClient.execute(request); } /** * 構建請求的 url * * @param host * @param path * @param queryMap * @return * @throws UnsupportedEncodingException */ private static String buildUrl(String host, String path, Map<String, String> queryMap) throws UnsupportedEncodingException { StringBuilder sbUrl = new StringBuilder(); if (!StringUtils.isBlank(host)) { sbUrl.append(host); } if (!StringUtils.isBlank(path)) { sbUrl.append(path); } if (null != queryMap) { StringBuilder sbQuery = new StringBuilder(); for (Map.Entry<String, String> query : queryMap.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) { HttpClient httpClient = HttpClientBuilder.create().build(); if (host != null && host.startsWith("https://")) { return sslClient(); } else if (StringUtils.isBlank(host) && path != null && path.startsWith("https://")) { return sslClient(); } return httpClient; } /** * 在呼叫SSL之前需要重寫驗證方法,取消檢測SSL * 建立ConnectionManager,新增Connection配置資訊 * * @return HttpClient 支援https */ private static HttpClient sslClient() { try { // 在呼叫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) .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); CloseableHttpClient closeableHttpClient = HttpClients.custom().setConnectionManager(connectionManager) .setDefaultRequestConfig(requestConfig).build(); return closeableHttpClient; } catch (KeyManagementException ex) { throw new RuntimeException(ex); } catch (NoSuchAlgorithmException ex) { throw new RuntimeException(ex); } } /** * 設定 連線超時、 請求超時 、 讀取超時 毫秒 * * @param requestConfig * @return */ private static RequestConfig setTimeOutConfig(RequestConfig requestConfig) { if (requestConfig == null) { requestConfig = RequestConfig.DEFAULT; } return RequestConfig.copy(requestConfig) .setConnectionRequestTimeout(60000) .setConnectTimeout(60000) .setSocketTimeout(10000) .build(); } /** * 將結果轉換成 String * * @param httpResponse * @return * @throws IOException */ public static String parseString(HttpResponse httpResponse) throws IOException { HttpEntity entity = httpResponse.getEntity(); String resp = EntityUtils.toString(entity, "UTF-8"); EntityUtils.consume(entity); return resp; } }