318 lines
9.3 KiB
JavaScript
318 lines
9.3 KiB
JavaScript
|
|
/**
|
||
|
|
* workerpool.js
|
||
|
|
* https://github.com/josdejong/workerpool
|
||
|
|
*
|
||
|
|
* Offload tasks to a pool of workers on node.js and in the browser.
|
||
|
|
*
|
||
|
|
* @version 6.5.1
|
||
|
|
* @date 2023-10-11
|
||
|
|
*
|
||
|
|
* @license
|
||
|
|
* Copyright (C) 2014-2022 Jos de Jong <wjosdejong@gmail.com>
|
||
|
|
*
|
||
|
|
* Licensed under the Apache License, Version 2.0 (the "License"); you may not
|
||
|
|
* use this file except in compliance with the License. You may obtain a copy
|
||
|
|
* of the License at
|
||
|
|
*
|
||
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||
|
|
*
|
||
|
|
* Unless required by applicable law or agreed to in writing, software
|
||
|
|
* distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
||
|
|
* WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
||
|
|
* License for the specific language governing permissions and limitations under
|
||
|
|
* the License.
|
||
|
|
*/
|
||
|
|
|
||
|
|
/******/ (function() { // webpackBootstrap
|
||
|
|
/******/ var __webpack_modules__ = ({
|
||
|
|
|
||
|
|
/***/ 577:
|
||
|
|
/***/ (function(module) {
|
||
|
|
|
||
|
|
/**
|
||
|
|
* The helper class for transferring data from the worker to the main thread.
|
||
|
|
*
|
||
|
|
* @param {Object} message The object to deliver to the main thread.
|
||
|
|
* @param {Object[]} transfer An array of transferable Objects to transfer ownership of.
|
||
|
|
*/
|
||
|
|
function Transfer(message, transfer) {
|
||
|
|
this.message = message;
|
||
|
|
this.transfer = transfer;
|
||
|
|
}
|
||
|
|
module.exports = Transfer;
|
||
|
|
|
||
|
|
/***/ })
|
||
|
|
|
||
|
|
/******/ });
|
||
|
|
/************************************************************************/
|
||
|
|
/******/ // The module cache
|
||
|
|
/******/ var __webpack_module_cache__ = {};
|
||
|
|
/******/
|
||
|
|
/******/ // The require function
|
||
|
|
/******/ function __webpack_require__(moduleId) {
|
||
|
|
/******/ // Check if module is in cache
|
||
|
|
/******/ var cachedModule = __webpack_module_cache__[moduleId];
|
||
|
|
/******/ if (cachedModule !== undefined) {
|
||
|
|
/******/ return cachedModule.exports;
|
||
|
|
/******/ }
|
||
|
|
/******/ // Create a new module (and put it into the cache)
|
||
|
|
/******/ var module = __webpack_module_cache__[moduleId] = {
|
||
|
|
/******/ // no module.id needed
|
||
|
|
/******/ // no module.loaded needed
|
||
|
|
/******/ exports: {}
|
||
|
|
/******/ };
|
||
|
|
/******/
|
||
|
|
/******/ // Execute the module function
|
||
|
|
/******/ __webpack_modules__[moduleId](module, module.exports, __webpack_require__);
|
||
|
|
/******/
|
||
|
|
/******/ // Return the exports of the module
|
||
|
|
/******/ return module.exports;
|
||
|
|
/******/ }
|
||
|
|
/******/
|
||
|
|
/************************************************************************/
|
||
|
|
var __webpack_exports__ = {};
|
||
|
|
// This entry need to be wrapped in an IIFE because it need to be isolated against other modules in the chunk.
|
||
|
|
!function() {
|
||
|
|
var exports = __webpack_exports__;
|
||
|
|
var __webpack_unused_export__;
|
||
|
|
function _typeof(o) { "@babel/helpers - typeof"; return _typeof = "function" == typeof Symbol && "symbol" == typeof Symbol.iterator ? function (o) { return typeof o; } : function (o) { return o && "function" == typeof Symbol && o.constructor === Symbol && o !== Symbol.prototype ? "symbol" : typeof o; }, _typeof(o); }
|
||
|
|
/**
|
||
|
|
* worker must be started as a child process or a web worker.
|
||
|
|
* It listens for RPC messages from the parent process.
|
||
|
|
*/
|
||
|
|
var Transfer = __webpack_require__(577);
|
||
|
|
|
||
|
|
// 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 exit() {}
|
||
|
|
};
|
||
|
|
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);
|
||
|
|
worker.exit = process.exit.bind(process);
|
||
|
|
} else {
|
||
|
|
worker.on = process.on.bind(process);
|
||
|
|
// ignore transfer argument since it is not supported by process
|
||
|
|
worker.send = function (message) {
|
||
|
|
process.send(message);
|
||
|
|
};
|
||
|
|
// 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);
|
||
|
|
};
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Custom handler for when the worker is terminated.
|
||
|
|
*/
|
||
|
|
worker.terminationHandler = undefined;
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Cleanup and exit the worker.
|
||
|
|
* @param {Number} code
|
||
|
|
* @returns
|
||
|
|
*/
|
||
|
|
worker.cleanupAndExit = function (code) {
|
||
|
|
var _exit = function _exit() {
|
||
|
|
worker.exit(code);
|
||
|
|
};
|
||
|
|
if (!worker.terminationHandler) {
|
||
|
|
return _exit();
|
||
|
|
}
|
||
|
|
var result = worker.terminationHandler(code);
|
||
|
|
if (isPromise(result)) {
|
||
|
|
result.then(_exit, _exit);
|
||
|
|
} else {
|
||
|
|
_exit();
|
||
|
|
}
|
||
|
|
};
|
||
|
|
var currentRequestId = null;
|
||
|
|
worker.on('message', function (request) {
|
||
|
|
if (request === TERMINATE_METHOD_ID) {
|
||
|
|
return worker.cleanupAndExit(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) {
|
||
|
|
if (result instanceof Transfer) {
|
||
|
|
worker.send({
|
||
|
|
id: request.id,
|
||
|
|
result: result.message,
|
||
|
|
error: null
|
||
|
|
}, result.transfer);
|
||
|
|
} else {
|
||
|
|
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
|
||
|
|
if (result instanceof Transfer) {
|
||
|
|
worker.send({
|
||
|
|
id: request.id,
|
||
|
|
result: result.message,
|
||
|
|
error: null
|
||
|
|
}, result.transfer);
|
||
|
|
} else {
|
||
|
|
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]
|
||
|
|
* @param {WorkerRegisterOptions} [options]
|
||
|
|
*/
|
||
|
|
worker.register = function (methods, options) {
|
||
|
|
if (methods) {
|
||
|
|
for (var name in methods) {
|
||
|
|
if (methods.hasOwnProperty(name)) {
|
||
|
|
worker.methods[name] = methods[name];
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
if (options) {
|
||
|
|
worker.terminationHandler = options.onTerminate;
|
||
|
|
}
|
||
|
|
worker.send('ready');
|
||
|
|
};
|
||
|
|
worker.emit = function (payload) {
|
||
|
|
if (currentRequestId) {
|
||
|
|
if (payload instanceof Transfer) {
|
||
|
|
worker.send({
|
||
|
|
id: currentRequestId,
|
||
|
|
isEvent: true,
|
||
|
|
payload: payload.message
|
||
|
|
}, payload.transfer);
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
worker.send({
|
||
|
|
id: currentRequestId,
|
||
|
|
isEvent: true,
|
||
|
|
payload: payload
|
||
|
|
});
|
||
|
|
}
|
||
|
|
};
|
||
|
|
if (true) {
|
||
|
|
__webpack_unused_export__ = worker.register;
|
||
|
|
__webpack_unused_export__ = worker.emit;
|
||
|
|
}
|
||
|
|
}();
|
||
|
|
/******/ })()
|
||
|
|
;
|
||
|
|
//# sourceMappingURL=worker.js.map
|