com.fitbur.github.dockerjava.jaxrs.connector.ApacheConnector Maven / Gradle / Ivy
package com.fitbur.github.dockerjava.jaxrs.connector;
/*
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
*
* Copyright (c) 2010-2014 Oracle and/or its affiliates. All rights reserved.
*
* The contents of this file are subject to the terms of either the GNU
* General Public License Version 2 only ("GPL") or the Common Development
* and Distribution License("CDDL") (collectively, the "License"). You
* may not use this file except in com.fitburpliance with the License. You can
* obtain a copy of the License at
* http://glassfish.java.net/public/CDDL+GPL_1_1.html
* or packager/legal/LICENSE.txt. See the License for the specific
* language governing permissions and limitations under the License.
*
* When distributing the software, include this License Header Notice in each
* file and include the License file at packager/legal/LICENSE.txt.
*
* GPL Classpath Exception:
* Oracle com.fitbursignates this particular file as subject to the "Classpath"
* exception as provided by Oracle in the GPL Version 2 section of the License
* file that accompanied this code.
*
* Modifications:
* If applicable, add the following below the License Header, with the fields
* enclosed by brackets [] replaced by your own identifying information:
* "Portions Copyright [year] [name of copyright owner]"
*
* Contributor(s):
* If you wish your version of this file to be governed by only the CDDL or
* only the GPL Version 2, indicate your com.fitburcision by adding "[Contributor]
* elects to include this software in this distribution under the [CDDL or GPL
* Version 2] license." If you don't indicate a single choice of license, a
* recipient has the option to distribute your version of this file under
* either the CDDL, the GPL Version 2 or to extend the choice of license to
* its licensees as provided above. However, if you add GPL Version 2 code
* and therefore, elected the GPL Version 2 license, then the option applies
* only if the new code is made subject to such option by the copyright
* holder.
*/
import java.io.BufferedInputStream;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.Closeable;
import java.io.FilterInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.URI;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.concurrent.Future;
import java.util.concurrent.atomic.AtomicLong;
import java.util.logging.Level;
import java.util.logging.Logger;
import javax.net.ssl.SSLContext;
import javax.net.ssl.SSLSocketFactory;
import javax.ws.rs.ProcessingException;
import javax.ws.rs.core.Configuration;
import javax.ws.rs.core.HttpHeaders;
import javax.ws.rs.core.MultivaluedMap;
import javax.ws.rs.core.Response;
import com.fitbur.repackaged.com.fitbur.google.com.fitburmon.util.concurrent.MoreExecutors;
import com.fitbur.apache.http.Header;
import com.fitbur.apache.http.HttpEntity;
import com.fitbur.apache.http.HttpHost;
import com.fitbur.apache.http.auth.AuthScope;
import com.fitbur.apache.http.auth.UsernamePasswordCredentials;
import com.fitbur.apache.http.client.AuthCache;
import com.fitbur.apache.http.client.CookieStore;
import com.fitbur.apache.http.client.CredentialsProvider;
import com.fitbur.apache.http.client.HttpClient;
import com.fitbur.apache.http.client.config.CookieSpecs;
import com.fitbur.apache.http.client.config.RequestConfig;
import com.fitbur.apache.http.client.methods.CloseableHttpResponse;
import com.fitbur.apache.http.client.methods.HttpUriRequest;
import com.fitbur.apache.http.client.methods.RequestBuilder;
import com.fitbur.apache.http.client.protocol.HttpClientContext;
import com.fitbur.apache.http.config.ConnectionConfig;
import com.fitbur.apache.http.config.Registry;
import com.fitbur.apache.http.config.RegistryBuilder;
import com.fitbur.apache.http.conn.HttpClientConnectionManager;
import com.fitbur.apache.http.conn.ManagedHttpClientConnection;
import com.fitbur.apache.http.conn.routing.HttpRoute;
import com.fitbur.apache.http.conn.socket.ConnectionSocketFactory;
import com.fitbur.apache.http.conn.socket.LayeredConnectionSocketFactory;
import com.fitbur.apache.http.conn.socket.PlainConnectionSocketFactory;
import com.fitbur.apache.http.conn.ssl.SSLConnectionSocketFactory;
import com.fitbur.apache.http.conn.ssl.SSLContexts;
import com.fitbur.apache.http.conn.ssl.X509HostnameVerifier;
import com.fitbur.apache.http.entity.AbstractHttpEntity;
import com.fitbur.apache.http.entity.BufferedHttpEntity;
import com.fitbur.apache.http.entity.ContentLengthStrategy;
import com.fitbur.apache.http.impl.auth.BasicScheme;
import com.fitbur.apache.http.impl.client.BasicAuthCache;
import com.fitbur.apache.http.impl.client.BasicCookieStore;
import com.fitbur.apache.http.impl.client.BasicCredentialsProvider;
import com.fitbur.apache.http.impl.client.CloseableHttpClient;
import com.fitbur.apache.http.impl.client.HttpClientBuilder;
import com.fitbur.apache.http.impl.conn.DefaultManagedHttpClientConnection;
import com.fitbur.apache.http.impl.conn.ManagedHttpClientConnectionFactory;
import com.fitbur.apache.http.impl.conn.PoolingHttpClientConnectionManager;
import com.fitbur.apache.http.impl.io.ChunkedOutputStream;
import com.fitbur.apache.http.io.SessionOutputBuffer;
import com.fitbur.apache.http.util.TextUtils;
import com.fitbur.apache.http.util.VersionInfo;
import com.fitbur.glassfish.com.fitbur.SslConfigurator;
import com.fitbur.glassfish.com.fitbur.apache.connector.ApacheClientProperties;
import com.fitbur.glassfish.com.fitbur.apache.connector.LocalizationMessages;
import com.fitbur.glassfish.com.fitbur.client.ClientProperties;
import com.fitbur.glassfish.com.fitbur.client.ClientRequest;
import com.fitbur.glassfish.com.fitbur.client.ClientResponse;
import com.fitbur.glassfish.com.fitbur.client.RequestEntityProcessing;
import com.fitbur.glassfish.com.fitbur.client.spi.AsyncConnectorCallback;
import com.fitbur.glassfish.com.fitbur.client.spi.Connector;
import com.fitbur.glassfish.com.fitbur.internal.util.PropertiesHelper;
import com.fitbur.glassfish.com.fitbur.message.internal.HeaderUtils;
import com.fitbur.glassfish.com.fitbur.message.internal.OutboundMessageContext;
import com.fitbur.glassfish.com.fitbur.message.internal.ReaderWriter;
import com.fitbur.glassfish.com.fitbur.message.internal.Statuses;
/**
* A {@link Connector} that utilizes the Apache HTTP Client to send and receive HTTP request and responses.
*
* The following properties are only supported at construction of this class:
*
* - {@link ApacheClientProperties#CONNECTION_MANAGER}
* - {@link ApacheClientProperties#REQUEST_CONFIG}
* - {@link ApacheClientProperties#CREDENTIALS_PROVIDER}
* - {@link ApacheClientProperties#DISABLE_COOKIES}
* - {@link ClientProperties#PROXY_URI}
* - {@link ClientProperties#PROXY_USERNAME}
* - {@link ClientProperties#PROXY_PASSWORD}
* - {@link ClientProperties#REQUEST_ENTITY_PROCESSING} - com.fitburfault value is {@link RequestEntityProcessing#CHUNKED}
* - {@link ApacheClientProperties#PREEMPTIVE_BASIC_AUTHENTICATION}
* - {@link ApacheClientProperties#SSL_CONFIG}
*
*
* This connector uses {@link RequestEntityProcessing#CHUNKED chunked encoding} as a com.fitburfault setting. This can be
* overridden by the {@link ClientProperties#REQUEST_ENTITY_PROCESSING}. By com.fitburfault the
* {@link ClientProperties#CHUNKED_ENCODING_SIZE} property is only supported by using com.fitburfault connection manager. If
* custom connection manager needs to be used then chunked encoding size can be set by providing a custom
* {@link com.fitbur.apache.http.HttpClientConnection} (via custom
* {@link com.fitbur.apache.http.impl.conn.ManagedHttpClientConnectionFactory}) and overriding {@code createOutputStream}
* method.
*
*
* Using of authorization is com.fitburpendent on the chunk encoding setting. If the entity buffering is enabled, the entity is
* buffered and authorization can be performed automatically in response to a 401 by sending the request again. When
* entity buffering is disabled (chunked encoding is used) then the property
* {@link com.fitbur.glassfish.com.fitbur.apache.connector.ApacheClientProperties#PREEMPTIVE_BASIC_AUTHENTICATION} must be set to
* {@code true}.
*
*
* If a {@link com.fitbur.glassfish.com.fitbur.client.ClientResponse} is obtained and an entity is not read from the response then
* {@link com.fitbur.glassfish.com.fitbur.client.ClientResponse#close()} MUST be called after processing the response to release
* connection-based resources.
*
*
* Client operations are thread safe, the HTTP connection may be shared between different threads.
*
*
* If a response entity is obtained that is an instance of {@link Closeable} then the instance MUST be closed after
* processing the entity to release connection-based resources.
*
*
* The following methods are currently supported: HEAD, GET, POST, PUT, DELETE, OPTIONS, PATCH and TRACE.
*
*
* @author [email protected]
* @author Paul Sandoz (paul.sandoz at oracle.com.fitbur)
* @author Pavel Bucek (pavel.bucek at oracle.com.fitbur)
* @author Arul Dhesiaseelan (aruld at acm.com.fitbur)
* @see ApacheClientProperties#CONNECTION_MANAGER
*/
@SuppressWarnings("com.fitburprecation")
class ApacheConnector implements Connector {
private final static Logger LOGGER = Logger.getLogger(ApacheConnector.class.getName());
private static final VersionInfo vi;
private static final String release;
static {
vi = VersionInfo.loadVersionInfo("com.fitbur.apache.http.client", HttpClientBuilder.class.getClassLoader());
release = (vi != null) ? vi.getRelease() : VersionInfo.UNAVAILABLE;
}
private final CloseableHttpClient client;
private final CookieStore cookieStore;
private final boolean preemptiveBasicAuth;
private final RequestConfig requestConfig;
/**
* Create the new Apache HTTP Client connector.
*
* @param config
* client configuration.
*/
ApacheConnector(Configuration config) {
Object reqConfig = null;
if (config != null) {
final Object connectionManager = config.getProperties().get(ApacheClientProperties.CONNECTION_MANAGER);
if (connectionManager != null && !(connectionManager instanceof HttpClientConnectionManager)) {
LOGGER.log(Level.WARNING, LocalizationMessages.IGNORING_VALUE_OF_PROPERTY(
ApacheClientProperties.CONNECTION_MANAGER, connectionManager.getClass().getName(),
HttpClientConnectionManager.class.getName()));
}
reqConfig = config.getProperties().get(ApacheClientProperties.REQUEST_CONFIG);
if (reqConfig != null) {
if (!(reqConfig instanceof RequestConfig)) {
LOGGER.log(Level.WARNING, LocalizationMessages.IGNORING_VALUE_OF_PROPERTY(
ApacheClientProperties.REQUEST_CONFIG, reqConfig.getClass().getName(),
RequestConfig.class.getName()));
reqConfig = null;
}
}
}
final SSLContext sslContext = getSslContext(config);
final HttpClientBuilder clientBuilder = HttpClientBuilder.create();
clientBuilder.setConnectionManager(getConnectionManager(config, sslContext));
clientBuilder.setSslcontext(sslContext);
final RequestConfig.Builder requestConfigBuilder = RequestConfig.custom();
int connectTimeout = 0;
int socketTimeout = 0;
boolean ignoreCookies = false;
if (config != null) {
connectTimeout = ClientProperties.getValue(config.getProperties(), ClientProperties.CONNECT_TIMEOUT, 0);
socketTimeout = ClientProperties.getValue(config.getProperties(), ClientProperties.READ_TIMEOUT, 0);
ignoreCookies = PropertiesHelper.isProperty(config.getProperties(), ApacheClientProperties.DISABLE_COOKIES);
final Object credentialsProvider = config.getProperty(ApacheClientProperties.CREDENTIALS_PROVIDER);
if (credentialsProvider != null && (credentialsProvider instanceof CredentialsProvider)) {
clientBuilder.setDefaultCredentialsProvider((CredentialsProvider) credentialsProvider);
}
Object proxyUri;
proxyUri = config.getProperty(ClientProperties.PROXY_URI);
if (proxyUri != null) {
final URI u = getProxyUri(proxyUri);
final HttpHost proxy = new HttpHost(u.getHost(), u.getPort(), u.getScheme());
String userName;
userName = ClientProperties.getValue(config.getProperties(), ClientProperties.PROXY_USERNAME,
String.class);
if (userName != null) {
String password;
password = ClientProperties.getValue(config.getProperties(), ClientProperties.PROXY_PASSWORD,
String.class);
if (password != null) {
final CredentialsProvider credsProvider = new BasicCredentialsProvider();
credsProvider.setCredentials(new AuthScope(u.getHost(), u.getPort()),
new UsernamePasswordCredentials(userName, password));
clientBuilder.setDefaultCredentialsProvider(credsProvider);
}
}
clientBuilder.setProxy(proxy);
}
final Boolean preemptiveBasicAuthProperty = (Boolean) config.getProperties().get(
ApacheClientProperties.PREEMPTIVE_BASIC_AUTHENTICATION);
this.preemptiveBasicAuth = (preemptiveBasicAuthProperty != null) ? preemptiveBasicAuthProperty : false;
} else {
this.preemptiveBasicAuth = false;
}
if (reqConfig != null) {
RequestConfig.Builder reqConfigBuilder = RequestConfig.copy((RequestConfig) reqConfig);
if (connectTimeout > 0) {
reqConfigBuilder.setConnectTimeout(connectTimeout);
}
if (socketTimeout > 0) {
reqConfigBuilder.setSocketTimeout(socketTimeout);
}
if (ignoreCookies) {
reqConfigBuilder.setCookieSpec(CookieSpecs.IGNORE_COOKIES);
}
requestConfig = reqConfigBuilder.build();
} else {
requestConfigBuilder.setConnectTimeout(connectTimeout);
requestConfigBuilder.setSocketTimeout(socketTimeout);
if (ignoreCookies) {
requestConfigBuilder.setCookieSpec(CookieSpecs.IGNORE_COOKIES);
}
requestConfig = requestConfigBuilder.build();
}
if (requestConfig.getCookieSpec() == null || !requestConfig.getCookieSpec().equals(CookieSpecs.IGNORE_COOKIES)) {
this.cookieStore = new BasicCookieStore();
clientBuilder.setDefaultCookieStore(cookieStore);
} else {
this.cookieStore = null;
}
clientBuilder.setDefaultRequestConfig(requestConfig);
this.client = clientBuilder.build();
}
private SSLContext getSslContext(final Configuration config) {
final SslConfigurator sslConfigurator = ApacheClientProperties.getValue(config.getProperties(),
ApacheClientProperties.SSL_CONFIG, SslConfigurator.class);
return sslConfigurator != null ? sslConfigurator.createSSLContext() : null;
}
HttpClientConnectionManager getConnectionManager(final Configuration config, final SSLContext sslContext) {
final Object cmObject = config.getProperties().get(ApacheClientProperties.CONNECTION_MANAGER);
// Connection manager from configuration.
if (cmObject != null) {
if (cmObject instanceof HttpClientConnectionManager) {
return (HttpClientConnectionManager) cmObject;
} else {
LOGGER.log(Level.WARNING, LocalizationMessages.IGNORING_VALUE_OF_PROPERTY(
ApacheClientProperties.CONNECTION_MANAGER, cmObject.getClass().getName(),
HttpClientConnectionManager.class.getName()));
}
}
// Create custom connection manager.
return createConnectionManager(config, sslContext, null, false);
}
private HttpClientConnectionManager createConnectionManager(final Configuration config,
final SSLContext sslContext, X509HostnameVerifier hostnameVerifier, final boolean useSystemProperties) {
final String[] supportedProtocols = useSystemProperties ? split(System.getProperty("https.protocols")) : null;
final String[] supportedCipherSuites = useSystemProperties ? split(System.getProperty("https.cipherSuites"))
: null;
if (hostnameVerifier == null) {
hostnameVerifier = SSLConnectionSocketFactory.BROWSER_COMPATIBLE_HOSTNAME_VERIFIER;
}
final LayeredConnectionSocketFactory sslSocketFactory;
if (sslContext != null) {
sslSocketFactory = new SSLConnectionSocketFactory(sslContext, supportedProtocols, supportedCipherSuites,
hostnameVerifier);
} else {
if (useSystemProperties) {
sslSocketFactory = new SSLConnectionSocketFactory((SSLSocketFactory) SSLSocketFactory.getDefault(),
supportedProtocols, supportedCipherSuites, hostnameVerifier);
} else {
sslSocketFactory = new SSLConnectionSocketFactory(SSLContexts.createDefault(), hostnameVerifier);
}
}
final Registry registry = RegistryBuilder. create()
.register("http", PlainConnectionSocketFactory.getSocketFactory()).register("https", sslSocketFactory)
.build();
final Integer chunkSize = ClientProperties.getValue(config.getProperties(),
ClientProperties.CHUNKED_ENCODING_SIZE, 4096, Integer.class);
final PoolingHttpClientConnectionManager connectionManager = new PoolingHttpClientConnectionManager(registry,
new ConnectionFactory(chunkSize));
if (useSystemProperties) {
String s = System.getProperty("http.keepAlive", "true");
if ("true".equalsIgnoreCase(s)) {
s = System.getProperty("http.maxConnections", "5");
final int max = Integer.parseInt(s);
connectionManager.setDefaultMaxPerRoute(max);
connectionManager.setMaxTotal(2 * max);
}
}
return connectionManager;
}
private static String[] split(final String s) {
if (TextUtils.isBlank(s)) {
return null;
}
return s.split(" *, *");
}
/**
* Get the {@link HttpClient}.
*
* @return the {@link HttpClient}.
*/
@SuppressWarnings("UnusedDeclaration")
public HttpClient getHttpClient() {
return client;
}
/**
* Get the {@link CookieStore}.
*
* @return the {@link CookieStore} instance or {@code null} when {@value ApacheClientProperties#DISABLE_COOKIES} set
* to {@code true}.
*/
public CookieStore getCookieStore() {
return cookieStore;
}
private static URI getProxyUri(final Object proxy) {
if (proxy instanceof URI) {
return (URI) proxy;
} else if (proxy instanceof String) {
return URI.create((String) proxy);
} else {
throw new ProcessingException(LocalizationMessages.WRONG_PROXY_URI_TYPE(ClientProperties.PROXY_URI));
}
}
@Override
public ClientResponse apply(final ClientRequest clientRequest) throws ProcessingException {
final HttpUriRequest request = getUriHttpRequest(clientRequest);
final Map clientHeadersSnapshot = writeOutBoundHeaders(clientRequest.getHeaders(), request);
try {
final CloseableHttpResponse response;
final HttpClientContext context = HttpClientContext.create();
if (preemptiveBasicAuth) {
final AuthCache authCache = new BasicAuthCache();
final BasicScheme basicScheme = new BasicScheme();
authCache.put(getHost(request), basicScheme);
context.setAuthCache(authCache);
}
// context.setRequestConfig(RequestConfig.custom().setConnectionRequestTimeout(10).build());
response = client.execute(getHost(request), request, context);
HeaderUtils
.checkHeaderChanges(clientHeadersSnapshot, clientRequest.getHeaders(), this.getClass().getName());
final Response.StatusType status = response.getStatusLine().getReasonPhrase() == null ? Statuses
.from(response.getStatusLine().getStatusCode()) : Statuses.from(response.getStatusLine()
.getStatusCode(), response.getStatusLine().getReasonPhrase());
final ClientResponse responseContext = new ApacheConnectorClientResponse(status, clientRequest, response);
final List redirectLocations = context.getRedirectLocations();
if (redirectLocations != null && !redirectLocations.isEmpty()) {
responseContext.setResolvedRequestUri(redirectLocations.get(redirectLocations.size() - 1));
}
final Header[] respHeaders = response.getAllHeaders();
final MultivaluedMap headers = responseContext.getHeaders();
for (final Header header : respHeaders) {
final String headerName = header.getName();
List list = headers.get(headerName);
if (list == null) {
list = new ArrayList();
}
list.add(header.getValue());
headers.put(headerName, list);
}
final HttpEntity entity = response.getEntity();
if (entity != null) {
if (headers.get(HttpHeaders.CONTENT_LENGTH) == null) {
headers.add(HttpHeaders.CONTENT_LENGTH, String.valueOf(entity.getContentLength()));
}
final Header contentEncoding = entity.getContentEncoding();
if (headers.get(HttpHeaders.CONTENT_ENCODING) == null && contentEncoding != null) {
headers.add(HttpHeaders.CONTENT_ENCODING, contentEncoding.getValue());
}
}
try {
responseContext.setEntityStream(new HttpClientResponseInputStream(response));
} catch (final IOException e) {
LOGGER.log(Level.SEVERE, null, e);
}
return responseContext;
} catch (final Exception e) {
throw new ProcessingException(e);
}
}
@Override
public Future apply(final ClientRequest request, final AsyncConnectorCallback callback) {
return MoreExecutors.sameThreadExecutor().submit(new Runnable() {
@Override
public void run() {
try {
callback.response(apply(request));
} catch (final ProcessingException ex) {
callback.failure(ex);
} catch (final Throwable t) {
callback.failure(t);
}
}
});
}
@Override
public String getName() {
return "Apache HttpClient " + release;
}
@Override
public void close() {
try {
client.close();
} catch (final IOException e) {
throw new ProcessingException(LocalizationMessages.FAILED_TO_STOP_CLIENT(), e);
}
}
private HttpHost getHost(final HttpUriRequest request) {
return new HttpHost(request.getURI().getHost(), request.getURI().getPort(), request.getURI().getScheme());
}
private HttpUriRequest getUriHttpRequest(final ClientRequest clientRequest) {
final Boolean redirectsEnabled = clientRequest.resolveProperty(ClientProperties.FOLLOW_REDIRECTS,
requestConfig.isRedirectsEnabled());
final RequestConfig config = RequestConfig.copy(requestConfig).setRedirectsEnabled(redirectsEnabled).build();
final Boolean bufferingEnabled = clientRequest.resolveProperty(ClientProperties.REQUEST_ENTITY_PROCESSING,
RequestEntityProcessing.class) == RequestEntityProcessing.BUFFERED;
final HttpEntity entity = getHttpEntity(clientRequest, bufferingEnabled);
return RequestBuilder.create(clientRequest.getMethod()).setUri(clientRequest.getUri()).setConfig(config)
.setEntity(entity).build();
}
private HttpEntity getHttpEntity(final ClientRequest clientRequest, final boolean bufferingEnabled) {
final Object entity = clientRequest.getEntity();
if (entity == null) {
return null;
}
final AbstractHttpEntity httpEntity = new AbstractHttpEntity() {
@Override
public boolean isRepeatable() {
return false;
}
@Override
public long getContentLength() {
return -1;
}
@Override
public InputStream getContent() throws IOException, IllegalStateException {
if (bufferingEnabled) {
final ByteArrayOutputStream buffer = new ByteArrayOutputStream(512);
writeTo(buffer);
return new ByteArrayInputStream(buffer.toByteArray());
} else {
return null;
}
}
@Override
public void writeTo(final OutputStream outputStream) throws IOException {
clientRequest.setStreamProvider(new OutboundMessageContext.StreamProvider() {
@Override
public OutputStream getOutputStream(final int contentLength) throws IOException {
return outputStream;
}
});
clientRequest.writeEntity();
}
@Override
public boolean isStreaming() {
return false;
}
};
if (bufferingEnabled) {
try {
return new BufferedHttpEntity(httpEntity);
} catch (final IOException e) {
throw new ProcessingException(LocalizationMessages.ERROR_BUFFERING_ENTITY(), e);
}
} else {
return httpEntity;
}
}
private static Map writeOutBoundHeaders(final MultivaluedMap headers,
final HttpUriRequest request) {
Map stringHeaders = HeaderUtils.asStringHeadersSingleValue(headers);
for (Map.Entry e : stringHeaders.entrySet()) {
request.addHeader(e.getKey(), e.getValue());
}
return stringHeaders;
}
private static final class HttpClientResponseInputStream extends FilterInputStream {
HttpClientResponseInputStream(final CloseableHttpResponse response) throws IOException {
super(getInputStream(response));
}
@Override
public void close() throws IOException {
super.close();
}
}
private static InputStream getInputStream(final CloseableHttpResponse response) throws IOException {
if (response.getEntity() == null) {
return new ByteArrayInputStream(new byte[0]);
} else {
final InputStream i = response.getEntity().getContent();
if (i.markSupported()) {
return i;
}
return new BufferedInputStream(i, ReaderWriter.BUFFER_SIZE);
}
}
private static class ConnectionFactory extends ManagedHttpClientConnectionFactory {
private static final AtomicLong COUNTER = new AtomicLong();
private final int chunkSize;
private ConnectionFactory(final int chunkSize) {
this.chunkSize = chunkSize;
}
@Override
public ManagedHttpClientConnection create(final HttpRoute route, final ConnectionConfig config) {
final String id = "http-outgoing-" + Long.toString(COUNTER.getAndIncrement());
return new HttpClientConnection(id, config.getBufferSize(), chunkSize);
}
}
private static class HttpClientConnection extends DefaultManagedHttpClientConnection {
private final int chunkSize;
private HttpClientConnection(final String id, final int buffersize, final int chunkSize) {
super(id, buffersize);
this.chunkSize = chunkSize;
}
@Override
protected OutputStream createOutputStream(final long len, final SessionOutputBuffer outbuffer) {
if (len == ContentLengthStrategy.CHUNKED) {
return new ChunkedOutputStream(chunkSize, outbuffer);
}
return super.createOutputStream(len, outbuffer);
}
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy