io.reactivex.netty.protocol.http.client.CompositeHttpClientBuilder Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of rx-netty Show documentation
rx-netty developed by Netflix
There is a newer version: 0.3.18
/*
 * Copyright 2014 Netflix, 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 io.reactivex.netty.protocol.http.client;

import io.netty.bootstrap.Bootstrap;
import io.reactivex.netty.channel.ObservableConnection;
import io.reactivex.netty.client.AbstractClientBuilder;
import io.reactivex.netty.client.ClientChannelFactory;
import io.reactivex.netty.client.ClientChannelFactoryImpl;
import io.reactivex.netty.client.ClientConnectionFactory;
import io.reactivex.netty.client.ClientMetricsEvent;
import io.reactivex.netty.client.ConnectionPoolBuilder;
import io.reactivex.netty.client.PoolLimitDeterminationStrategy;
import io.reactivex.netty.client.RxClient;
import io.reactivex.netty.client.UnpooledClientConnectionFactory;
import io.reactivex.netty.metrics.MetricEventsListener;
import io.reactivex.netty.metrics.MetricEventsListenerFactory;
import io.reactivex.netty.pipeline.PipelineConfigurators;

/**
 * @param  The type of the content of request.
 * @param  The type of the content of response.
 *
 * @author Nitesh Kant
 */
public class CompositeHttpClientBuilder
        extends AbstractClientBuilder, HttpClientResponse, CompositeHttpClientBuilder,
        CompositeHttpClient> {

    private static final RxClient.ServerInfo defaultServer = new RxClient.ServerInfo();

    public CompositeHttpClientBuilder() {
        this(new Bootstrap());
    }

    public CompositeHttpClientBuilder(Bootstrap bootstrap) {
        this(bootstrap, new UnpooledClientConnectionFactory, HttpClientRequest>(),
             new ClientChannelFactoryImpl, HttpClientRequest>(bootstrap));
    }

    public CompositeHttpClientBuilder(Bootstrap bootstrap,
                                      ClientConnectionFactory, HttpClientRequest,
                                              ? extends ObservableConnection, HttpClientRequest>> connectionFactory,
                                      ClientChannelFactory, HttpClientRequest> factory) {
        super(bootstrap, defaultServer.getHost(), defaultServer.getPort(), connectionFactory, factory);
        clientConfig = HttpClient.HttpClientConfig.Builder.newDefaultConfig();
        pipelineConfigurator(PipelineConfigurators.httpClientConfigurator());
    }

    public CompositeHttpClientBuilder(Bootstrap bootstrap,
                                      ConnectionPoolBuilder, HttpClientRequest> poolBuilder) {
        super(bootstrap, defaultServer.getHost(), defaultServer.getPort(), poolBuilder);
        clientConfig = HttpClient.HttpClientConfig.Builder.newDefaultConfig();
        pipelineConfigurator(PipelineConfigurators.httpClientConfigurator());
    }

    @Override
    public CompositeHttpClientBuilder withConnectionPoolLimitStrategy(PoolLimitDeterminationStrategy strategy) {
        if (strategy instanceof CloneablePoolLimitDeterminationStrategy) {
            return withConnectionPoolLimitStrategy((CloneablePoolLimitDeterminationStrategy) strategy);
        } else {
            throw new IllegalArgumentException("Only " + CloneablePoolLimitDeterminationStrategy.class.getName() +
                                               " strategy implementations are allowed.");
        }
    }

    public CompositeHttpClientBuilder withConnectionPoolLimitStrategy(CloneablePoolLimitDeterminationStrategy strategy) {
        super.withConnectionPoolLimitStrategy(strategy);
        return this;
    }

    @Override
    public CompositeHttpClientBuilder withMaxConnections(int maxConnections) {
        return super.withMaxConnections(maxConnections);
    }

    @Override
    protected CompositeHttpClient createClient() {
        if (null == poolBuilder) {
            return new CompositeHttpClient(getOrCreateName(), serverInfo, bootstrap, pipelineConfigurator,
                                                 clientConfig, channelFactory, connectionFactory, eventsSubject);
        } else {
            return new CompositeHttpClient(getOrCreateName(), serverInfo, bootstrap, pipelineConfigurator,
                                                 clientConfig, poolBuilder, eventsSubject);
        }
    }

    @Override
    protected String generatedNamePrefix() {
        return "HttpClient-";
    }

    @Override
    protected MetricEventsListener>
    newMetricsListener(MetricEventsListenerFactory factory, CompositeHttpClient client) {
        return factory.forHttpClient(client);
    }

    public interface CloneablePoolLimitDeterminationStrategy extends PoolLimitDeterminationStrategy {

        CloneablePoolLimitDeterminationStrategy copy();
    }
}