com.metamx.http.client.HttpClientConfig.java Source code

Introduction

Here is the source code for com.metamx.http.client.HttpClientConfig.java

Source

/*
 * Copyright 2011 - 2015 Metamarkets Group Inc.
 *
 * Licensed 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.metamx.http.client;

import org.joda.time.Duration;

import javax.net.ssl.SSLContext;

/**
 */
public class HttpClientConfig {
    // Default from NioClientSocketChannelFactory.DEFAULT_BOSS_COUNT, which is private:
    private static final int DEFAULT_BOSS_COUNT = 1;

    // Default from SelectorUtil.DEFAULT_IO_THREADS, which is private:
    private static final int DEFAULT_WORKER_COUNT = Runtime.getRuntime().availableProcessors() * 2;

    public static Builder builder() {
        return new Builder();
    }

    private final int numConnections;
    private final SSLContext sslContext;
    private final Duration readTimeout;
    private final Duration sslHandshakeTimeout;
    private final int bossPoolSize;
    private final int workerPoolSize;

    @Deprecated // Use the builder instead
    public HttpClientConfig(int numConnections, SSLContext sslContext) {
        this(numConnections, sslContext, Duration.ZERO, null, DEFAULT_BOSS_COUNT, DEFAULT_WORKER_COUNT);
    }

    @Deprecated // Use the builder instead
    public HttpClientConfig(int numConnections, SSLContext sslContext, Duration readTimeout) {
        this(numConnections, sslContext, readTimeout, null, DEFAULT_BOSS_COUNT, DEFAULT_WORKER_COUNT);
    }

    @Deprecated // Use the builder instead
    public HttpClientConfig(int numConnections, SSLContext sslContext, Duration readTimeout,
            Duration sslHandshakeTimeout) {
        this(numConnections, sslContext, readTimeout, sslHandshakeTimeout, DEFAULT_BOSS_COUNT,
                DEFAULT_WORKER_COUNT);
    }

    private HttpClientConfig(int numConnections, SSLContext sslContext, Duration readTimeout,
            Duration sslHandshakeTimeout, int bossPoolSize, int workerPoolSize) {
        this.numConnections = numConnections;
        this.sslContext = sslContext;
        this.readTimeout = readTimeout;
        this.sslHandshakeTimeout = sslHandshakeTimeout;
        this.bossPoolSize = bossPoolSize;
        this.workerPoolSize = workerPoolSize;
    }

    public int getNumConnections() {
        return numConnections;
    }

    public SSLContext getSslContext() {
        return sslContext;
    }

    public Duration getReadTimeout() {
        return readTimeout;
    }

    public Duration getSslHandshakeTimeout() {
        return sslHandshakeTimeout;
    }

    public int getBossPoolSize() {
        return bossPoolSize;
    }

    public int getWorkerPoolSize() {
        return workerPoolSize;
    }

    public static class Builder {
        private int numConnections = 1;
        private SSLContext sslContext = null;
        private Duration readTimeout = null;
        private Duration sslHandshakeTimeout = null;
        private int bossCount = DEFAULT_BOSS_COUNT;
        private int workerCount = DEFAULT_WORKER_COUNT;

        private Builder() {
        }

        public Builder withNumConnections(int numConnections) {
            this.numConnections = numConnections;
            return this;
        }

        public Builder withSslContext(SSLContext sslContext) {
            this.sslContext = sslContext;
            return this;
        }

        public Builder withSslContext(String keyStorePath, String keyStorePassword) {
            this.sslContext = HttpClientInit.sslContextWithTrustedKeyStore(keyStorePath, keyStorePassword);
            return this;
        }

        public Builder withReadTimeout(Duration readTimeout) {
            this.readTimeout = readTimeout;
            return this;
        }

        public Builder withSslHandshakeTimeout(Duration sslHandshakeTimeout) {
            this.sslHandshakeTimeout = sslHandshakeTimeout;
            return this;
        }

        public Builder withBossCount(int bossCount) {
            this.bossCount = bossCount;
            return this;
        }

        public Builder withWorkerCount(int workerCount) {
            this.workerCount = workerCount;
            return this;
        }

        public HttpClientConfig build() {
            return new HttpClientConfig(numConnections, sslContext, readTimeout, sslHandshakeTimeout, bossCount,
                    workerCount);
        }
    }
}