flipkart.mongo.replicator.cluster.ClusterReplicator.java Source code

Introduction

Here is the source code for flipkart.mongo.replicator.cluster.ClusterReplicator.java

Source

/*
 * Copyright 2012-2015, the original author or authors.
 *
 * 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 flipkart.mongo.replicator.cluster;

import com.google.common.util.concurrent.AbstractService;
import com.google.common.util.concurrent.Service;
import com.google.common.util.concurrent.ServiceManager;
import flipkart.mongo.replicator.core.model.Cluster;
import flipkart.mongo.replicator.core.model.ReplicaSetConfig;
import flipkart.mongo.replicator.core.model.TaskContext;
import flipkart.mongo.replicator.node.ReplicaSetReplicator;

import java.util.LinkedHashSet;
import java.util.Set;

/**
 * Created by pradeep on 29/10/14.
 */
public class ClusterReplicator extends AbstractService {

    private ServiceManager replicasReplicatorServiceManager;
    private final Cluster cluster;
    private TaskContext taskContext;

    public ClusterReplicator(Cluster cluster, TaskContext taskContext) {

        this.cluster = cluster;
        this.taskContext = taskContext;
    }

    @Override
    protected void doStart() {

        Set<Service> replicaSetServices = new LinkedHashSet<Service>();
        for (ReplicaSetConfig rsConfig : cluster.getReplicaSets()) {
            replicaSetServices.add(new ReplicaSetReplicator(taskContext, rsConfig));
        }

        /**
         * getting set of replicaSetReplicators for defined replicas and
         * attaching them to serviceManager for starting and stopping
         */
        replicasReplicatorServiceManager = new ServiceManager(replicaSetServices);
        replicasReplicatorServiceManager.addListener(new ServiceManager.Listener() {
            @Override
            public void healthy() {

            }

            @Override
            public void stopped() {

            }

            @Override
            public void failure(Service service) {
                service.startAsync();
            }
        });

        replicasReplicatorServiceManager.startAsync();
    }

    @Override
    protected void doStop() {
        replicasReplicatorServiceManager.stopAsync();
    }
}