async_hooks: use resource stack for AsyncLocalStorage run

PR-URL: #39890
Reviewed-By: James M Snell <[email protected]>
Reviewed-By: Vladimir de Turckheim <[email protected]>
Reviewed-By: Gerhard Stöbich <[email protected]>
Reviewed-By: Minwoo Jung <[email protected]>
Reviewed-By: Zijian Liu <[email protected]>

Author: Stephen Belanger

benchmark/async_hooks/async-local-storage-run.js

@@ -0,0 +1,21 @@
+'use strict';
+const common = require('../common.js');
+const { AsyncLocalStorage } = require('async_hooks');
+
+const bench = common.createBenchmark(main, {
+  n: [1e7]
+});
+
+async function run(store, n) {
+  for (let i = 0; i < n; i++) {
+    await new Promise((resolve) => store.run(i, resolve));
+  }
+}
+
+function main({ n }) {
+  const store = new AsyncLocalStorage();
+  bench.start();
+  run(store, n).then(() => {
+    bench.end(n);
+  });
+}

lib/async_hooks.js

@@ -263,7 +263,6 @@ const storageHook = createHook({
   }
 });
 
-const defaultAlsResourceOpts = { requireManualDestroy: true };
 class AsyncLocalStorage {
   constructor() {
     this.kResourceStore = Symbol('kResourceStore');
@@ -309,14 +308,19 @@ class AsyncLocalStorage {
     if (ObjectIs(store, this.getStore())) {
       return ReflectApply(callback, null, args);
     }
-    const resource = new AsyncResource('AsyncLocalStorage',
-                                       defaultAlsResourceOpts);
-    // Calling emitDestroy before runInAsyncScope avoids a try/finally
-    // It is ok because emitDestroy only schedules calling the hook
-    return resource.emitDestroy().runInAsyncScope(() => {
-      this.enterWith(store);
+
+    this._enable();
+
+    const resource = executionAsyncResource();
+    const oldStore = resource[this.kResourceStore];
+
+    resource[this.kResourceStore] = store;
+
+    try {
       return ReflectApply(callback, null, args);
-    });
+    } finally {
+      resource[this.kResourceStore] = oldStore;
+    }
   }
 
   exit(callback, ...args) {