top.jfunc.common.http.base.AbstractApacheHttpTemplate Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of httpclient Show documentation
Show all versions of httpclient Show documentation
http客户端请求工具类，有多种实现：OkHttp3、ApacheHttpClient、HttpURLConnection、Jodd-Http，可以随意切换http实现
package top.jfunc.common.http.base;
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 top.jfunc.common.http.Method;
import top.jfunc.common.http.ParamUtil;
import top.jfunc.common.http.base.ssl.SSLSocketFactoryBuilder;
import top.jfunc.common.utils.ArrayListMultimap;
import top.jfunc.common.utils.IoUtil;
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.*;
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 AbstractApacheHttpTemplate extends AbstractConfigurableHttp implements HttpTemplate {
protected int _maxRetryTimes = 1;
@Override
public R template(String url, Method method , String contentType, ContentCallback contentCallback, ArrayListMultimap headers, Integer connectTimeout, Integer readTimeout, String resultCharset , boolean includeHeader , ResultCallback resultCallback) throws IOException {
//1.获取完成的URL，创建请求
String completedUrl = addBaseUrlIfNecessary(url);
///*URIBuilder builder = new URIBuilder(url);
//URI uri = builder.build();*/
HttpUriRequest httpUriRequest = createHttpUriRequest(completedUrl , method);
//2.设置请求头
setRequestHeaders(httpUriRequest, contentType, mergeDefaultHeaders(headers));
//3.设置请求参数
setRequestProperty((HttpRequestBase) httpUriRequest,
getConnectionTimeoutWithDefault(connectTimeout),
getReadTimeoutWithDefault(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(completedUrl ,getHostnameVerifier() , getSSLContext());
//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 = top.jfunc.common.http.IoUtil.emptyInputStream();
}*/
InputStream inputStream = getStreamFrom(entity, false);
R convert = resultCallback.convert(statusCode , inputStream, getResultCharsetWithDefault(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);
}
}
protected InputStream getStreamFrom(HttpEntity entity , boolean ignoreResponseBody) throws IOException {
//忽略返回body的情况下，直接返回空的
if(ignoreResponseBody){
return top.jfunc.common.http.IoUtil.emptyInputStream();
}
InputStream inputStream = entity.getContent();
if(null == inputStream){
inputStream = top.jfunc.common.http.IoUtil.emptyInputStream();
}
return inputStream;
}
protected 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
*/
protected 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 = ParamUtil.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{
//default do nothing, give children a chance to do more config
}
/**
* @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) {
setRequestProperty(request, connectTimeout, readTimeout , null);
}
protected void setRequestProperty(HttpRequestBase request,
int connectTimeout,
int readTimeout,
Proxy proxy) {
RequestConfig.Builder builder = RequestConfig.custom()
.setConnectTimeout(connectTimeout)
.setConnectionRequestTimeout(readTimeout)
.setSocketTimeout(readTimeout)
//.setStaleConnectionCheckEnabled(true)
;
//代理设置
if(null != proxy){
InetSocketAddress address = (InetSocketAddress)proxy.address();
HttpHost proxyHost = new HttpHost(address.getHostName() , address.getPort());
builder.setProxy(proxyHost);
}
request.setConfig(builder.build());
}
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);
}
}
protected 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