com.netflix.eureka2.server.channel.ReplicationTransportClient Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of eureka-write-server Show documentation
eureka-write-server developed by Netflix
The newest version!
/*
 * 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 com.netflix.eureka2.server.channel;

import com.netflix.eureka2.transport.TransportClient;
import com.netflix.eureka2.transport.EurekaTransports;
import com.netflix.eureka2.transport.EurekaTransports.Codec;
import com.netflix.eureka2.transport.MessageConnection;
import com.netflix.eureka2.transport.base.BaseMessageConnection;
import com.netflix.eureka2.transport.base.HeartBeatConnection;
import com.netflix.eureka2.metric.MessageConnectionMetrics;
import com.netflix.eureka2.transport.base.SelfClosingConnection;
import io.reactivex.netty.RxNetty;
import io.reactivex.netty.channel.ObservableConnection;
import io.reactivex.netty.client.RxClient;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import rx.Observable;
import rx.functions.Action0;
import rx.functions.Func1;
import rx.schedulers.Schedulers;

import java.net.InetSocketAddress;

/**
 * {@link ReplicationTransportClient} is always associated with single write server.
 *
 * @author Tomasz Bak
 */
public class ReplicationTransportClient implements TransportClient {

    private static final Logger logger = LoggerFactory.getLogger(ReplicationTransportClient.class);

    private final InetSocketAddress address;
    private final RxClient rxClient;
    private final MessageConnectionMetrics metrics;

    public ReplicationTransportClient(InetSocketAddress address, Codec codec, MessageConnectionMetrics metrics) {
        this.address = address;
        this.metrics = metrics;
        this.rxClient = RxNetty.newTcpClientBuilder(address.getHostName(), address.getPort())
                .pipelineConfigurator(EurekaTransports.replicationPipeline(codec))
                .build();
    }

    @Override
    public Observable connect() {
        return rxClient.connect()
                .take(1)
                .map(new Func1, MessageConnection>() {
                    @Override
                    public MessageConnection call(ObservableConnection connection) {
                        return new SelfClosingConnection(
                            new HeartBeatConnection(
                                    new BaseMessageConnection("replicationClient", connection, metrics), 30000, 3, Schedulers.computation()
                            ),
                            -1  // TODO: configure to not self terminate for now, re-evaluate later
                        );
                    }
                })
                .doOnCompleted(new Action0() {
                    @Override
                    public void call() {
                        logger.info("Connected to replication peer {}", address);
                    }
                });
    }

    @Override
    public void shutdown() {
        rxClient.shutdown();
    }
}