worker.js

/*!
 * hot-restart - worker.js
 * Copyright(c) 2014 dead_horse <dead_horse@qq.com>
 * MIT Licensed
 */

'use strict';

/**
 * Module dependencies.
 */

var debug = require('debug')('hot-restart:worker');
var cluster = require('cluster');
var ms = require('humanize-ms');

module.exports = function (options) {
  var disconnectHandle = options.disconnectHandle || function () {};
  var exitCode = options.exitCode == null
    ? 100
    : ms(options.exitCode);
  var disconnectTime = options.disconnectTime == null
    ? 10000
    : ms(options.disconnectTime);
  var exitTime = options.exitTime == null
    ? 30000
    : ms(options.exitTime);

  process.on('message', function (msg) {
    if (!msg || msg.action !== '__reload__') {
      return;
    }
    debug('worker:%s get reload message', process.pid);

    setTimeout(disconnect, disconnectTime);
    setTimeout(exit, exitTime);
  })

  function disconnect() {
    if (cluster.worker && !cluster.worker.suicide) {
      debug('cluster worker disconnect');
      cluster.worker.disconnect();
    }
    disconnectHandle();
  }

  function exit() {
    process.exit(exitCode);
  }
}