| /** |
| * worker must be started as a child process or a web worker. |
| * It listens for RPC messages from the parent process. |
| */ |
| |
| // source of inspiration: https://github.com/sindresorhus/require-fool-webpack |
| var requireFoolWebpack = eval( |
| 'typeof require !== \'undefined\'' + |
| ' ? require' + |
| ' : function (module) { throw new Error(\'Module " + module + " not found.\') }' |
| ); |
| |
| /** |
| * Special message sent by parent which causes the worker to terminate itself. |
| * Not a "message object"; this string is the entire message. |
| */ |
| var TERMINATE_METHOD_ID = '__workerpool-terminate__'; |
| |
| // var nodeOSPlatform = require('./environment').nodeOSPlatform; |
| |
| // create a worker API for sending and receiving messages which works both on |
| // node.js and in the browser |
| var worker = { |
| exit: function() {} |
| }; |
| if (typeof self !== 'undefined' && typeof postMessage === 'function' && typeof addEventListener === 'function') { |
| // worker in the browser |
| worker.on = function (event, callback) { |
| addEventListener(event, function (message) { |
| callback(message.data); |
| }) |
| }; |
| worker.send = function (message) { |
| postMessage(message); |
| }; |
| } |
| else if (typeof process !== 'undefined') { |
| // node.js |
| |
| var WorkerThreads; |
| try { |
| WorkerThreads = requireFoolWebpack('worker_threads'); |
| } catch(error) { |
| if (typeof error === 'object' && error !== null && error.code === 'MODULE_NOT_FOUND') { |
| // no worker_threads, fallback to sub-process based workers |
| } else { |
| throw error; |
| } |
| } |
| |
| if (WorkerThreads && |
| /* if there is a parentPort, we are in a WorkerThread */ |
| WorkerThreads.parentPort !== null) { |
| var parentPort = WorkerThreads.parentPort; |
| worker.send = parentPort.postMessage.bind(parentPort); |
| worker.on = parentPort.on.bind(parentPort); |
| } else { |
| worker.on = process.on.bind(process); |
| worker.send = process.send.bind(process); |
| // register disconnect handler only for subprocess worker to exit when parent is killed unexpectedly |
| worker.on('disconnect', function () { |
| process.exit(1); |
| }); |
| worker.exit = process.exit.bind(process); |
| } |
| } |
| else { |
| throw new Error('Script must be executed as a worker'); |
| } |
| |
| function convertError(error) { |
| return Object.getOwnPropertyNames(error).reduce(function(product, name) { |
| return Object.defineProperty(product, name, { |
| value: error[name], |
| enumerable: true |
| }); |
| }, {}); |
| } |
| |
| /** |
| * Test whether a value is a Promise via duck typing. |
| * @param {*} value |
| * @returns {boolean} Returns true when given value is an object |
| * having functions `then` and `catch`. |
| */ |
| function isPromise(value) { |
| return value && (typeof value.then === 'function') && (typeof value.catch === 'function'); |
| } |
| |
| // functions available externally |
| worker.methods = {}; |
| |
| /** |
| * Execute a function with provided arguments |
| * @param {String} fn Stringified function |
| * @param {Array} [args] Function arguments |
| * @returns {*} |
| */ |
| worker.methods.run = function run(fn, args) { |
| var f = new Function('return (' + fn + ').apply(null, arguments);'); |
| return f.apply(f, args); |
| }; |
| |
| /** |
| * Get a list with methods available on this worker |
| * @return {String[]} methods |
| */ |
| worker.methods.methods = function methods() { |
| return Object.keys(worker.methods); |
| }; |
| |
| var currentRequestId = null; |
| |
| worker.on('message', function (request) { |
| if (request === TERMINATE_METHOD_ID) { |
| return worker.exit(0); |
| } |
| try { |
| var method = worker.methods[request.method]; |
| |
| if (method) { |
| currentRequestId = request.id; |
| |
| // execute the function |
| var result = method.apply(method, request.params); |
| |
| if (isPromise(result)) { |
| // promise returned, resolve this and then return |
| result |
| .then(function (result) { |
| worker.send({ |
| id: request.id, |
| result: result, |
| error: null |
| }); |
| currentRequestId = null; |
| }) |
| .catch(function (err) { |
| worker.send({ |
| id: request.id, |
| result: null, |
| error: convertError(err) |
| }); |
| currentRequestId = null; |
| }); |
| } |
| else { |
| // immediate result |
| worker.send({ |
| id: request.id, |
| result: result, |
| error: null |
| }); |
| |
| currentRequestId = null; |
| } |
| } |
| else { |
| throw new Error('Unknown method "' + request.method + '"'); |
| } |
| } |
| catch (err) { |
| worker.send({ |
| id: request.id, |
| result: null, |
| error: convertError(err) |
| }); |
| } |
| }); |
| |
| /** |
| * Register methods to the worker |
| * @param {Object} methods |
| */ |
| worker.register = function (methods) { |
| |
| if (methods) { |
| for (var name in methods) { |
| if (methods.hasOwnProperty(name)) { |
| worker.methods[name] = methods[name]; |
| } |
| } |
| } |
| |
| worker.send('ready'); |
| |
| }; |
| |
| worker.emit = function (payload) { |
| if (currentRequestId) { |
| worker.send({ |
| id: currentRequestId, |
| isEvent: true, |
| payload |
| }); |
| } |
| }; |
| |
| if (typeof exports !== 'undefined') { |
| exports.add = worker.register; |
| exports.emit = worker.emit; |
| } |