aboutsummaryrefslogtreecommitdiff
path: root/node_modules/jest-worker/build/workers/ChildProcessWorker.js
diff options
context:
space:
mode:
Diffstat (limited to 'node_modules/jest-worker/build/workers/ChildProcessWorker.js')
-rw-r--r--node_modules/jest-worker/build/workers/ChildProcessWorker.js333
1 files changed, 333 insertions, 0 deletions
diff --git a/node_modules/jest-worker/build/workers/ChildProcessWorker.js b/node_modules/jest-worker/build/workers/ChildProcessWorker.js
new file mode 100644
index 0000000..f8a42c9
--- /dev/null
+++ b/node_modules/jest-worker/build/workers/ChildProcessWorker.js
@@ -0,0 +1,333 @@
+'use strict';
+
+Object.defineProperty(exports, '__esModule', {
+ value: true
+});
+exports.default = void 0;
+
+function _child_process() {
+ const data = require('child_process');
+
+ _child_process = function () {
+ return data;
+ };
+
+ return data;
+}
+
+function _stream() {
+ const data = require('stream');
+
+ _stream = function () {
+ return data;
+ };
+
+ return data;
+}
+
+function _mergeStream() {
+ const data = _interopRequireDefault(require('merge-stream'));
+
+ _mergeStream = function () {
+ return data;
+ };
+
+ return data;
+}
+
+function _supportsColor() {
+ const data = require('supports-color');
+
+ _supportsColor = function () {
+ 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;
+}
+
+const SIGNAL_BASE_EXIT_CODE = 128;
+const SIGKILL_EXIT_CODE = SIGNAL_BASE_EXIT_CODE + 9;
+const SIGTERM_EXIT_CODE = SIGNAL_BASE_EXIT_CODE + 15; // How long to wait after SIGTERM before sending SIGKILL
+
+const SIGKILL_DELAY = 500;
+/**
+ * This class wraps the child process and provides a nice interface to
+ * communicate with. It takes care of:
+ *
+ * - Re-spawning the process if it dies.
+ * - Queues calls while the worker is busy.
+ * - Re-sends the requests if the worker blew up.
+ *
+ * The reason for queueing them here (since childProcess.send also has an
+ * internal queue) is because the worker could be doing asynchronous work, and
+ * this would lead to the child process to read its receiving buffer and start a
+ * second call. By queueing calls here, we don't send the next call to the
+ * children until we receive the result of the previous one.
+ *
+ * As soon as a request starts to be processed by a worker, its "processed"
+ * field is changed to "true", so that other workers which might encounter the
+ * same call skip it.
+ */
+
+class ChildProcessWorker {
+ constructor(options) {
+ _defineProperty(this, '_child', void 0);
+
+ _defineProperty(this, '_options', void 0);
+
+ _defineProperty(this, '_request', void 0);
+
+ _defineProperty(this, '_retries', void 0);
+
+ _defineProperty(this, '_onProcessEnd', void 0);
+
+ _defineProperty(this, '_onCustomMessage', 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);
+
+ this._options = options;
+ this._request = null;
+ this._fakeStream = null;
+ this._stdout = null;
+ this._stderr = null;
+ this._exitPromise = new Promise(resolve => {
+ this._resolveExitPromise = resolve;
+ });
+ this.initialize();
+ }
+
+ initialize() {
+ const forceColor = _supportsColor().stdout
+ ? {
+ FORCE_COLOR: '1'
+ }
+ : {};
+ const child = (0, _child_process().fork)(
+ require.resolve('./processChild'),
+ [],
+ {
+ cwd: process.cwd(),
+ env: {
+ ...process.env,
+ JEST_WORKER_ID: String(this._options.workerId + 1),
+ // 0-indexed workerId, 1-indexed JEST_WORKER_ID
+ ...forceColor
+ },
+ // Suppress --debug / --inspect flags while preserving others (like --harmony).
+ execArgv: process.execArgv.filter(v => !/^--(debug|inspect)/.test(v)),
+ silent: true,
+ ...this._options.forkOptions
+ }
+ );
+
+ if (child.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(child.stdout);
+ }
+
+ if (child.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(child.stderr);
+ }
+
+ child.on('message', this._onMessage.bind(this));
+ child.on('exit', this._onExit.bind(this));
+ child.send([
+ _types.CHILD_MESSAGE_INITIALIZE,
+ false,
+ this._options.workerPath,
+ this._options.setupArgs
+ ]);
+ this._child = child;
+ 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(
+ `Jest worker encountered ${this._retries} child process exceptions, exceeding retry limit`
+ );
+
+ this._onMessage([
+ _types.PARENT_MESSAGE_CLIENT_ERROR,
+ error.name,
+ error.message,
+ error.stack,
+ {
+ type: 'WorkerError'
+ }
+ ]);
+ }
+ }
+
+ _shutdown() {
+ // End the temporary streams so the merged streams end too
+ if (this._fakeStream) {
+ this._fakeStream.end();
+
+ this._fakeStream = null;
+ }
+
+ this._resolveExitPromise();
+ }
+
+ _onMessage(response) {
+ // TODO: Add appropriate type check
+ 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-expect-error: 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) {
+ error[key] = extra[key];
+ }
+ }
+
+ this._onProcessEnd(error, null);
+
+ break;
+
+ case _types.PARENT_MESSAGE_SETUP_ERROR:
+ error = new Error('Error when calling setup: ' + response[2]);
+ error.type = response[1];
+ error.stack = response[3];
+
+ this._onProcessEnd(error, null);
+
+ break;
+
+ case _types.PARENT_MESSAGE_CUSTOM:
+ this._onCustomMessage(response[1]);
+
+ break;
+
+ default:
+ throw new TypeError('Unexpected response from worker: ' + response[0]);
+ }
+ }
+
+ _onExit(exitCode) {
+ if (
+ exitCode !== 0 &&
+ exitCode !== null &&
+ exitCode !== SIGTERM_EXIT_CODE &&
+ exitCode !== SIGKILL_EXIT_CODE
+ ) {
+ this.initialize();
+
+ if (this._request) {
+ this._child.send(this._request);
+ }
+ } else {
+ this._shutdown();
+ }
+ }
+
+ send(request, onProcessStart, onProcessEnd, onCustomMessage) {
+ 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._onCustomMessage = (...arg) => onCustomMessage(...arg);
+
+ this._request = request;
+ this._retries = 0;
+
+ this._child.send(request, () => {});
+ }
+
+ waitForExit() {
+ return this._exitPromise;
+ }
+
+ forceExit() {
+ this._child.kill('SIGTERM');
+
+ const sigkillTimeout = setTimeout(
+ () => this._child.kill('SIGKILL'),
+ SIGKILL_DELAY
+ );
+
+ this._exitPromise.then(() => clearTimeout(sigkillTimeout));
+ }
+
+ 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 = ChildProcessWorker;