com.netflix.eureka2.client.channel.ClientChannelFactoryImpl Maven / Gradle / Ivy

Go to download
package com.netflix.eureka2.client.channel;

import com.netflix.eureka2.client.metric.EurekaClientMetricFactory;
import com.netflix.eureka2.client.registry.EurekaClientRegistry;
import com.netflix.eureka2.client.transport.TransportClient;
import com.netflix.eureka2.registry.InstanceInfo;
import com.netflix.eureka2.service.InterestChannel;
import com.netflix.eureka2.service.RegistrationChannel;
import rx.functions.Func0;
import rx.schedulers.Schedulers;

/**
 * @author Nitesh Kant
 */
public class ClientChannelFactoryImpl implements ClientChannelFactory {

    private final TransportClient readServerClient; /*Null for write server only service*/
    private final TransportClient writeServerClient; /*Null for read server only service*/
    private final EurekaClientMetricFactory metricFactory;
    private final Mode channelMode;
    private final long retryInitialDelayMs;

    public ClientChannelFactoryImpl(TransportClient writeServerClient,
                                    TransportClient readServerClient,
                                    long retryInitialDelayMs,
                                    EurekaClientMetricFactory metricFactory) {
        this.retryInitialDelayMs = retryInitialDelayMs;
        if (writeServerClient == null && readServerClient == null) {
            throw new IllegalArgumentException("Both read and write transport clients are null");
        }
        this.writeServerClient = writeServerClient;
        this.readServerClient = readServerClient;
        this.metricFactory = metricFactory;
        this.channelMode = writeServerClient == null ? Mode.Read : readServerClient == null ? Mode.Write : Mode.ReadWrite;
    }

    /**
     * Returns an {@link InterestChannel} which is not yet connected to any eureka servers. The connection is done
     * lazily when any operation is invoked on the channel.
     *
     * This makes it possible for clients to create this channel eagerly and use it when required.
     *
     * @return An {@link InterestChannel} which is not yet connected to any eureka servers.
     */
    @Override
    public ClientInterestChannel newInterestChannel(EurekaClientRegistry clientRegistry) {
        return new InterestChannelInvoker(new InterestChannelImpl(clientRegistry, readServerClient, metricFactory.getInterestChannelMetrics()));
    }


    /**
     * Returns an {@link RegistrationChannel} which is not yet connected to any eureka servers. The connection is done
     * lazily when any operation is invoked on the channel.
     *
     * This makes it possible for clients to create this channel eagerly and use it when required.
     *
     * @return An {@link RegistrationChannel} which is not yet connected to any eureka servers.
     */
    @Override
    public RegistrationChannel newRegistrationChannel() {
        return new RegistrationChannelInvoker(
                new RetryableRegistrationChannel(new Func0() {
                    @Override
                    public RegistrationChannel call() {
                        return new RegistrationChannelImpl(writeServerClient, metricFactory.getRegistrationChannelMetrics());
                    }
                }, retryInitialDelayMs, Schedulers.computation())
        );
    }

    @Override
    public Mode mode() {
        return channelMode;
    }

    @Override
    public void shutdown() {
        if (null != readServerClient) {
            readServerClient.shutdown();
        }
        if (null != writeServerClient) {
            writeServerClient.shutdown();
        }
    }
}