com.openteach.diamond.network.waverider.master.DefaultMasterNode.java Source code

Introduction

Here is the source code for com.openteach.diamond.network.waverider.master.DefaultMasterNode.java

Source

/**
 * Copyright 2013 openteach
 *
 *  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.openteach.diamond.network.waverider.master;

import java.util.concurrent.atomic.AtomicLong;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;

import com.openteach.diamond.network.waverider.MasterNode;
import com.openteach.diamond.network.waverider.State;
import com.openteach.diamond.network.waverider.command.CommandDispatcher;
import com.openteach.diamond.network.waverider.command.CommandHandler;
import com.openteach.diamond.network.waverider.command.MasterHeartbeatCommandHandler;
import com.openteach.diamond.network.waverider.command.SampleCommandDispatcher;
import com.openteach.diamond.network.waverider.config.WaveriderConfig;
import com.openteach.diamond.network.waverider.network.DefaultNetWorkServer;
import com.openteach.diamond.network.waverider.network.NetWorkServer;
import com.openteach.diamond.network.waverider.session.DefaultSessionManager;
import com.openteach.diamond.network.waverider.session.Session;
import com.openteach.diamond.network.waverider.session.SessionListener;
import com.openteach.diamond.network.waverider.session.SessionManager;

/**
 * <p>
 * ?Master
 * </p>
 * 
 * @author <a href="mailto:sihai@taobao.com">sihai</a>
 *
 */
public class DefaultMasterNode implements MasterNode {

    private static final Log logger = LogFactory.getLog(DefaultMasterNode.class);

    private WaveriderConfig config; // ??
    private SessionManager sessionManager; // Slave session ?
    private volatile SlaveListenter listener; // ?Slave?
    private NetWorkServer netWorkServer; // ?
    private CommandDispatcher commandDispatcher; // ?
    private AtomicLong stateIDGenrator = new AtomicLong(0); // ?

    public DefaultMasterNode(WaveriderConfig config) {
        this.config = config;
        this.commandDispatcher = new SampleCommandDispatcher();
    }

    @Override
    public boolean init() {
        commandDispatcher.addCommandHandler(0L, new MasterHeartbeatCommandHandler(this));
        netWorkServer = new DefaultNetWorkServer(config.getPort());
        sessionManager = new DefaultSessionManager(config);
        sessionManager.setCommandDispatcher(commandDispatcher);
        sessionManager.registerSessionListener(new SessionListenerSupport());
        netWorkServer.setSessionManager(sessionManager);
        return netWorkServer.init() && sessionManager.init();
    }

    @Override
    public boolean start() {
        return netWorkServer.start() && sessionManager.start();
    }

    @Override
    public boolean stop() {
        return sessionManager.stop() && netWorkServer.stop();
    }

    @Override
    public boolean restart() {
        return sessionManager.restart() && netWorkServer.restart();
    }

    @Override
    public void addCommandHandler(Long command, CommandHandler handler) {
        if (command == null || command.equals(0L)) {
            throw new IllegalArgumentException("command must not be null or 0");
        }
        commandDispatcher.addCommandHandler(command, handler);
    }

    @Override
    public State gatherStatistics() {
        MasterState masterState = new MasterState();
        masterState.setId(stateIDGenrator.addAndGet(1));
        masterState.setIp(netWorkServer.getIp());
        masterState.setPort(netWorkServer.getPort());
        masterState.setSessionStateList(sessionManager.generateSessionState());
        return masterState;
    }

    @Override
    public void acceptStatistics(State state) {
        //logger.info(new StringBuilder("Master Accept Slave state : ").append(((SlaveState)state).toString()).toString());
    }

    @Override
    public void registerSlaveListener(SlaveListenter listener) {
        this.listener = listener;
    }

    /**
     * 
     */
    private class SessionListenerSupport implements SessionListener {

        @Override
        public void sessionAllocated(Session session) {
            //
            logger.warn(String.format("New slave:%s joined !!!", session.getSlaveWorker()));
            // 
            if (listener != null) {
                listener.joined(session.getSlaveWorker());
            }
        }

        @Override
        public void sessionReleased(Session session) {
            //
            logger.warn(String.format("Slave:%s left !!!", session.getSlaveWorker()));
            // 
            if (listener != null) {
                listener.left(session.getSlaveWorker());
            }
        }
    }
}