clustery
Version:
A node cluster manager
94 lines (75 loc) • 2.93 kB
JavaScript
;
Object.defineProperty(exports, "__esModule", {
value: true
});
var _extends = Object.assign || function (target) { for (var i = 1; i < arguments.length; i++) { var source = arguments[i]; for (var key in source) { if (Object.prototype.hasOwnProperty.call(source, key)) { target[key] = source[key]; } } } return target; };
var _createClass = function () { function defineProperties(target, props) { for (var i = 0; i < props.length; i++) { var descriptor = props[i]; descriptor.enumerable = descriptor.enumerable || false; descriptor.configurable = true; if ("value" in descriptor) descriptor.writable = true; Object.defineProperty(target, descriptor.key, descriptor); } } return function (Constructor, protoProps, staticProps) { if (protoProps) defineProperties(Constructor.prototype, protoProps); if (staticProps) defineProperties(Constructor, staticProps); return Constructor; }; }();
var _cluster = require('cluster');
var _cluster2 = _interopRequireDefault(_cluster);
function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
function _classCallCheck(instance, Constructor) { if (!(instance instanceof Constructor)) { throw new TypeError("Cannot call a class as a function"); } }
var getContext = function getContext() {
if (_cluster2.default.isMaster) {
return {
processType: 'master'
};
}
return {
processType: 'worker',
workerId: process.env.CLUSTERY_WORKER_ID
};
};
var master = function master(instance) {
instance.clusterMap.forEach(function (id, index) {
var worker = _cluster2.default.fork({ CLUSTERY_WORKER_ID: id });
});
};
var init = function init(instance) {
if (_cluster2.default.isMaster) master(instance);
};
var _class = function () {
function _class(map, options) {
var _this = this;
_classCallCheck(this, _class);
this.clusterMap = map;
this.context = getContext();
this.options = _extends({
prefix: 'clusteryMessage: ',
identifierDelimiter: ':'
}, options);
// Makes sure we can safely register all our events before we start
// our cluster
process.nextTick(function () {
return init(_this);
});
}
_createClass(_class, [{
key: 'whichWorker',
value: function whichWorker() {
if (_cluster2.default.isMaster) return;
var context = getContext();
if (!context.workerId) return;
return context.workerId;
}
}, {
key: 'is',
value: function is(processType, workerId) {
var context = getContext();
if (context.processType !== processType.toLowerCase()) {
return false;
}
if (workerId && context.workerId !== workerId) {
return false;
}
return true;
}
}, {
key: 'which',
value: function which() {
return this.whichWorker();
}
}]);
return _class;
}();
exports.default = _class;
module.exports = exports['default'];