com.windowsazure.messaging.HttpClientManager Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of Notification-Hubs-java-sdk Show documentation
Azure Notification Hubs Java SDK for interacting with the data and management plane operations.
The newest version!
//----------------------------------------------------------------
// Copyright (c) Microsoft Corporation. All rights reserved.
//----------------------------------------------------------------

package com.windowsazure.messaging;

import org.apache.hc.client5.http.HttpRequestRetryStrategy;
import org.apache.hc.client5.http.config.RequestConfig;
import org.apache.hc.client5.http.impl.async.CloseableHttpAsyncClient;
import org.apache.hc.client5.http.impl.async.HttpAsyncClients;
import org.apache.hc.core5.reactor.IOReactorConfig;
import org.apache.hc.core5.reactor.IOReactorStatus;
import org.apache.hc.core5.util.Timeout;

/**
 * This class manages the interaction with the HTTP async client.
 */
public class HttpClientManager {

    private static CloseableHttpAsyncClient httpAsyncClient;

    private static final int DEFAULT_WAIT_TIMEOUT_MILLISECONDS = (60 * 1000);
    private static final int DEFAULT_CONNECTION_TIMEOUT_MILLISECONDS = (60 * 1000);

    // The timeout in milliseconds used when requesting a connection from the connection manager.
    private static int connectionRequestTimeout = DEFAULT_WAIT_TIMEOUT_MILLISECONDS;

    // The timeout in milliseconds until a connection is established.
    private static int connectionTimeout = DEFAULT_CONNECTION_TIMEOUT_MILLISECONDS;

    private static HttpRequestRetryStrategy retryStrategy = BasicRetryStrategy.INSTANCE;

    private static void initializeHttpAsyncClient() {
        synchronized (HttpClientManager.class) {
            if (httpAsyncClient == null) {
                final IOReactorConfig ioReactorConfig = IOReactorConfig.custom()
                    .setSoTimeout(Timeout.ofSeconds(5))
                    .build();

                final RequestConfig config = RequestConfig.custom()
                    .setConnectionRequestTimeout(Timeout.ofMilliseconds(connectionRequestTimeout))
                    .setConnectTimeout(Timeout.ofMilliseconds(connectionTimeout))
                    .build();

                final CloseableHttpAsyncClient client = HttpAsyncClients.customHttp2()
                    .setIOReactorConfig(ioReactorConfig)
                    .setDefaultRequestConfig(config)
                    .setRetryStrategy(retryStrategy)
                    .build();

                client.start();
                httpAsyncClient = client;
            }
        }
    }

    /**
     * Gets the current HTTP async client.
     * @return The current HTTP async client.
     */
    public static CloseableHttpAsyncClient getHttpAsyncClient() {
        if (HttpClientManager.httpAsyncClient == null) {
            initializeHttpAsyncClient();
        }
        return HttpClientManager.httpAsyncClient;
    }

    /**
     * Sets the current HTTP async client.
     * @param httpAsyncClient The HTTP async client to set.
     */
    public static void setHttpAsyncClient(CloseableHttpAsyncClient httpAsyncClient) {
        synchronized (HttpClientManager.class) {
            if (HttpClientManager.httpAsyncClient == null) {
                HttpClientManager.httpAsyncClient = httpAsyncClient;

                if (HttpClientManager.httpAsyncClient.getStatus() == IOReactorStatus.ACTIVE) {
                    HttpClientManager.httpAsyncClient.start();
                }
            } else {
                throw new RuntimeException("Cannot setHttpAsyncClient after having previously set, or after default already initialized from earlier call to getHttpAsyncClient.");
            }
        }
    }

    /**
     * Sets the timeout in milliseconds used when requesting a connection from the connection manager.
     * @param timeout The timeout in milliseconds used when requesting a connection from the connection manager.
     */
    public static void setConnectionRequestTimeout(int timeout) {
        if (HttpClientManager.httpAsyncClient == null) {
            connectionRequestTimeout = timeout;
        } else {
            throw new RuntimeException("Cannot setConnectionRequestTimeout after previously setting httpAsyncClient, or after default already initialized from earlier call to getHttpAsyncClient.");
        }
    }

    /**
     * Sets the timeout in milliseconds until a connection is established.
     * @param timeout The timeout in milliseconds to set.
     */
    public static void setConnectTimeout(int timeout) {
        if (HttpClientManager.httpAsyncClient == null) {
            connectionTimeout = timeout;
        } else {
            throw new RuntimeException("Cannot setConnectTimeout after previously setting httpAsyncClient, or after default already initialized from earlier call to getHttpAsyncClient.");
        }
    }

    /**
     * Sets the retry strategy for the HTTP client.
     * @param strategy The retry strategy for the HTTP client.
     */
    public static void setRetryStrategy(HttpRequestRetryStrategy strategy) {
        if (HttpClientManager.httpAsyncClient == null) {
            retryStrategy = strategy;
        } else {
            throw new RuntimeException("Cannot setRetryStrategy after previously setting httpAsyncClient, or after default already initialized from earlier call to getHttpAsyncClient.");
        }
    }
}