com.aliyun.openservices.log.http.comm.DefaultServiceClient Maven / Gradle / Ivy
/*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
*/
package com.aliyun.openservices.log.http.comm;
import com.aliyun.openservices.log.exception.LogException;
import com.aliyun.openservices.log.http.client.ClientConfiguration;
import com.aliyun.openservices.log.http.client.ClientErrorCode;
import com.aliyun.openservices.log.http.client.ClientException;
import com.aliyun.openservices.log.http.utils.ExceptionFactory;
import com.aliyun.openservices.log.http.utils.HttpHeaders;
import com.aliyun.openservices.log.http.utils.HttpUtil;
import com.aliyun.openservices.log.http.utils.IOUtils;
import org.apache.commons.codec.binary.Base64;
import org.apache.http.Header;
import org.apache.http.HttpHost;
import org.apache.http.auth.AUTH;
import org.apache.http.auth.AuthScope;
import org.apache.http.auth.NTCredentials;
import org.apache.http.client.AuthCache;
import org.apache.http.client.CredentialsProvider;
import org.apache.http.client.config.RequestConfig;
import org.apache.http.client.methods.CloseableHttpResponse;
import org.apache.http.client.methods.HttpRequestBase;
import org.apache.http.client.protocol.HttpClientContext;
import org.apache.http.config.Registry;
import org.apache.http.config.RegistryBuilder;
import org.apache.http.config.SocketConfig;
import org.apache.http.conn.HttpClientConnectionManager;
import org.apache.http.conn.socket.ConnectionSocketFactory;
import org.apache.http.conn.socket.PlainConnectionSocketFactory;
import org.apache.http.conn.ssl.NoopHostnameVerifier;
import org.apache.http.conn.ssl.SSLConnectionSocketFactory;
import org.apache.http.conn.ssl.TrustStrategy;
import org.apache.http.impl.auth.BasicScheme;
import org.apache.http.impl.client.BasicAuthCache;
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.SSLContextBuilder;
import javax.net.ssl.SSLContext;
import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.security.cert.CertificateException;
import java.security.cert.X509Certificate;
/**
* Default implementation of {@link ServiceClient}.
*/
public class DefaultServiceClient extends ServiceClient {
protected static HttpRequestFactory httpRequestFactory = new HttpRequestFactory();
protected CloseableHttpClient httpClient;
protected HttpClientConnectionManager connectionManager;
protected RequestConfig requestConfig;
protected CredentialsProvider credentialsProvider;
protected HttpHost proxyHttpHost;
protected AuthCache authCache;
public DefaultServiceClient(ClientConfiguration config) {
super(config);
this.connectionManager = createHttpClientConnectionManager();
this.httpClient = createHttpClient(this.connectionManager, config);
RequestConfig.Builder requestConfigBuilder = RequestConfig.custom();
requestConfigBuilder.setConnectTimeout(config.getConnectionTimeout());
requestConfigBuilder.setSocketTimeout(config.getSocketTimeout());
requestConfigBuilder.setConnectionRequestTimeout(config.getConnectionRequestTimeout());
String proxyHost = config.getProxyHost();
int proxyPort = config.getProxyPort();
if (proxyHost != null && proxyPort > 0) {
this.proxyHttpHost = new HttpHost(proxyHost, proxyPort);
requestConfigBuilder.setProxy(proxyHttpHost);
String proxyUsername = config.getProxyUsername();
String proxyPassword = config.getProxyPassword();
String proxyDomain = config.getProxyDomain();
String proxyWorkstation = config.getProxyWorkstation();
if (proxyUsername != null && proxyPassword != null) {
this.credentialsProvider = new BasicCredentialsProvider();
this.credentialsProvider.setCredentials(new AuthScope(proxyHost, proxyPort),
new NTCredentials(proxyUsername, proxyPassword, proxyWorkstation, proxyDomain));
this.authCache = new BasicAuthCache();
authCache.put(this.proxyHttpHost, new BasicScheme());
}
}
this.requestConfig = requestConfigBuilder.build();
}
@Override
public ResponseMessage sendRequestCore(ServiceClient.Request request, String charset) throws IOException, LogException {
HttpRequestBase httpRequest = httpRequestFactory.createHttpRequest(request, charset);
setProxyAuthorizationIfNeed(httpRequest);
HttpClientContext httpContext = createHttpContext();
httpContext.setRequestConfig(this.requestConfig);
CloseableHttpResponse httpResponse = null;
try {
httpResponse = httpClient.execute(httpRequest, httpContext);
} catch (IOException ex) {
httpRequest.abort();
throw ExceptionFactory.createNetworkException(ex);
}
return buildResponse(request, httpResponse);
}
protected static ResponseMessage buildResponse(ServiceClient.Request request, CloseableHttpResponse httpResponse)
throws IOException {
assert (httpResponse != null);
ResponseMessage response = new ResponseMessage();
response.setUrl(request.getUri());
if (httpResponse.getStatusLine() != null) {
response.setStatusCode(httpResponse.getStatusLine().getStatusCode());
}
if (httpResponse.getEntity() != null) {
if (response.isSuccessful()) {
response.setContent(httpResponse.getEntity().getContent());
} else {
readAndSetErrorResponse(httpResponse.getEntity().getContent(), response);
}
}
for (Header header : httpResponse.getAllHeaders()) {
if (HttpHeaders.CONTENT_LENGTH.equals(header.getName())) {
response.setContentLength(Long.parseLong(header.getValue()));
}
response.addHeader(header.getName(), header.getValue());
}
HttpUtil.convertHeaderCharsetFromIso88591(response.getHeaders());
return response;
}
private static void readAndSetErrorResponse(InputStream originalContent, ResponseMessage response)
throws IOException {
byte[] contentBytes = IOUtils.readStreamAsByteArray(originalContent);
response.setErrorResponseAsString(new String(contentBytes));
response.setContent(new ByteArrayInputStream(contentBytes));
}
@Override
protected RetryStrategy getDefaultRetryStrategy() {
if (config.getRetryDisabled()) {
return new NeverRetryStrategy();
}
return new DefaultRetryStrategy();
}
private static class NeverRetryStrategy extends RetryStrategy {
@Override
public boolean shouldRetry(Exception ex, RequestMessage request, int retries) {
return false;
}
}
private static class DefaultRetryStrategy extends RetryStrategy {
@Override
public boolean shouldRetry(Exception ex, RequestMessage request, int retries) {
if (ex instanceof ClientException) {
String errorCode = ((ClientException) ex).getErrorCode();
if (errorCode.equals(ClientErrorCode.CONNECTION_TIMEOUT)
|| errorCode.equals(ClientErrorCode.SOCKET_TIMEOUT)
|| errorCode.equals(ClientErrorCode.CONNECTION_REFUSED)
|| errorCode.equals(ClientErrorCode.UNKNOWN_HOST)
|| errorCode.equals(ClientErrorCode.SOCKET_EXCEPTION)
|| errorCode.equals(ClientErrorCode.SSL_EXCEPTION)) {
return true;
}
// Don't retry when request input stream is non-repeatable
if (errorCode.equals(ClientErrorCode.NONREPEATABLE_REQUEST)) {
return false;
}
}
return false;
}
}
protected CloseableHttpClient createHttpClient(HttpClientConnectionManager connectionManager,
ClientConfiguration config) {
return HttpClients.custom()
.setConnectionManager(connectionManager)
.setConnectionManagerShared(config.isConnManagerShared())
.disableContentCompression()
.disableAutomaticRetries()
.build();
}
protected HttpClientConnectionManager createHttpClientConnectionManager() {
SSLContext sslContext = null;
try {
sslContext = new SSLContextBuilder().loadTrustMaterial(null, new TrustStrategy() {
@Override
public boolean isTrusted(X509Certificate[] chain, String authType) throws CertificateException {
return true;
}
}).build();
} catch (Exception e) {
throw new ClientException(e.getMessage());
}
SSLConnectionSocketFactory sslSocketFactory = new SSLConnectionSocketFactory(sslContext,
NoopHostnameVerifier.INSTANCE);
Registry socketFactoryRegistry = RegistryBuilder.create()
.register(Protocol.HTTP.toString(), PlainConnectionSocketFactory.getSocketFactory())
.register(Protocol.HTTPS.toString(), sslSocketFactory).build();
PoolingHttpClientConnectionManager connectionManager = new PoolingHttpClientConnectionManager(
socketFactoryRegistry);
connectionManager.setDefaultMaxPerRoute(config.getMaxConnections());
connectionManager.setMaxTotal(config.getMaxConnections());
connectionManager.setValidateAfterInactivity(config.getValidateAfterInactivity());
connectionManager.setDefaultSocketConfig(
SocketConfig.custom().setSoTimeout(config.getSocketTimeout()).setTcpNoDelay(true).build());
if (config.isUseReaper()) {
IdleConnectionReaper.setIdleConnectionTime(config.getIdleConnectionTime());
IdleConnectionReaper.registerConnectionManager(connectionManager);
}
return connectionManager;
}
protected HttpClientContext createHttpContext() {
HttpClientContext httpContext = HttpClientContext.create();
httpContext.setRequestConfig(this.requestConfig);
if (this.credentialsProvider != null) {
httpContext.setCredentialsProvider(this.credentialsProvider);
httpContext.setAuthCache(this.authCache);
}
return httpContext;
}
private void setProxyAuthorizationIfNeed(HttpRequestBase httpRequest) {
if (this.credentialsProvider != null) {
String auth = this.config.getProxyUsername() + ":" + this.config.getProxyPassword();
byte[] encodedAuth = Base64.encodeBase64(auth.getBytes());
String authHeader = "Basic " + new String(encodedAuth);
httpRequest.addHeader(AUTH.PROXY_AUTH_RESP, authHeader);
}
}
@Override
public void shutdown() {
IdleConnectionReaper.removeConnectionManager(this.connectionManager);
this.connectionManager.shutdown();
}
@Override
public HttpClientConnectionManager getConnectionManager() {
return this.connectionManager;
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy