diff options
author | jonmv <venstad@gmail.com> | 2022-04-30 07:37:53 +0200 |
---|---|---|
committer | jonmv <venstad@gmail.com> | 2022-05-02 12:02:02 +0200 |
commit | d0621873642dfaba4ddfe3bae03a30f58fc7261c (patch) | |
tree | e1ab55b273b6ddc4453733a51b8e1b5e57b0c8db /http-utils | |
parent | eaee8f4042eb135a2c3f20ecbeba5459175dd2de (diff) |
Add DefaultHttpClientBuidler for easy SSLContext
Diffstat (limited to 'http-utils')
-rw-r--r-- | http-utils/src/main/java/ai/vespa/util/http/hc5/DefaultHttpClientBuilder.java | 49 |
1 files changed, 49 insertions, 0 deletions
diff --git a/http-utils/src/main/java/ai/vespa/util/http/hc5/DefaultHttpClientBuilder.java b/http-utils/src/main/java/ai/vespa/util/http/hc5/DefaultHttpClientBuilder.java new file mode 100644 index 00000000000..8ad9d63cd1a --- /dev/null +++ b/http-utils/src/main/java/ai/vespa/util/http/hc5/DefaultHttpClientBuilder.java @@ -0,0 +1,49 @@ +package ai.vespa.util.http.hc5; + +import org.apache.hc.client5.http.config.RequestConfig; +import org.apache.hc.client5.http.impl.classic.HttpClientBuilder; +import org.apache.hc.client5.http.impl.classic.HttpClients; +import org.apache.hc.client5.http.impl.io.PoolingHttpClientConnectionManagerBuilder; +import org.apache.hc.client5.http.ssl.SSLConnectionSocketFactoryBuilder; +import org.apache.hc.core5.http.ContentType; +import org.apache.hc.core5.http.HttpHeaders; +import org.apache.hc.core5.util.Timeout; + +import javax.net.ssl.SSLContext; +import java.time.Duration; +import java.util.Map; +import java.util.function.Supplier; + +/** + * Like {@link VespaHttpClientBuilder}, but with standard TLS based on provided SSL context. + * + * @author jonmv + */ +public class DefaultHttpClientBuilder { + + public static final Duration connectTimeout = Duration.ofSeconds(5); + public static final Duration socketTimeout = Duration.ofSeconds(5); + + private DefaultHttpClientBuilder() { } + + public static HttpClientBuilder create(SSLContext sslContext, String userAgent) { + return create(() -> sslContext, userAgent); + } + + /** Creates an HTTP client builder with the given SSL context, and using the provided timeouts for requests where config is not overridden. */ + public static HttpClientBuilder create(Supplier<SSLContext> sslContext, String userAgent) { + return HttpClientBuilder.create() + .setConnectionManager(PoolingHttpClientConnectionManagerBuilder + .create() + .setSSLSocketFactory(SSLConnectionSocketFactoryBuilder + .create() + .setSslContext(sslContext.get()) + .build()) + .build()) + .setUserAgent(userAgent) + .disableCookieManagement() + .disableAutomaticRetries() + .disableAuthCaching(); + } + +} |