top.jfunc.common.http.base.AbstractApacheHttp Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of network Show documentation
Show all versions of network Show documentation
http客户端请求工具类，有多种实现：OKHttp3、ApacheHttpClient、HttpURLConnection，可以随意切换http实现
package top.jfunc.common.http.base;
import top.jfunc.common.http.Method;
import top.jfunc.common.http.base.ssl.SSLSocketFactoryBuilder;
import top.jfunc.common.http.smart.Request;
import top.jfunc.common.http.smart.SSLRequest;
import top.jfunc.common.utils.ArrayListMultimap;
import top.jfunc.common.utils.IoUtil;
import org.apache.http.*;
import org.apache.http.client.HttpRequestRetryHandler;
import org.apache.http.client.config.RequestConfig;
import org.apache.http.client.methods.*;
import org.apache.http.client.protocol.HttpClientContext;
import org.apache.http.config.Registry;
import org.apache.http.config.RegistryBuilder;
import org.apache.http.conn.routing.HttpRoute;
import org.apache.http.conn.socket.ConnectionSocketFactory;
import org.apache.http.conn.socket.LayeredConnectionSocketFactory;
import org.apache.http.conn.socket.PlainConnectionSocketFactory;
import org.apache.http.conn.ssl.SSLConnectionSocketFactory;
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.protocol.HttpContext;
import org.apache.http.util.EntityUtils;
import javax.net.ssl.HostnameVerifier;
import javax.net.ssl.SSLContext;
import javax.net.ssl.SSLException;
import javax.net.ssl.SSLHandshakeException;
import java.io.IOException;
import java.io.InputStream;
import java.io.InterruptedIOException;
import java.net.ConnectException;
import java.net.UnknownHostException;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Set;
/**
* 使用Apache SmartHttpClient 实现的Http请求类
* @author 熊诗言2018/6/6
*/
public abstract class AbstractApacheHttp extends AbstractHttp implements HttpTemplate {
protected int _maxRetryTimes = 1;
@Override
public R template(Request request, Method method , ContentCallback contentCallback , ResultCallback resultCallback) throws IOException {
HttpUriRequest httpUriRequest = createHttpUriRequest(request.getUrl(), method);
//2.设置请求头
setRequestHeaders(httpUriRequest, request.getContentType(), request.getHeaders());
//3.设置请求参数
setRequestProperty((HttpRequestBase) httpUriRequest, request.getConnectionTimeout(), request.getReadTimeout());
//4.创建请求内容，如果有的话
if(httpUriRequest instanceof HttpEntityEnclosingRequest){
if(contentCallback != null){
contentCallback.doWriteWith((HttpEntityEnclosingRequest)httpUriRequest);
}
}
CloseableHttpClient httpClient = null;
CloseableHttpResponse response = null;
HttpEntity entity = null;
try {
////////////////////////////////////ssl处理///////////////////////////////////
HostnameVerifier hostnameVerifier = null;
SSLContext sslContext = null;
//https默认设置这些
if(isHttps(request.getUrl())){
hostnameVerifier = getDefaultHostnameVerifier();
sslContext = getDefaultSSLContext();
//客户传过来就用客户的
if(request instanceof SSLRequest){
SSLRequest sslRequest = (SSLRequest)request;
hostnameVerifier = sslRequest.getHostnameVerifier();
sslContext = sslRequest.getSslContext();
}
}
////////////////////////////////////ssl处理///////////////////////////////////
httpClient = getCloseableHttpClient(request.getUrl() , hostnameVerifier , sslContext);
//6.发送请求
response = httpClient.execute(httpUriRequest , HttpClientContext.create());
int statusCode = response.getStatusLine().getStatusCode();
entity = response.getEntity();
InputStream inputStream = entity.getContent();
if(null == inputStream){
inputStream = emptyInputStream();
}
R convert = resultCallback.convert(statusCode , inputStream, request.getResultCharset(), request.isIncludeHeaders() ? parseHeaders(response) : new HashMap<>(0));
IoUtil.close(inputStream);
return convert;
///
/*Response convert = Response.with(statusCode , inputStream , request.getResultCharset() , request.isIncludeHeaders() ? parseHeaders(response) : new HashMap<>(0));
IoUtil.close(inputStream);
return convert;*/
} catch (IOException e) {
throw e;
} catch (Exception e){
throw new RuntimeException(e);
}finally {
EntityUtils.consumeQuietly(entity);
IoUtil.close(response);
IoUtil.close(httpClient);
}
}
@Override
public R template(String url, Method method , String contentType, ContentCallback contentCallback, ArrayListMultimap headers, int connectTimeout, int readTimeout, String resultCharset , boolean includeHeader , ResultCallback resultCallback) throws IOException {
//1.创建请求
///*URIBuilder builder = new URIBuilder(url);
//URI uri = builder.build();*/
HttpUriRequest httpUriRequest = createHttpUriRequest(url , method);
//2.设置请求头
setRequestHeaders(httpUriRequest, contentType, headers);
//3.设置请求参数
setRequestProperty((HttpRequestBase) httpUriRequest, connectTimeout, readTimeout);
//4.创建请求内容，如果有的话
if(httpUriRequest instanceof HttpEntityEnclosingRequest){
if(contentCallback != null){
contentCallback.doWriteWith((HttpEntityEnclosingRequest)httpUriRequest);
}
}
CloseableHttpClient httpClient = null;
CloseableHttpResponse response = null;
HttpEntity entity = null;
try {
//5.创建http客户端
//CloseableHttpClient httpClient = HttpClients.createDefault();
httpClient = getCloseableHttpClient(url ,getDefaultHostnameVerifier() , getDefaultSSLContext());
//6.发送请求
response = httpClient.execute(httpUriRequest , HttpClientContext.create());
int statusCode = response.getStatusLine().getStatusCode();
/*String resultString = EntityUtils.toString(response.getEntity(), resultCharset);*/
entity = response.getEntity();
// if (HttpStatus.HTTP_OK == statusCode) {
// //7.封装返回参数
// InputStream inputStream = entity.getContent();
// R convert = resultCallback.convert(inputStream, resultCharset, includeHeader ? parseHeaders(response) : new HashMap<>(0));
// IoUtil.close(inputStream);
// return convert;
// }else {
// String errorMsg = EntityUtils.toString(entity, resultCharset);
// throw new HttpException(statusCode,errorMsg,parseHeaders(response));
// }
InputStream inputStream = entity.getContent();
// R convert;
// if (HttpStatus.HTTP_OK == statusCode) {
// //7.封装返回参数
// convert = resultCallback.convert(HttpStatus.HTTP_OK , inputStream, resultCharset, includeHeader ? parseHeaders(response) : new HashMap<>(0));
// }else {
// convert = resultCallback.convert(statusCode , inputStream , resultCharset , parseHeaders(response));
// }
if(null == inputStream){
inputStream = emptyInputStream();
}
R convert = resultCallback.convert(statusCode , inputStream, resultCharset, includeHeader ? parseHeaders(response) : new HashMap<>(0));
IoUtil.close(inputStream);
return convert;
} catch (IOException e) {
throw e;
} catch (Exception e){
throw new RuntimeException(e);
}finally {
EntityUtils.consumeQuietly(entity);
IoUtil.close(response);
IoUtil.close(httpClient);
}
}
private HttpUriRequest createHttpUriRequest(String url, Method method) {
switch (method){
case GET : return new HttpGet(url);
case POST : return new HttpPost(url);
case PUT : return new HttpPut(url);
case OPTIONS : return new HttpOptions(url);
case HEAD : return new HttpHead(url);
case PATCH : return new HttpPatch(url);
case TRACE : return new HttpTrace(url);
case DELETE : return new HttpDelete(url);
default : throw new IllegalArgumentException("不支持的http方法 : " + method.name());
}
}
/**
* https://ss.xx.xx.ss:8080/dsda
*/
private CloseableHttpClient getCloseableHttpClient(String url, HostnameVerifier hostnameVerifier , SSLContext sslContext) throws Exception{
return createHttpClient(200, 40, 100, url , hostnameVerifier , sslContext);
}
private boolean retryIf(IOException exception, int executionCount, HttpContext context) {
if (executionCount >= _maxRetryTimes) {
return false;
}
// 如果服务器丢掉了连接，那么就重试
if (exception instanceof NoHttpResponseException) {
return true;
}
// 不要重试SSL握手异常
if (exception instanceof SSLHandshakeException) {
return false;
}
// 超时
if (exception instanceof InterruptedIOException) {
return false;
}
// 目标服务器不可达
if (exception instanceof UnknownHostException) {
return false;
}
// 连接被拒绝
if (exception instanceof ConnectException) {
return false;
}
// SSL握手异常
if (exception instanceof SSLException) {
return false;
}
HttpClientContext clientContext = HttpClientContext
.adapt(context);
HttpRequest request = clientContext.getRequest();
// 如果请求是幂等的，就再次尝试
if (!(request instanceof HttpEntityEnclosingRequest)) {
return true;
}
return false;
}
protected CloseableHttpClient createHttpClient(int maxTotal, int maxPerRoute, int maxRoute, String url , HostnameVerifier hostnameVerifier , SSLContext sslContext) throws Exception{
String hostname = url.split("/")[2];
boolean isHttps = isHttps(url);
int port = isHttps ? 443 : 80;
if (hostname.contains(":")) {
String[] arr = hostname.split(":");
hostname = arr[0];
port = Integer.parseInt(arr[1]);
}
ConnectionSocketFactory csf = PlainConnectionSocketFactory
.getSocketFactory();
LayeredConnectionSocketFactory sslsf = SSLConnectionSocketFactory
.getSocketFactory();
Registry registry = RegistryBuilder
. create().register("http", csf)
.register("https", sslsf).build();
PoolingHttpClientConnectionManager cm = new PoolingHttpClientConnectionManager(registry);
// 将最大连接数增加
cm.setMaxTotal(maxTotal);
// 将每个路由基础的连接增加
cm.setDefaultMaxPerRoute(maxPerRoute);
// 设置不活动的连接1000ms之后Validate
cm.setValidateAfterInactivity(1000);
HttpHost httpHost = new HttpHost(hostname, port);
// 将目标主机的最大连接数增加
cm.setMaxPerRoute(new HttpRoute(httpHost), maxRoute);
// 请求重试处理
HttpRequestRetryHandler httpRequestRetryHandler = this::retryIf ;
HttpClientBuilder httpClientBuilder = HttpClients.custom()
.setConnectionManager(cm)
.setRetryHandler(httpRequestRetryHandler);
if(isHttps){
initSSL(httpClientBuilder , hostnameVerifier , sslContext);
}
//给子类复写的机会
doWithClient(httpClientBuilder , isHttps);
return httpClientBuilder.build();
}
protected void doWithClient(HttpClientBuilder httpClientBuilder , boolean isHttps) throws Exception{
}
/**
* @see SSLSocketFactoryBuilder#getSSLContext()
* @see SSLSocketFactoryBuilder#getSSLContext(String, String)
*/
protected void initSSL(HttpClientBuilder httpClientBuilder , HostnameVerifier hostnameVerifier , SSLContext sslContext) {
// 验证域
if(null != hostnameVerifier){
httpClientBuilder.setSSLHostnameVerifier(hostnameVerifier);
}
if(null != sslContext){
httpClientBuilder.setSSLContext(sslContext);
}
}
protected void setRequestBody(HttpEntityEnclosingRequest request, String body, String bodyCharset) {
if(body == null || bodyCharset == null){return;}
StringEntity entity = new StringEntity(body, bodyCharset);
entity.setContentEncoding(bodyCharset);
request.setEntity(entity);
}
protected void setRequestProperty(HttpRequestBase request, int connectTimeout, int readTimeout) {
RequestConfig config = RequestConfig.custom()
.setConnectTimeout(connectTimeout)
.setConnectionRequestTimeout(readTimeout)
.setSocketTimeout(readTimeout)
//.setStaleConnectionCheckEnabled(true)
.build();
request.setConfig(config);
}
protected void setRequestHeaders(HttpUriRequest request, String contentType, ArrayListMultimap headers) {
if(null != headers) {
Set keySet = headers.keySet();
keySet.forEach((k)->headers.get(k).forEach((v)->request.addHeader(k,v)));
}
if(null != contentType){
request.setHeader(top.jfunc.common.http.Header.CONTENT_TYPE.toString(), contentType);
}
}
private Map> parseHeaders(CloseableHttpResponse response) {
Header[] allHeaders = response.getAllHeaders();
ArrayListMultimap arrayListMultimap = new ArrayListMultimap<>(allHeaders.length);
for (Header header : allHeaders) {
arrayListMultimap.put(header.getName() , header.getValue());
}
return arrayListMultimap.getMap();
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy