kong.unirest.apache.ApacheClient Maven / Gradle / Ivy
The newest version!
/**
* The MIT License
*
* Copyright for portions of unirest-java are held by Kong Inc (c) 2013.
*
* Permission is hereby granted, free of charge, to any person obtaining
* a copy of this software and associated documentation files (the
* "Software"), to deal in the Software without restriction, including
* without limitation the rights to use, copy, modify, merge, publish,
* distribute, sublicense, and/or sell copies of the Software, and to
* permit persons to whom the Software is furnished to do so, subject to
* the following conditions:
*
* The above copyright notice and this permission notice shall be
* included in all copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
* NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
* LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
* OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
* WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
*/
package kong.unirest.apache;
import kong.unirest.*;
import org.apache.http.HttpHost;
import org.apache.http.client.HttpClient;
import org.apache.http.client.methods.HttpRequestBase;
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 java.io.Closeable;
import java.util.Objects;
import java.util.Optional;
import java.util.concurrent.TimeUnit;
import java.util.function.Consumer;
import java.util.function.Function;
import java.util.stream.Stream;
public class ApacheClient extends BaseApacheClient implements Client {
private final HttpClient client;
private final Config config;
private final SecurityConfig security;
private final PoolingHttpClientConnectionManager manager;
private boolean hookset;
public ApacheClient(Config config) {
this(config, a -> {
});
}
public ApacheClient(Config config, Consumer builderConfig) {
this.config = config;
security = new SecurityConfig(config);
manager = security.createManager();
HttpClientBuilder cb = HttpClients.custom();
setOptions(cb);
builderConfig.accept(cb);
client = cb.build();
}
@Deprecated // Use the builder instead, also, the PoolingHttpClientConnectionManager and SyncIdleConnectionMonitorThread don't get used here anyway
public ApacheClient(HttpClient httpClient, Config config, PoolingHttpClientConnectionManager clientManager) {
this.client = httpClient;
this.security = new SecurityConfig(config);
this.config = config;
this.manager = clientManager;
}
public ApacheClient(HttpClient httpClient, Config config) {
this.client = httpClient;
this.config = config;
this.security = new SecurityConfig(config);
this.manager = null;
}
public static Builder builder() {
return new Builder();
}
public static Builder builder(Consumer configOptions) {
return new Builder(configOptions);
}
@Override
public HttpResponse request(HttpRequest request,
Function> transformer) {
HttpRequestSummary reqSum = request.toSummary();
config.getUniInterceptor().onRequest(request, config);
HttpRequestBase requestObj = new RequestPrep(request, config, false).prepare(configFactory);
MetricContext metric = config.getMetric().begin(reqSum);
try {
HttpHost host = determineTarget(requestObj, request.getHeaders());
org.apache.http.HttpResponse execute = client.execute(host, requestObj);
ApacheResponse t = new ApacheResponse(execute, config);
metric.complete(t.toSummary(), null);
HttpResponse httpResponse = transformBody(transformer, t);
requestObj.releaseConnection();
config.getUniInterceptor().onResponse(httpResponse, reqSum, config);
return httpResponse;
} catch (Exception e) {
metric.complete(null, e);
return (HttpResponse) config.getUniInterceptor().onFail(e, reqSum, config);
} finally {
requestObj.releaseConnection();
}
}
@Override
public HttpClient getClient() {
return client;
}
public PoolingHttpClientConnectionManager getManager() {
return manager;
}
@Override
public void registerShutdownHook() {
if (!hookset) {
hookset = true;
Runtime.getRuntime().addShutdownHook(new Thread(this::close, "Unirest Apache Client Shutdown Hook"));
}
}
private void setOptions(HttpClientBuilder cb) {
cb.setDefaultRequestConfig(RequestOptions.toRequestConfig(config))
.setDefaultCredentialsProvider(toApacheCreds(config.getProxy()))
.setConnectionManager(manager)
.evictIdleConnections(30, TimeUnit.SECONDS);
security.configureSecurity(cb);
if (!config.isAutomaticRetries()) {
cb.disableAutomaticRetries();
}
if (!config.isRequestCompressionOn()) {
cb.disableContentCompression();
}
if (config.useSystemProperties()) {
cb.useSystemProperties();
}
if (!config.getFollowRedirects()) {
cb.disableRedirectHandling();
}
if (!config.getEnabledCookieManagement()) {
cb.disableCookieManagement();
}
config.getInterceptor().stream().forEach(cb::addInterceptorFirst);
if (config.shouldAddShutdownHook()) {
registerShutdownHook();
}
}
@Override
public Stream close() {
return Util.collectExceptions(Util.tryCast(client, CloseableHttpClient.class)
.map(c -> Util.tryDo(c, Closeable::close))
.filter(Optional::isPresent)
.map(Optional::get),
Util.tryDo(manager, PoolingHttpClientConnectionManager::close)
);
}
public static Builder builder(HttpClient baseClient) {
return new Builder(baseClient);
}
public static class Builder implements Function {
private final Optional baseClient;
private RequestConfigFactory configFactory;
private Consumer options = c -> {
};
public Builder(HttpClient baseClient) {
this.baseClient = Optional.ofNullable(baseClient);
}
public Builder() {
baseClient = Optional.empty();
}
public Builder(Consumer configOptions) {
Objects.requireNonNull(configOptions, "Config Options Cannot Be Null");
baseClient = Optional.empty();
this.options = configOptions;
}
@Override
public Client apply(Config config) {
ApacheClient apacheClient = baseClient.map(c -> new ApacheClient(c, config))
.orElseGet(() -> new ApacheClient(config, options));
if (configFactory != null) {
apacheClient.setConfigFactory(configFactory);
}
return apacheClient;
}
public Builder withRequestConfig(RequestConfigFactory factory) {
Objects.requireNonNull(factory);
this.configFactory = factory;
return this;
}
}
}