From c5b3c551b845b5c3f79499656ffa30cce2628aa5 Mon Sep 17 00:00:00 2001 From: egan Date: Mon, 25 Jun 2018 09:27:42 +0800 Subject: [PATCH] =?UTF-8?q?http=E8=BF=9E=E6=8E=A5=E6=B1=A0?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../pay/common/http/HttpConfigStorage.java | 33 ++++++++- .../pay/common/http/HttpRequestTemplate.java | 69 +++++++++++++++---- 2 files changed, 84 insertions(+), 18 deletions(-) diff --git a/pay-java-common/src/main/java/com/egzosn/pay/common/http/HttpConfigStorage.java b/pay-java-common/src/main/java/com/egzosn/pay/common/http/HttpConfigStorage.java index f455b4a..737cda0 100644 --- a/pay-java-common/src/main/java/com/egzosn/pay/common/http/HttpConfigStorage.java +++ b/pay-java-common/src/main/java/com/egzosn/pay/common/http/HttpConfigStorage.java @@ -32,11 +32,22 @@ public class HttpConfigStorage { */ private boolean isPath = true; - //https请求所需的证书（PKCS12） + /** + * https请求所需的证书（PKCS12） + */ private String keystore; - //证书对应的密码 + /** + * 证书对应的密码 + */ private String storePassword; - + /** + * 最大连接数 + */ + private int maxTotal = 0; + /** + * 默认的每个路由的最大连接数 + */ + private int defaultMaxPerRoute = 0; /** * http代理地址 @@ -190,4 +201,20 @@ public class HttpConfigStorage { public void setStorePassword(String storePassword) { this.storePassword = storePassword; } + + public int getMaxTotal() { + return maxTotal; + } + + public void setMaxTotal(int maxTotal) { + this.maxTotal = maxTotal; + } + + public int getDefaultMaxPerRoute() { + return defaultMaxPerRoute; + } + + public void setDefaultMaxPerRoute(int defaultMaxPerRoute) { + this.defaultMaxPerRoute = defaultMaxPerRoute; + } } diff --git a/pay-java-common/src/main/java/com/egzosn/pay/common/http/HttpRequestTemplate.java b/pay-java-common/src/main/java/com/egzosn/pay/common/http/HttpRequestTemplate.java index b005495..bfbe3c6 100644 --- a/pay-java-common/src/main/java/com/egzosn/pay/common/http/HttpRequestTemplate.java +++ b/pay-java-common/src/main/java/com/egzosn/pay/common/http/HttpRequestTemplate.java @@ -8,11 +8,16 @@ import org.apache.http.auth.AuthScope; import org.apache.http.auth.UsernamePasswordCredentials; import org.apache.http.client.CredentialsProvider; import org.apache.http.client.methods.CloseableHttpResponse; +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.DefaultHostnameVerifier; import org.apache.http.conn.ssl.SSLConnectionSocketFactory; import org.apache.http.impl.client.BasicCredentialsProvider; 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.ssl.SSLContexts; import javax.net.ssl.SSLContext; @@ -35,8 +40,11 @@ public class HttpRequestTemplate { protected CloseableHttpClient httpClient; + protected PoolingHttpClientConnectionManager connectionManager; + protected HttpHost httpProxy; + HttpConfigStorage configStorage; /** * 获取代理带代理地址的 HttpHost * @return 获取代理带代理地址的 HttpHost @@ -46,7 +54,27 @@ public class HttpRequestTemplate { } public CloseableHttpClient getHttpClient() { + if (null != httpClient) { + return httpClient; + } + if (null == configStorage) { + return httpClient = HttpClients.createDefault(); + } + + CloseableHttpClient httpClient = HttpClients + .custom() + //网络提供者 + .setDefaultCredentialsProvider(createCredentialsProvider(configStorage)) + //设置httpclient的SSLSocketFactory + .setSSLSocketFactory(createSSL(configStorage)) + .setConnectionManager(connectionManager(configStorage)) + .build(); + if (null == connectionManager) { + return this.httpClient = httpClient; + } + return httpClient; + } /** @@ -126,6 +154,28 @@ public class HttpRequestTemplate { return credsProvider; } + /** + * 初始化连接池 + * @param configStorage 配置 + * @return 连接池对象 + */ + public PoolingHttpClientConnectionManager connectionManager(HttpConfigStorage configStorage){ + if (null != connectionManager){ + return connectionManager; + } + if (0 == configStorage.getMaxTotal() || 0 == configStorage.getDefaultMaxPerRoute()){ + return null; + } + Registry socketFactoryRegistry = RegistryBuilder. create() + .register("https", createSSL(configStorage)) + .register("http", new PlainConnectionSocketFactory()) + .build(); + connectionManager = new PoolingHttpClientConnectionManager(socketFactoryRegistry); + connectionManager.setMaxTotal(configStorage.getMaxTotal()); + connectionManager.setDefaultMaxPerRoute(configStorage.getDefaultMaxPerRoute()); + + return connectionManager; + } /** * 设置HTTP请求的配置 @@ -134,24 +184,13 @@ public class HttpRequestTemplate { * @return 当前HTTP请求的客户端模板 */ public HttpRequestTemplate setHttpConfigStorage(HttpConfigStorage configStorage) { - - if (null == configStorage) { - httpClient = HttpClients.createDefault(); - return this; - } - - httpClient = HttpClients - .custom() - //网络提供者 - .setDefaultCredentialsProvider(createCredentialsProvider(configStorage)) - //设置httpclient的SSLSocketFactory - .setSSLSocketFactory(createSSL(configStorage)) - .build(); - + this.configStorage = configStorage; return this; } + + /** * * post @@ -275,7 +314,7 @@ public class HttpRequestTemplate { httpRequest.setProxy(httpProxy); } httpRequest.setResponseType(responseType); - try (CloseableHttpResponse response = httpClient.execute(httpRequest)) { + try (CloseableHttpResponse response = getHttpClient().execute(httpRequest)) { return httpRequest.handleResponse(response); }catch (IOException e){ e.printStackTrace();