package weixin.popular.client;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.InputStream;
import java.security.KeyStore;
import java.security.KeyStoreException;
import java.security.NoSuchAlgorithmException;
import java.security.cert.CertificateException;
import java.util.HashMap;
import java.util.Map;
import java.util.UUID;
import org.apache.http.HttpEntity;
import org.apache.http.client.ResponseHandler;
import org.apache.http.client.methods.CloseableHttpResponse;
import org.apache.http.client.methods.HttpEntityEnclosingRequestBase;
import org.apache.http.client.methods.HttpUriRequest;
import org.apache.http.client.protocol.HttpClientContext;
import org.apache.http.entity.ContentType;
import org.apache.http.impl.client.CloseableHttpClient;
import org.apache.http.util.EntityUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
public class LocalHttpClient {
private static final Logger logger = LoggerFactory.getLogger(LocalHttpClient.class);
private static int timeout = 8000;
private static int retryExecutionCount = 2;
protected static CloseableHttpClient httpClient = HttpClientFactory.createHttpClient(100,10,timeout,retryExecutionCount);
private static Map<String,CloseableHttpClient> httpClient_mchKeyStore = new HashMap<String, CloseableHttpClient>();
private static ResultErrorHandler resultErrorHandler;
/**
* @since 2.7.0
* @param timeout timeout
*/
public static void setTimeout(int timeout) {
LocalHttpClient.timeout = timeout;
}
/**
* @since 2.7.0
* @param retryExecutionCount retryExecutionCount
*/
public static void setRetryExecutionCount(int retryExecutionCount) {
LocalHttpClient.retryExecutionCount = retryExecutionCount;
}
/**
* @since 2.8.3
* @param resultErrorHandler 数据返回错误处理
*/
public static void setResultErrorHandler(ResultErrorHandler resultErrorHandler) {
LocalHttpClient.resultErrorHandler = resultErrorHandler;
}
/**
*
* @param maxTotal maxTotal
* @param maxPerRoute maxPerRoute
*/
public static void init(int maxTotal,int maxPerRoute){
try {
httpClient.close();
} catch (IOException e) {
e.printStackTrace();
}
httpClient = HttpClientFactory.createHttpClient(maxTotal,maxPerRoute,timeout,retryExecutionCount);
}
/**
* 初始化 MCH HttpClient KeyStore
* @param mch_id mch_id
* @param keyStoreFilePath keyStoreFilePath
*/
public static void initMchKeyStore(String mch_id,String keyStoreFilePath){
try {
initMchKeyStore(mch_id, new FileInputStream(new File(keyStoreFilePath)));
} catch (FileNotFoundException e) {
e.printStackTrace();
}
}
/**
* 初始化 MCH HttpClient KeyStore
* @since 2.8.7
* @param mch_id mch_id
* @param inputStream p12 文件流
*/
public static void initMchKeyStore(String mch_id, InputStream inputStream) {
try {
KeyStore keyStore = KeyStore.getInstance("PKCS12");
keyStore.load(inputStream,mch_id.toCharArray());
inputStream.close();
CloseableHttpClient httpClient = HttpClientFactory.createKeyMaterialHttpClient(keyStore, mch_id,timeout,retryExecutionCount);
httpClient_mchKeyStore.put(mch_id, httpClient);
} catch (KeyStoreException e) {
e.printStackTrace();
} catch (FileNotFoundException e) {
e.printStackTrace();
} catch (NoSuchAlgorithmException e) {
e.printStackTrace();
} catch (CertificateException e) {
e.printStackTrace();
} catch (IOException e) {
e.printStackTrace();
}
}
public static CloseableHttpResponse execute(HttpUriRequest request){
loggerRequest(request);
try {
return httpClient.execute(request,HttpClientContext.create());
} catch (Exception e) {
e.printStackTrace();
logger.error(e.getMessage());
}
return null;
}
public static <T> T execute(HttpUriRequest request,ResponseHandler<T> responseHandler){
String uriId = loggerRequest(request);
if(responseHandler instanceof LocalResponseHandler){
LocalResponseHandler lrh = (LocalResponseHandler) responseHandler;
lrh.setUriId(uriId);
}
try {
T t = httpClient.execute(request, responseHandler,HttpClientContext.create());
if(resultErrorHandler != null){
resultErrorHandler.doHandle(uriId, request, t);
}
return t;
} catch (Exception e) {
e.printStackTrace();
logger.error(e.getMessage());
}
return null;
}
/**
* 数据返回自动JSON对象解析
* @param request request
* @param clazz clazz
* @return result
*/
public static <T> T executeJsonResult(HttpUriRequest request,Class<T> clazz){
return execute(request,JsonResponseHandler.createResponseHandler(clazz));
}
/**
* 数据返回自动XML对象解析
* @param request request
* @param clazz clazz
* @return result
*/
public static <T> T executeXmlResult(HttpUriRequest request,Class<T> clazz){
return execute(request,XmlResponseHandler.createResponseHandler(clazz));
}
/**
* 数据返回自动XML对象解析
* @param request request
* @param clazz clazz
* @param sign_type 数据返回验证签名类型
* @param key 数据返回验证签名key
* @return result
* @since 2.8.5
*/
public static <T> T executeXmlResult(HttpUriRequest request,Class<T> clazz,String sign_type,String key){
return execute(request,XmlResponseHandler.createResponseHandler(clazz,sign_type,key));
}
/**
* MCH keyStore 请求 数据返回自动XML对象解析
* @param mch_id mch_id
* @param request request
* @param clazz clazz
* @return result
*/
public static <T> T keyStoreExecuteXmlResult(String mch_id,HttpUriRequest request,Class<T> clazz){
return keyStoreExecuteXmlResult(mch_id, request, clazz, null,null);
}
/**
* MCH keyStore 请求 数据返回自动XML对象解析
* @param mch_id mch_id
* @param request request
* @param sign_type 数据返回验证签名类型
* @param key 数据返回验证签名key
* @since 2.8.5
* @return result
*/
public static <T> T keyStoreExecuteXmlResult(String mch_id,HttpUriRequest request,Class<T> clazz,String sign_type,String key){
String uriId = loggerRequest(request);
ResponseHandler<T> responseHandler = XmlResponseHandler.createResponseHandler(clazz,sign_type,key);
if(responseHandler instanceof LocalResponseHandler){
LocalResponseHandler lrh = (LocalResponseHandler) responseHandler;
lrh.setUriId(uriId);
}
try {
T t = httpClient_mchKeyStore.get(mch_id).execute(request,responseHandler,HttpClientContext.create());
if(resultErrorHandler != null){
resultErrorHandler.doHandle(uriId, request, t);
}
return t;
} catch (Exception e) {
e.printStackTrace();
logger.error(e.getMessage());
}
return null;
}
/**
* 日志记录
* @param request request
* @return log request id
*/
private static String loggerRequest(HttpUriRequest request){
String id = UUID.randomUUID().toString();
if(logger.isInfoEnabled()||logger.isDebugEnabled()){
if(request instanceof HttpEntityEnclosingRequestBase){
HttpEntityEnclosingRequestBase request_base = (HttpEntityEnclosingRequestBase)request;
HttpEntity entity = request_base.getEntity();
String content = null;
//MULTIPART_FORM_DATA 请求类型判断
if(entity.getContentType().toString().indexOf(ContentType.MULTIPART_FORM_DATA.getMimeType()) == -1){
try {
content = EntityUtils.toString(entity);
} catch (Exception e) {
e.printStackTrace();
logger.error(e.getMessage());
}
}
logger.info("URI[{}] {} {} ContentLength:{} Content:{}",
id,
request.getURI().toString(),
entity.getContentType(),
entity.getContentLength(),
content == null?"multipart_form_data":content);
}else{
logger.info("URI[{}] {}",id,request.getURI().toString());
}
}
return id;
}
}