fix(zone.js): enable monkey patching of the queueMicrotask() API in node.js (#50530)

alan-agius4 · alxhub · commit 7837f7119f8c · 2023-06-07T12:42:33.000-07:00
Similar to the browser, now we also monkey patch `queueMicrotask` in Node.js. This API was added in Node.js 11. PR Close #50530
diff --git a/packages/zone.js/lib/browser/browser.ts b/packages/zone.js/lib/browser/browser.ts
@@ -11,6 +11,7 @@
  */
 
 import {findEventTasks} from '../common/events';
+import {patchQueueMicrotask} from '../common/queue-microtask';
 import {patchTimer} from '../common/timers';
 import {patchClass, patchMethod, patchPrototype, scheduleMacroTaskWithCurrentZone, ZONE_SYMBOL_ADD_EVENT_LISTENER, ZONE_SYMBOL_REMOVE_EVENT_LISTENER, zoneSymbol,} from '../common/utils';
 
@@ -25,15 +26,6 @@ Zone.__load_patch('legacy', (global: any) => {
   }
 });
 
-Zone.__load_patch('queueMicrotask', (global: any, Zone: ZoneType, api: _ZonePrivate) => {
-  api.patchMethod(global, 'queueMicrotask', delegate => {
-    return function(self: any, args: any[]) {
-      Zone.current.scheduleMicroTask('queueMicrotask', args[0]);
-    }
-  });
-});
-
-
 Zone.__load_patch('timers', (global: any) => {
   const set = 'set';
   const clear = 'clear';
@@ -300,3 +292,7 @@ Zone.__load_patch('PromiseRejectionEvent', (global: any, Zone: ZoneType) => {
         findPromiseRejectionHandler('rejectionhandled');
   }
 });
+
+Zone.__load_patch('queueMicrotask', (global: any, Zone: ZoneType, api: _ZonePrivate) => {
+  patchQueueMicrotask(global, api);
+});
diff --git a/packages/zone.js/lib/common/queue-microtask.ts b/packages/zone.js/lib/common/queue-microtask.ts
@@ -0,0 +1,19 @@
+/**
+ * @license
+ * Copyright Google LLC All Rights Reserved.
+ *
+ * Use of this source code is governed by an MIT-style license that can be
+ * found in the LICENSE file at https://angular.io/license
+ */
+/**
+ * @fileoverview
+ * @suppress {missingRequire}
+ */
+
+export function patchQueueMicrotask(global: any, api: _ZonePrivate) {
+  api.patchMethod(global, 'queueMicrotask', (delegate) => {
+    return function(self: any, args: any[]) {
+      Zone.current.scheduleMicroTask('queueMicrotask', args[0]);
+    };
+  });
+}
diff --git a/packages/zone.js/lib/node/node.ts b/packages/zone.js/lib/node/node.ts
@@ -11,6 +11,7 @@ import './events';
 import './fs';
 
 import {findEventTasks} from '../common/events';
+import {patchQueueMicrotask} from '../common/queue-microtask';
 import {patchTimer} from '../common/timers';
 import {ArraySlice, isMix, patchMacroTask, patchMicroTask} from '../common/utils';
 
@@ -152,3 +153,7 @@ Zone.__load_patch('console', (global: any, Zone: ZoneType) => {
     }
   });
 });
+
+Zone.__load_patch('queueMicrotask', (global: any, Zone: ZoneType, api: _ZonePrivate) => {
+  patchQueueMicrotask(global, api);
+});
diff --git a/packages/zone.js/lib/zone.configurations.api.ts b/packages/zone.js/lib/zone.configurations.api.ts
@@ -296,9 +296,9 @@ interface ZoneGlobalConfigurations {
 
   /**
    *
-   * Disable the monkey patching of the browser's `queueMicrotask()` API.
+   * Disable the monkey patching of the `queueMicrotask()` API.
    *
-   * By default, `zone.js` monkey patches the browser's `queueMicrotask()` API
+   * By default, `zone.js` monkey patches the `queueMicrotask()` API
    * to ensure that `queueMicrotask()` callback is invoked in the same zone as zone used to invoke
    * `queueMicrotask()`. And also the callback is running as `microTask` like
    * `Promise.prototype.then()`.
diff --git a/packages/zone.js/test/browser_entry_point.ts b/packages/zone.js/test/browser_entry_point.ts
@@ -28,4 +28,3 @@ import './mocha-patch.spec';
 import './jasmine-patch.spec';
 import './browser/messageport.spec';
 import './extra/cordova.spec';
-import './browser/queue-microtask.spec';
diff --git a/packages/zone.js/test/common/queue-microtask.spec.ts b/packages/zone.js/test/common/queue-microtask.spec.ts
diff --git a/packages/zone.js/test/common_tests.ts b/packages/zone.js/test/common_tests.ts
@@ -11,6 +11,7 @@ import './common/zone.spec';
 import './common/task.spec';
 import './common/util.spec';
 import './common/Promise.spec';
+import './common/queue-microtask.spec';
 import './common/fetch.spec';
 import './common/Error.spec';
 import './common/setInterval.spec';

Original file line number	Diff line number	Diff line change
`@@ -296,9 +296,9 @@ interface ZoneGlobalConfigurations {`
`296`	`296`
`297`	`297`	`/**`
`298`	`298`	`*`
`299`		- * Disable the monkey patching of the browser's `queueMicrotask()` API.
	`299`	+ * Disable the monkey patching of the `queueMicrotask()` API.
`300`	`300`	`*`
`301`		- * By default, `zone.js` monkey patches the browser's `queueMicrotask()` API
	`301`	+ * By default, `zone.js` monkey patches the `queueMicrotask()` API
`302`	`302`	* to ensure that `queueMicrotask()` callback is invoked in the same zone as zone used to invoke
`303`	`303`	* `queueMicrotask()`. And also the callback is running as `microTask` like
`304`	`304`	* `Promise.prototype.then()`.