123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134 |
- 'use strict';
- Object.defineProperty(exports, '__esModule', {
- value: true
- });
- exports.default = void 0;
- function _path() {
- const data = _interopRequireDefault(require('path'));
- _path = function _path() {
- return data;
- };
- return data;
- }
- function _mergeStream() {
- const data = _interopRequireDefault(require('merge-stream'));
- _mergeStream = function _mergeStream() {
- return data;
- };
- return data;
- }
- var _types = require('../types');
- function _interopRequireDefault(obj) {
- return obj && obj.__esModule ? obj : {default: obj};
- }
- function _defineProperty(obj, key, value) {
- if (key in obj) {
- Object.defineProperty(obj, key, {
- value: value,
- enumerable: true,
- configurable: true,
- writable: true
- });
- } else {
- obj[key] = value;
- }
- return obj;
- }
- /* istanbul ignore next */
- const emptyMethod = () => {};
- class BaseWorkerPool {
- constructor(workerPath, options) {
- _defineProperty(this, '_stderr', void 0);
- _defineProperty(this, '_stdout', void 0);
- _defineProperty(this, '_options', void 0);
- _defineProperty(this, '_workers', void 0);
- this._options = options;
- this._workers = new Array(options.numWorkers);
- if (!_path().default.isAbsolute(workerPath)) {
- workerPath = require.resolve(workerPath);
- }
- const stdout = (0, _mergeStream().default)();
- const stderr = (0, _mergeStream().default)();
- const forkOptions = options.forkOptions,
- maxRetries = options.maxRetries,
- setupArgs = options.setupArgs;
- for (let i = 0; i < options.numWorkers; i++) {
- const workerOptions = {
- forkOptions,
- maxRetries,
- setupArgs,
- workerId: i,
- workerPath
- };
- const worker = this.createWorker(workerOptions);
- const workerStdout = worker.getStdout();
- const workerStderr = worker.getStderr();
- if (workerStdout) {
- stdout.add(workerStdout);
- }
- if (workerStderr) {
- stderr.add(workerStderr);
- }
- this._workers[i] = worker;
- }
- this._stdout = stdout;
- this._stderr = stderr;
- }
- getStderr() {
- return this._stderr;
- }
- getStdout() {
- return this._stdout;
- }
- getWorkers() {
- return this._workers;
- }
- getWorkerById(workerId) {
- return this._workers[workerId];
- }
- createWorker(_workerOptions) {
- throw Error('Missing method createWorker in WorkerPool');
- }
- end() {
- // We do not cache the request object here. If so, it would only be only
- // processed by one of the workers, and we want them all to close.
- for (let i = 0; i < this._workers.length; i++) {
- this._workers[i].send(
- [_types.CHILD_MESSAGE_END, false],
- emptyMethod,
- emptyMethod
- );
- }
- }
- }
- exports.default = BaseWorkerPool;
|