2019-02-12 14:22:33 +00:00
|
|
|
import * as util from 'util';
|
2018-09-22 12:28:50 +00:00
|
|
|
|
2019-06-24 17:18:29 +00:00
|
|
|
const timers = require('timers');
|
|
|
|
|
2019-02-12 14:22:33 +00:00
|
|
|
type AnyFn = (...args: any[]) => any
|
2016-01-12 02:40:23 +00:00
|
|
|
|
2016-01-14 18:44:21 +00:00
|
|
|
// setImmediate and process.nextTick makes use of uv_check and uv_prepare to
|
|
|
|
// run the callbacks, however since we only run uv loop on requests, the
|
|
|
|
// callbacks wouldn't be called until something else activated the uv loop,
|
|
|
|
// which would delay the callbacks for arbitrary long time. So we should
|
|
|
|
// initiatively activate the uv loop once setImmediate and process.nextTick is
|
|
|
|
// called.
|
2019-02-12 14:22:33 +00:00
|
|
|
const wrapWithActivateUvLoop = function <T extends AnyFn> (func: T): T {
|
2018-07-30 01:14:04 +00:00
|
|
|
return wrap(func, function (func) {
|
2019-02-12 14:22:33 +00:00
|
|
|
return function (this: any, ...args: any[]) {
|
2018-07-30 01:14:04 +00:00
|
|
|
process.activateUvLoop();
|
2019-02-12 14:22:33 +00:00
|
|
|
return func.apply(this, args);
|
2018-07-30 01:14:04 +00:00
|
|
|
};
|
2019-02-12 14:22:33 +00:00
|
|
|
}) as T;
|
2018-07-30 01:14:04 +00:00
|
|
|
};
|
|
|
|
|
2019-02-12 14:22:33 +00:00
|
|
|
/**
|
|
|
|
* Casts to any below for func are due to Typescript not supporting symbols
|
|
|
|
* in index signatures
|
|
|
|
*
|
|
|
|
* Refs: https://github.com/Microsoft/TypeScript/issues/1863
|
|
|
|
*/
|
|
|
|
function wrap <T extends AnyFn> (func: T, wrapper: (fn: AnyFn) => T) {
|
2018-07-30 01:14:04 +00:00
|
|
|
const wrapped = wrapper(func);
|
2019-02-12 14:22:33 +00:00
|
|
|
if ((func as any)[util.promisify.custom]) {
|
|
|
|
(wrapped as any)[util.promisify.custom] = wrapper((func as any)[util.promisify.custom]);
|
2016-03-25 19:57:17 +00:00
|
|
|
}
|
2018-07-30 01:14:04 +00:00
|
|
|
return wrapped;
|
2016-03-25 19:57:17 +00:00
|
|
|
}
|
2016-01-12 02:40:23 +00:00
|
|
|
|
2022-10-20 05:49:49 +00:00
|
|
|
// process.nextTick and setImmediate make use of uv_check and uv_prepare to
|
|
|
|
// run the callbacks, however since we only run uv loop on requests, the
|
|
|
|
// callbacks wouldn't be called until something else activated the uv loop,
|
|
|
|
// which would delay the callbacks for arbitrary long time. So we should
|
|
|
|
// initiatively activate the uv loop once process.nextTick and setImmediate is
|
|
|
|
// called.
|
2016-03-25 19:57:17 +00:00
|
|
|
process.nextTick = wrapWithActivateUvLoop(process.nextTick);
|
2019-06-24 17:18:29 +00:00
|
|
|
global.setImmediate = timers.setImmediate = wrapWithActivateUvLoop(timers.setImmediate);
|
2016-03-25 19:57:17 +00:00
|
|
|
global.clearImmediate = timers.clearImmediate;
|
2016-01-12 02:40:23 +00:00
|
|
|
|
2019-06-24 17:18:29 +00:00
|
|
|
// setTimeout needs to update the polling timeout of the event loop, when
|
|
|
|
// called under Chromium's event loop the node's event loop won't get a chance
|
|
|
|
// to update the timeout, so we have to force the node's event loop to
|
2022-10-20 05:49:49 +00:00
|
|
|
// recalculate the timeout in the process.
|
2019-06-24 17:18:29 +00:00
|
|
|
timers.setTimeout = wrapWithActivateUvLoop(timers.setTimeout);
|
|
|
|
timers.setInterval = wrapWithActivateUvLoop(timers.setInterval);
|
|
|
|
|
2022-10-20 05:49:49 +00:00
|
|
|
// Update the global version of the timer apis to use the above wrapper
|
|
|
|
// only in the process that runs node event loop alongside chromium
|
|
|
|
// event loop. We skip renderer with nodeIntegration here because node globals
|
|
|
|
// are deleted in these processes, see renderer/init.js for reference.
|
|
|
|
if (process.type === 'browser' ||
|
|
|
|
process.type === 'utility') {
|
2019-06-24 17:18:29 +00:00
|
|
|
global.setTimeout = timers.setTimeout;
|
|
|
|
global.setInterval = timers.setInterval;
|
2016-01-12 02:40:23 +00:00
|
|
|
}
|
2016-04-21 06:55:56 +00:00
|
|
|
|
2016-09-07 21:45:56 +00:00
|
|
|
if (process.platform === 'win32') {
|
|
|
|
// Always returns EOF for stdin stream.
|
2018-09-13 16:10:51 +00:00
|
|
|
const { Readable } = require('stream');
|
2016-09-07 21:45:56 +00:00
|
|
|
const stdin = new Readable();
|
|
|
|
stdin.push(null);
|
2019-02-12 14:22:33 +00:00
|
|
|
Object.defineProperty(process, 'stdin', {
|
|
|
|
configurable: false,
|
|
|
|
enumerable: true,
|
|
|
|
get () {
|
|
|
|
return stdin;
|
|
|
|
}
|
2016-09-07 21:45:56 +00:00
|
|
|
});
|
2016-04-21 06:55:56 +00:00
|
|
|
}
|