123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289 |
- '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 _stream() {
- const data = require('stream');
- _stream = function _stream() {
- return data;
- };
- return data;
- }
- function _worker_threads() {
- const data = require('worker_threads');
- _worker_threads = function _worker_threads() {
- 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 _objectSpread(target) {
- for (var i = 1; i < arguments.length; i++) {
- var source = arguments[i] != null ? arguments[i] : {};
- var ownKeys = Object.keys(source);
- if (typeof Object.getOwnPropertySymbols === 'function') {
- ownKeys = ownKeys.concat(
- Object.getOwnPropertySymbols(source).filter(function(sym) {
- return Object.getOwnPropertyDescriptor(source, sym).enumerable;
- })
- );
- }
- ownKeys.forEach(function(key) {
- _defineProperty(target, key, source[key]);
- });
- }
- return target;
- }
- 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;
- }
- class ExperimentalWorker {
- constructor(options) {
- _defineProperty(this, '_worker', void 0);
- _defineProperty(this, '_options', void 0);
- _defineProperty(this, '_onProcessEnd', void 0);
- _defineProperty(this, '_request', void 0);
- _defineProperty(this, '_retries', void 0);
- _defineProperty(this, '_stderr', void 0);
- _defineProperty(this, '_stdout', void 0);
- _defineProperty(this, '_fakeStream', void 0);
- this._options = options;
- this._request = null;
- this._stderr = null;
- this._stdout = null;
- this._fakeStream = null;
- this.initialize();
- }
- initialize() {
- this._worker = new (_worker_threads()).Worker(
- _path().default.resolve(__dirname, './threadChild.js'),
- {
- eval: false,
- stderr: true,
- stdout: true,
- workerData: _objectSpread(
- {
- cwd: process.cwd(),
- env: _objectSpread({}, process.env, {
- JEST_WORKER_ID: String(this._options.workerId + 1) // 0-indexed workerId, 1-indexed JEST_WORKER_ID
- }),
- // Suppress --debug / --inspect flags while preserving others (like --harmony).
- execArgv: process.execArgv.filter(
- v => !/^--(debug|inspect)/.test(v)
- ),
- silent: true
- },
- this._options.forkOptions
- )
- }
- );
- if (this._worker.stdout) {
- if (!this._stdout) {
- // We need to add a permanent stream to the merged stream to prevent it
- // from ending when the subprocess stream ends
- this._stdout = (0, _mergeStream().default)(this._getFakeStream());
- }
- this._stdout.add(this._worker.stdout);
- }
- if (this._worker.stderr) {
- if (!this._stderr) {
- // We need to add a permanent stream to the merged stream to prevent it
- // from ending when the subprocess stream ends
- this._stderr = (0, _mergeStream().default)(this._getFakeStream());
- }
- this._stderr.add(this._worker.stderr);
- }
- this._worker.on('message', this.onMessage.bind(this));
- this._worker.on('exit', this.onExit.bind(this));
- this._worker.postMessage([
- _types.CHILD_MESSAGE_INITIALIZE,
- false,
- this._options.workerPath,
- this._options.setupArgs
- ]);
- this._retries++; // If we exceeded the amount of retries, we will emulate an error reply
- // coming from the child. This avoids code duplication related with cleaning
- // the queue, and scheduling the next call.
- if (this._retries > this._options.maxRetries) {
- const error = new Error('Call retries were exceeded');
- this.onMessage([
- _types.PARENT_MESSAGE_CLIENT_ERROR,
- error.name,
- error.message,
- error.stack,
- {
- type: 'WorkerError'
- }
- ]);
- }
- }
- _shutdown() {
- // End the permanent stream so the merged stream end too
- if (this._fakeStream) {
- this._fakeStream.end();
- this._fakeStream = null;
- }
- }
- onMessage(response) {
- let error;
- switch (response[0]) {
- case _types.PARENT_MESSAGE_OK:
- this._onProcessEnd(null, response[1]);
- break;
- case _types.PARENT_MESSAGE_CLIENT_ERROR:
- error = response[4];
- if (error != null && typeof error === 'object') {
- const extra = error; // @ts-ignore: no index
- const NativeCtor = global[response[1]];
- const Ctor = typeof NativeCtor === 'function' ? NativeCtor : Error;
- error = new Ctor(response[2]);
- error.type = response[1];
- error.stack = response[3];
- for (const key in extra) {
- // @ts-ignore: no index
- error[key] = extra[key];
- }
- }
- this._onProcessEnd(error, null);
- break;
- case _types.PARENT_MESSAGE_SETUP_ERROR:
- error = new Error('Error when calling setup: ' + response[2]); // @ts-ignore: adding custom properties to errors.
- error.type = response[1];
- error.stack = response[3];
- this._onProcessEnd(error, null);
- break;
- default:
- throw new TypeError('Unexpected response from worker: ' + response[0]);
- }
- }
- onExit(exitCode) {
- if (exitCode !== 0) {
- this.initialize();
- if (this._request) {
- this._worker.postMessage(this._request);
- }
- } else {
- this._shutdown();
- }
- }
- send(request, onProcessStart, onProcessEnd) {
- onProcessStart(this);
- this._onProcessEnd = (...args) => {
- // Clean the request to avoid sending past requests to workers that fail
- // while waiting for a new request (timers, unhandled rejections...)
- this._request = null;
- return onProcessEnd(...args);
- };
- this._request = request;
- this._retries = 0;
- this._worker.postMessage(request);
- }
- getWorkerId() {
- return this._options.workerId;
- }
- getStdout() {
- return this._stdout;
- }
- getStderr() {
- return this._stderr;
- }
- _getFakeStream() {
- if (!this._fakeStream) {
- this._fakeStream = new (_stream()).PassThrough();
- }
- return this._fakeStream;
- }
- }
- exports.default = ExperimentalWorker;
|