Merge pull request #73 from himself65/async_hooks

fix: missing apis in AsyncLocalStorage
wasmerio · Apr 26, 2024 · 806b6eb · 806b6eb
2 parents cf9b7ca + 08f1ae4
commit 806b6eb
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/src/builtins/internal_js_modules/node/async_hooks.js b/src/builtins/internal_js_modules/node/async_hooks.js
@@ -249,6 +249,15 @@ export class AsyncLocalStorage {
         const currentFrame = AsyncContextFrame.current();
         return currentFrame.get(this.#key);
     }
+    static bind(fn) {
+        return AsyncResource.bind(fn);
+    }
+    static snapshot() {
+        return AsyncLocalStorage.bind((
+            cb,
+            ...args
+        ) => cb(...args));
+    }
 }
 export function executionAsyncId() {
     return 1;

diff --git a/src/builtins/internal_js_modules/node/async_hooks.ts b/src/builtins/internal_js_modules/node/async_hooks.ts
@@ -313,6 +313,17 @@ export class AsyncLocalStorage {
     const currentFrame = AsyncContextFrame.current();
     return currentFrame.get(this.#key);
   }
+
+  static bind(fn: (...args: unknown[]) => unknown) {
+    return AsyncResource.bind(fn);
+  }
+
+  static snapshot() {
+    return AsyncLocalStorage.bind((
+      cb: (...args: unknown[]) => unknown,
+      ...args: unknown[]
+    ) => cb(...args));
+  }
 }
 
 export function executionAsyncId() {