UNPKG

clustery

Version:
94 lines (75 loc) 2.93 kB
'use strict'; 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'];