HTTP

package com.chargerlink.util;


import java.io.IOException;
import java.net.SocketTimeoutException;
import java.nio.charset.Charset;
import java.security.GeneralSecurityException;
import java.security.cert.CertificateException;
import java.security.cert.X509Certificate;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.Map.Entry;


import javax.net.ssl.SSLContext;
import javax.net.ssl.SSLException;
import javax.net.ssl.SSLSession;
import javax.net.ssl.SSLSocket;


import org.apache.http.HttpEntity;
import org.apache.http.HttpStatus;
import org.apache.http.NameValuePair;
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.conn.HttpHostConnectException;
import org.apache.http.conn.ssl.SSLConnectionSocketFactory;
import org.apache.http.conn.ssl.SSLContextBuilder;
import org.apache.http.conn.ssl.TrustStrategy;
import org.apache.http.conn.ssl.X509HostnameVerifier;
import org.apache.http.entity.StringEntity;
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.util.EntityUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;


/**
 * HTTP 请求工具类
 */
public class HttpPlusUtil {


private final static Logger logger = LoggerFactory.getLogger(HttpPlusUtil.class);


private static PoolingHttpClientConnectionManager connMgr;
private static RequestConfig requestConfig;
private static final int MAX_TIMEOUT = 7000;


static {
// 设置连接池
connMgr = new PoolingHttpClientConnectionManager();
// 设置连接池大小
connMgr.setMaxTotal(1200);
connMgr.setDefaultMaxPerRoute(connMgr.getMaxTotal());


RequestConfig.Builder configBuilder = RequestConfig.custom();
// 设置连接超时
configBuilder.setConnectTimeout(MAX_TIMEOUT);
// 设置读取超时
configBuilder.setSocketTimeout(MAX_TIMEOUT);
// 设置从连接池获取连接实例的超时
configBuilder.setConnectionRequestTimeout(MAX_TIMEOUT);
// 在提交请求之前 测试连接是否可用
configBuilder.setStaleConnectionCheckEnabled(true);
requestConfig = configBuilder.build();
}


/**
* 发送 GET 请求（HTTP），不带输入数据
*
* @param url
* @return
*/
public static String doGet(String url) {
return doGet(url, new HashMap<String, Object>());
}


/**
* 发送 GET 请求（HTTP），K-V形式
*
* @param url
* @param params
* @return
*/
public static String doGet(String url, Map<String, Object> params) {
String apiUrl = url;
StringBuffer param = new StringBuffer();
int i = 0;
for (String key : params.keySet()) {
if (i == 0)
param.append("?");
else
param.append("&");
param.append(key).append("=").append(params.get(key));
i++;
}
apiUrl += param;
String httpStr = null;
CloseableHttpResponse response = null;
CloseableHttpClient httpClient = HttpClients.custom().setConnectionManager(connMgr)
.setDefaultRequestConfig(requestConfig).build();
try {
HttpGet httpGet = new HttpGet(apiUrl);
response = httpClient.execute(httpGet);
int statusCode = response.getStatusLine().getStatusCode();


System.out.println("执行状态码 : " + statusCode);


HttpEntity entity = response.getEntity();
httpStr = EntityUtils.toString(entity, "utf-8");


} catch (IOException e) {
logger.error(e.getMessage(), e);
httpStr = proccessExption(e);
} finally {
closeResponse(response);
}
return httpStr;
}


/**
* 发送 GET 请求（HTTP），K-V形式
*
* @param url
* @param params
* @return
*/
public static String doGet(String url, Map<String, Object> params, Map<String, String> headerParams) {
String apiUrl = url;
StringBuffer param = new StringBuffer();
int i = 0;
for (String key : params.keySet()) {
if (i == 0)
param.append("?");
else
param.append("&");
param.append(key).append("=").append(params.get(key));
i++;
}
apiUrl += param;
String httpStr = null;
CloseableHttpResponse response = null;
CloseableHttpClient httpClient = HttpClients.custom().setConnectionManager(connMgr)
.setDefaultRequestConfig(requestConfig).build();
try {
HttpGet httpGet = new HttpGet(apiUrl);
if (headerParams.size() > 0) {
Set<Entry<String, String>> set = headerParams.entrySet();
for (Entry<String, String> entry : set) {
httpGet.addHeader(entry.getKey(), entry.getValue());
}
}
response = httpClient.execute(httpGet);
int statusCode = response.getStatusLine().getStatusCode();


System.out.println("执行状态码 : " + statusCode);


HttpEntity entity = response.getEntity();
httpStr = EntityUtils.toString(entity, "utf-8");


} catch (IOException e) {
logger.error(e.getMessage(), e);
httpStr = proccessExption(e);
} finally {
closeResponse(response);
}
return httpStr;
}


/**
* 发送 POST 请求（HTTP），不带输入数据
*
* @param apiUrl
* @return
*/
public static String doPost(String apiUrl) {
return doPost(apiUrl, new HashMap<String, Object>());
}


/**
* 发送 POST 请求（HTTP），K-V形式
*
* @param apiUrl
*            API接口URL
* @param params
*            参数map
* @return
*/
public static String doPost(String apiUrl, Map<String, Object> params, Map<String, String> headerParams) {
CloseableHttpClient httpClient = HttpClients.custom().setConnectionManager(connMgr)
.setDefaultRequestConfig(requestConfig).build();
String httpStr = null;
HttpPost httpPost = new HttpPost(apiUrl);
CloseableHttpResponse response = null;
try {
httpPost.setConfig(requestConfig);
List<NameValuePair> pairList = new ArrayList<>(params.size());
for (Map.Entry<String, Object> entry : params.entrySet()) {
NameValuePair pair = new BasicNameValuePair(entry.getKey(), entry.getValue().toString());
pairList.add(pair);
}
httpPost.setEntity(new UrlEncodedFormEntity(pairList, Charset.forName("UTF-8")));
if (headerParams.size() > 0) {
Set<Entry<String, String>> set = headerParams.entrySet();
for (Entry<String, String> entry : set) {
httpPost.addHeader(entry.getKey(), entry.getValue());
}
}
response = httpClient.execute(httpPost);
System.out.println(response.toString());
HttpEntity entity = response.getEntity();
httpStr = EntityUtils.toString(entity, "UTF-8");
} catch (IOException e) {
logger.error(e.getMessage(), e);
httpStr = proccessExption(e);
} finally {
closeResponse(response);
}
return httpStr;
}


/**
* 发送 POST 请求（HTTP），K-V形式
*
* @param apiUrl
*            API接口URL
* @param params
*            参数map
* @return
*/
public static String doPost(String apiUrl, Map<String, Object> params) {
logger.debug("doPost api:"+apiUrl);
CloseableHttpClient httpClient = HttpClients.custom().setConnectionManager(connMgr)
.setDefaultRequestConfig(requestConfig).build();
String httpStr = null;
HttpPost httpPost = new HttpPost(apiUrl);
CloseableHttpResponse response = null;
try {
httpPost.setConfig(requestConfig);
List<NameValuePair> pairList = new ArrayList<>(params.size());
for (Map.Entry<String, Object> entry : params.entrySet()) {
NameValuePair pair = new BasicNameValuePair(entry.getKey(), entry.getValue().toString());
pairList.add(pair);
}
httpPost.setEntity(new UrlEncodedFormEntity(pairList, Charset.forName("UTF-8")));
response = httpClient.execute(httpPost);
System.out.println(response.toString());
HttpEntity entity = response.getEntity();
httpStr = EntityUtils.toString(entity, "UTF-8");
} catch (IOException e) {
logger.error(e.getMessage(), e);
httpStr = proccessExption(e);
} finally {
closeResponse(response);
}
return httpStr;
}


/**
* 发送 POST 请求（HTTP），JSON形式
*
* @param apiUrl
* @param json
*            json对象
* @return
*/
public static String doPost(String apiUrl, Object json) {
logger.debug("doPost api:"+apiUrl);
CloseableHttpClient httpClient = HttpClients.custom().setConnectionManager(connMgr)
.setDefaultRequestConfig(requestConfig).build();
String httpStr = null;
HttpPost httpPost = new HttpPost(apiUrl);
CloseableHttpResponse response = null;


try {
StringEntity stringEntity = new StringEntity(json.toString(), "UTF-8");// 解决中文乱码问题
stringEntity.setContentEncoding("UTF-8");
stringEntity.setContentType("application/json");
httpPost.setEntity(stringEntity);
response = httpClient.execute(httpPost);
HttpEntity entity = response.getEntity();
System.out.println(response.getStatusLine().getStatusCode());
httpStr = EntityUtils.toString(entity, "UTF-8");
} catch (IOException e) {
logger.error(e.getMessage(), e);
httpStr = proccessExption(e);
} finally {
closeResponse(response);
}
return httpStr;
}


/**
* 发送 SSL POST 请求（HTTPS），K-V形式
*
* @param apiUrl
*            API接口URL
* @param params
*            参数map
* @return
*/
public static String doPostSSL(String apiUrl, Map<String, Object> params) {
CloseableHttpClient httpClient = HttpClients.custom().setSSLSocketFactory(createSSLConnSocketFactory())
.setConnectionManager(connMgr).setDefaultRequestConfig(requestConfig).build();
HttpPost httpPost = new HttpPost(apiUrl);
CloseableHttpResponse response = null;
String httpStr = null;


try {
httpPost.setConfig(requestConfig);
List<NameValuePair> pairList = new ArrayList<NameValuePair>(params.size());
for (Map.Entry<String, Object> entry : params.entrySet()) {
NameValuePair pair = new BasicNameValuePair(entry.getKey(), entry.getValue().toString());
pairList.add(pair);
}
httpPost.setEntity(new UrlEncodedFormEntity(pairList, Charset.forName("utf-8")));
response = httpClient.execute(httpPost);
int statusCode = response.getStatusLine().getStatusCode();
if (statusCode != HttpStatus.SC_OK) {
return null;
}
HttpEntity entity = response.getEntity();
if (entity == null) {
return null;
}
httpStr = EntityUtils.toString(entity, "utf-8");
} catch (Exception e) {
logger.error(e.getMessage(), e);
httpStr = proccessExption(e);
} finally {
closeResponse(response);
}
return httpStr;
}


/**
* 发送 SSL POST 请求（HTTPS），JSON形式
*
* @param apiUrl
*            API接口URL
* @param json
*            JSON对象
* @return
*/
public static String doPostSSL(String apiUrl, Object json) {
CloseableHttpClient httpClient = HttpClients.custom().setSSLSocketFactory(createSSLConnSocketFactory())
.setConnectionManager(connMgr).setDefaultRequestConfig(requestConfig).build();
HttpPost httpPost = new HttpPost(apiUrl);
CloseableHttpResponse response = null;
String httpStr = null;


try {
httpPost.setConfig(requestConfig);
StringEntity stringEntity = new StringEntity(json.toString(), "UTF-8");// 解决中文乱码问题
stringEntity.setContentEncoding("UTF-8");
stringEntity.setContentType("application/json");
httpPost.setEntity(stringEntity);
response = httpClient.execute(httpPost);
int statusCode = response.getStatusLine().getStatusCode();
if (statusCode != HttpStatus.SC_OK) {
return null;
}
HttpEntity entity = response.getEntity();
if (entity == null) {
return null;
}
httpStr = EntityUtils.toString(entity, "utf-8");
} catch (Exception e) {
e.printStackTrace();
httpStr = proccessExption(e);
} finally {
closeResponse(response);
}
return httpStr;
}


/**
* 创建SSL安全连接
*
* @return
*/
private static SSLConnectionSocketFactory createSSLConnSocketFactory() {
SSLConnectionSocketFactory sslsf = null;
try {
SSLContext sslContext = new SSLContextBuilder().loadTrustMaterial(null, new TrustStrategy() {


public boolean isTrusted(X509Certificate[] chain, String authType) throws CertificateException {
return true;
}
}).build();
sslsf = new SSLConnectionSocketFactory(sslContext, new X509HostnameVerifier() {


@Override
public boolean verify(String arg0, SSLSession arg1) {
return true;
}


@Override
public void verify(String host, SSLSocket ssl) throws IOException {
}


@Override
public void verify(String host, X509Certificate cert) throws SSLException {
}


@Override
public void verify(String host, String[] cns, String[] subjectAlts) throws SSLException {
}
});
} catch (GeneralSecurityException e) {
logger.error(e.getMessage(), e);
}
return sslsf;
}


/**
* 设置超时时间
*
* @param timeout
* @return
*/
public static void setTimeout(int timeout) {
RequestConfig.Builder configBuilder = RequestConfig.custom();
// 设置连接超时
configBuilder.setConnectTimeout(timeout);
// 设置读取超时
configBuilder.setSocketTimeout(timeout);
// 设置从连接池获取连接实例的超时
configBuilder.setConnectionRequestTimeout(timeout);
// 在提交请求之前 测试连接是否可用
configBuilder.setStaleConnectionCheckEnabled(true);
requestConfig = configBuilder.build();
}


private static void closeResponse(CloseableHttpResponse response) {
if (response != null) {
try {
EntityUtils.consume(response.getEntity());
} catch (IOException e) {
logger.error(e.getMessage(), e);
}
}
}


private static String proccessExption(Exception e) {
String message;
if (e instanceof SocketTimeoutException) {
int connectTimeout = requestConfig.getConnectTimeout();
message = String.format("HttpConnectTimeout  :%d", connectTimeout);
} else if (e instanceof HttpHostConnectException) {
HttpHostConnectException hc = (HttpHostConnectException) e;
message = String.format("HttpHostConnectException host :%s : %s", hc.getHost(), hc.getMessage());
} else {
message = e.getMessage();
}
return message;
}


/**
* 测试方法
*
* @param args
*/
public static void main(String[] args) throws Exception {


}
}