You can not select more than 25 topics
			Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
		
		
		
		
		
			
		
			
				
					
					
						
							353 lines
						
					
					
						
							8.0 KiB
						
					
					
				
			
		
		
	
	
							353 lines
						
					
					
						
							8.0 KiB
						
					
					
				| '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, '_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);
 | |
| 
 | |
|     _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,
 | |
|         // @ts-expect-error: added in newer versions
 | |
|         resourceLimits: this._options.resourceLimits,
 | |
|         stderr: true,
 | |
|         stdout: true,
 | |
|         workerData: {
 | |
|           cwd: process.cwd(),
 | |
|           env: {
 | |
|             ...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;
 | |
|     }
 | |
| 
 | |
|     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-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) {
 | |
|             // @ts-expect-error: 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-expect-error: adding custom properties to errors.
 | |
| 
 | |
|         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 && !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, 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._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;
 |