2020-03-31 20:42:07 +08:00
|
|
|
'use strict';
|
|
|
|
|
|
|
|
Object.defineProperty(exports, '__esModule', {
|
|
|
|
value: true
|
|
|
|
});
|
|
|
|
exports.default = void 0;
|
|
|
|
|
|
|
|
function path() {
|
|
|
|
const data = _interopRequireWildcard(require('path'));
|
|
|
|
|
|
|
|
path = function () {
|
|
|
|
return data;
|
|
|
|
};
|
|
|
|
|
|
|
|
return data;
|
|
|
|
}
|
|
|
|
|
|
|
|
function _stream() {
|
|
|
|
const data = require('stream');
|
|
|
|
|
|
|
|
_stream = function () {
|
|
|
|
return data;
|
|
|
|
};
|
|
|
|
|
|
|
|
return data;
|
|
|
|
}
|
|
|
|
|
|
|
|
function _worker_threads() {
|
|
|
|
const data = require('worker_threads');
|
|
|
|
|
|
|
|
_worker_threads = function () {
|
|
|
|
return data;
|
|
|
|
};
|
|
|
|
|
|
|
|
return data;
|
|
|
|
}
|
|
|
|
|
|
|
|
function _mergeStream() {
|
|
|
|
const data = _interopRequireDefault(require('merge-stream'));
|
|
|
|
|
|
|
|
_mergeStream = function () {
|
|
|
|
return data;
|
|
|
|
};
|
|
|
|
|
|
|
|
return data;
|
|
|
|
}
|
|
|
|
|
|
|
|
function _types() {
|
|
|
|
const data = require('../types');
|
|
|
|
|
|
|
|
_types = function () {
|
|
|
|
return data;
|
|
|
|
};
|
|
|
|
|
|
|
|
return data;
|
|
|
|
}
|
|
|
|
|
|
|
|
function _interopRequireDefault(obj) {
|
|
|
|
return obj && obj.__esModule ? obj : {default: obj};
|
|
|
|
}
|
|
|
|
|
|
|
|
function _getRequireWildcardCache() {
|
|
|
|
if (typeof WeakMap !== 'function') return null;
|
|
|
|
var cache = new WeakMap();
|
|
|
|
_getRequireWildcardCache = function () {
|
|
|
|
return cache;
|
|
|
|
};
|
|
|
|
return cache;
|
|
|
|
}
|
|
|
|
|
|
|
|
function _interopRequireWildcard(obj) {
|
|
|
|
if (obj && obj.__esModule) {
|
|
|
|
return obj;
|
|
|
|
}
|
|
|
|
if (obj === null || (typeof obj !== 'object' && typeof obj !== 'function')) {
|
|
|
|
return {default: obj};
|
|
|
|
}
|
|
|
|
var cache = _getRequireWildcardCache();
|
|
|
|
if (cache && cache.has(obj)) {
|
|
|
|
return cache.get(obj);
|
|
|
|
}
|
|
|
|
var newObj = {};
|
|
|
|
var hasPropertyDescriptor =
|
|
|
|
Object.defineProperty && Object.getOwnPropertyDescriptor;
|
|
|
|
for (var key in obj) {
|
|
|
|
if (Object.prototype.hasOwnProperty.call(obj, key)) {
|
|
|
|
var desc = hasPropertyDescriptor
|
|
|
|
? Object.getOwnPropertyDescriptor(obj, key)
|
|
|
|
: null;
|
|
|
|
if (desc && (desc.get || desc.set)) {
|
|
|
|
Object.defineProperty(newObj, key, desc);
|
|
|
|
} else {
|
|
|
|
newObj[key] = obj[key];
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
newObj.default = obj;
|
|
|
|
if (cache) {
|
|
|
|
cache.set(obj, newObj);
|
|
|
|
}
|
|
|
|
return newObj;
|
|
|
|
}
|
|
|
|
|
|
|
|
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, '_request', void 0);
|
|
|
|
|
|
|
|
_defineProperty(this, '_retries', void 0);
|
|
|
|
|
|
|
|
_defineProperty(this, '_onProcessEnd', void 0);
|
|
|
|
|
|
|
|
_defineProperty(this, '_fakeStream', void 0);
|
|
|
|
|
|
|
|
_defineProperty(this, '_stdout', void 0);
|
|
|
|
|
|
|
|
_defineProperty(this, '_stderr', void 0);
|
|
|
|
|
|
|
|
_defineProperty(this, '_exitPromise', void 0);
|
|
|
|
|
|
|
|
_defineProperty(this, '_resolveExitPromise', void 0);
|
|
|
|
|
|
|
|
_defineProperty(this, '_forceExited', void 0);
|
|
|
|
|
|
|
|
this._options = options;
|
|
|
|
this._request = null;
|
|
|
|
this._fakeStream = null;
|
|
|
|
this._stdout = null;
|
|
|
|
this._stderr = null;
|
|
|
|
this._exitPromise = new Promise(resolve => {
|
|
|
|
this._resolveExitPromise = resolve;
|
|
|
|
});
|
|
|
|
this._forceExited = false;
|
|
|
|
this.initialize();
|
|
|
|
}
|
|
|
|
|
|
|
|
initialize() {
|
|
|
|
this._worker = new (_worker_threads().Worker)(
|
|
|
|
path().resolve(__dirname, './threadChild.js'),
|
|
|
|
{
|
|
|
|
eval: false,
|
|
|
|
stderr: true,
|
|
|
|
stdout: true,
|
2020-04-30 20:40:07 +08:00
|
|
|
workerData: {
|
|
|
|
cwd: process.cwd(),
|
|
|
|
env: {
|
|
|
|
...process.env,
|
|
|
|
JEST_WORKER_ID: String(this._options.workerId + 1) // 0-indexed workerId, 1-indexed JEST_WORKER_ID
|
2020-03-31 20:42:07 +08:00
|
|
|
},
|
2020-04-30 20:40:07 +08:00
|
|
|
// Suppress --debug / --inspect flags while preserving others (like --harmony).
|
|
|
|
execArgv: process.execArgv.filter(v => !/^--(debug|inspect)/.test(v)),
|
|
|
|
silent: true,
|
|
|
|
...this._options.forkOptions
|
|
|
|
}
|
2020-03-31 20:42:07 +08:00
|
|
|
}
|
|
|
|
);
|
|
|
|
|
|
|
|
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;
|
|
|
|
}
|
|
|
|
|
|
|
|
this._resolveExitPromise();
|
|
|
|
}
|
|
|
|
|
|
|
|
_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._forceExited) {
|
|
|
|
this.initialize();
|
|
|
|
|
|
|
|
if (this._request) {
|
|
|
|
this._worker.postMessage(this._request);
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
this._shutdown();
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
waitForExit() {
|
|
|
|
return this._exitPromise;
|
|
|
|
}
|
|
|
|
|
|
|
|
forceExit() {
|
|
|
|
this._forceExited = true;
|
|
|
|
|
|
|
|
this._worker.terminate();
|
|
|
|
}
|
|
|
|
|
|
|
|
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;
|