eazy-pomelo/lib/master/master.js

NetEase Pomelo Of EazyGame OEM

var starter = require('./starter');
var logger = require('pomelo-logger').getLogger('pomelo', __filename);
var crashLogger = require('pomelo-logger').getLogger('crash-log', __filename);
var adminLogger = require('pomelo-logger').getLogger('admin-log', __filename);
var admin = require('pomelo-admin');
var util = require('util');
var utils = require('../util/utils');
var moduleUtil = require('../util/moduleUtil');
var Constants = require('../util/constants');

var Server = function(app, opts) {
  this.app = app;
  this.masterInfo = app.getMaster();
  this.registered = {};
  this.modules = [];
  opts = opts || {};
  
  opts.port = this.masterInfo.port;
  opts.env = this.app.get(Constants.RESERVED.ENV);
  this.closeWatcher = opts.closeWatcher;
  this.masterConsole = admin.createMasterConsole(opts);
};

module.exports = Server;

Server.prototype.start = function(cb) {
  moduleUtil.registerDefaultModules(true, this.app, this.closeWatcher);
  moduleUtil.loadModules(this, this.masterConsole);

  var self = this;
  // start master console
  this.masterConsole.start(function(err) {
    if(err) {
      process.exit(0);
    }
    moduleUtil.startModules(self.modules, function(err) {
      if(err) {
        utils.invokeCallback(cb, err);
        return;
      }

      if(self.app.get(Constants.RESERVED.MODE) !== Constants.RESERVED.STAND_ALONE) {
        starter.runServers(self.app);
      }
      utils.invokeCallback(cb);
    });
  });
  
  this.masterConsole.on('error', function(err) {
    if(!!err) {
      logger.error('masterConsole encounters with error: ' + err.stack);
      return;
    }
  });
  
  this.masterConsole.on('reconnect', function(info){
    self.app.addServers([info]);
  });

  // monitor servers disconnect event
  this.masterConsole.on('disconnect', function(id, type, info, reason) {
    crashLogger.info(util.format('[%s],[%s],[%s],[%s]', type, id, Date.now(), reason || 'disconnect'));
    var count = 0;
    var time = 0;
    var pingTimer = null;
    var server = self.app.getServerById(id);
    var stopFlags = self.app.get(Constants.RESERVED.STOP_SERVERS) || [];
    if(!!server && (server[Constants.RESERVED.AUTO_RESTART] === 'true' || server[Constants.RESERVED.RESTART_FORCE] === 'true') && stopFlags.indexOf(id) < 0) {
      var setTimer = function(time) {
        pingTimer = setTimeout(function() {
          utils.ping(server.host, function(flag) {
            if(flag)  {
              handle();
            } else {
              count++;
              if(count > 3) {
                time = Constants.TIME.TIME_WAIT_MAX_PING;
              } else {
                time = Constants.TIME.TIME_WAIT_PING * count;
              }
              setTimer(time);
            }
          });
        }, time);
      };
      setTimer(time);
      var handle = function() {
        clearTimeout(pingTimer);
        utils.checkPort(server, function(status) {
          if(status === 'error') {
            utils.invokeCallback(cb, new Error('Check port command executed with error.'));
            return;
          } else if(status === 'busy') {
            if(!!server[Constants.RESERVED.RESTART_FORCE]) {
              starter.kill([info.pid], [server]);
            } else {
              utils.invokeCallback(cb, new Error('Port occupied already, check your server to add.'));
              return;
            }
          }
          setTimeout(function() {
            starter.run(self.app, server, null);
          }, Constants.TIME.TIME_WAIT_STOP);
        });
      };
    }
  });

  // monitor servers register event
  this.masterConsole.on('register', function(record) {
    starter.bindCpu(record.id, record.pid, record.host);
  });

  this.masterConsole.on('admin-log', function(log, error) {
    if(error) {
      adminLogger.error(JSON.stringify(log));
    } else {
      adminLogger.info(JSON.stringify(log));
    }
  });
};

Server.prototype.stop = function(cb) {
  this.masterConsole.stop();
  process.nextTick(cb);
};