feat(node): allow overriding AsyncHook, AsyncLocalStorage and `As…

…yncResource` with `globalThis` (#126)
unjs · Aug 8, 2023 · 9141c15 · 9141c15
1 parent ae04a73
commit 9141c15
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 3 deletions.
diff --git a/src/runtime/node/async_hooks/_async-hook.ts b/src/runtime/node/async_hooks/_async-hook.ts
@@ -2,7 +2,7 @@ import type asyncHooks from "node:async_hooks";
 
 // https://nodejs.org/api/async_hooks.html
 
-export class AsyncHook implements asyncHooks.HookCallbacks {
+class _AsyncHook implements asyncHooks.HookCallbacks {
   readonly __unenv__ = true;
 
   _enabled: boolean = false;
@@ -53,6 +53,9 @@ export class AsyncHook implements asyncHooks.HookCallbacks {
   }
 }
 
+export const AsyncHook: asyncHooks.AsyncHook =
+  (globalThis as any).AsyncHook || _AsyncHook;
+
 export const createHook: typeof asyncHooks.createHook = function createHook(
   callbacks,
 ) {

diff --git a/src/runtime/node/async_hooks/_async-local-storage.ts b/src/runtime/node/async_hooks/_async-local-storage.ts
@@ -2,7 +2,7 @@ import type asyncHooks from "node:async_hooks";
 
 // https://nodejs.org/api/async_context.html#class-asynclocalstorage
 
-export class AsyncLocalStorage<T> implements asyncHooks.AsyncLocalStorage<T> {
+class _AsyncLocalStorage<T> implements asyncHooks.AsyncLocalStorage<T> {
   readonly __unenv__ = true;
 
   _currentStore: undefined | T;
@@ -51,3 +51,6 @@ export class AsyncLocalStorage<T> implements asyncHooks.AsyncLocalStorage<T> {
     throw new Error("[unenv] `AsyncLocalStorage.snapshot` is not implemented!");
   }
 }
+
+export const AsyncLocalStorage: typeof asyncHooks.AsyncLocalStorage =
+  (globalThis as any).AsyncLocalStorage || _AsyncLocalStorage;
diff --git a/src/runtime/node/async_hooks/_async-resource.ts b/src/runtime/node/async_hooks/_async-resource.ts
@@ -5,7 +5,7 @@ import { executionAsyncId } from "./_async-hook";
 
 let _asyncIdCounter = 100;
 
-export class AsyncResource implements asyncHooks.AsyncResource {
+class _AsyncResource implements asyncHooks.AsyncResource {
   readonly __unenv__ = true;
 
   type: string;
@@ -63,3 +63,6 @@ export class AsyncResource implements asyncHooks.AsyncResource {
     return this._triggerAsyncId as number;
   }
 }
+
+export const AsyncResource: typeof asyncHooks.AsyncResource =
+  (globalThis as any).AsyncResource || _AsyncResource;