fix: should return AsyncLocalStorage type

node-modules · Dec 19, 2023 · f9dae20 · f9dae20
1 parent 79695af
commit f9dae20
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/src/index.ts b/src/index.ts
@@ -2,10 +2,10 @@ import { AsyncLocalStorage } from 'node:async_hooks';
 
 export const kGALS = Symbol.for('gals#asyncLocalStorage');
 
-export function getAsyncLocalStorage() {
+export function getAsyncLocalStorage<T = any>(): AsyncLocalStorage<T> {
   const g: any = globalThis;
   if (!g[kGALS]) {
-    g[kGALS] = new AsyncLocalStorage();
+    g[kGALS] = new AsyncLocalStorage<T>();
   }
   return g[kGALS];
 }
diff --git a/test/index.test.ts b/test/index.test.ts
@@ -4,9 +4,11 @@ import { getAsyncLocalStorage, kGALS } from '../src/index.js';
 describe('test/index.test.ts', () => {
   it('should only create one als instance', () => {
     const als1 = getAsyncLocalStorage();
-    const als2 = getAsyncLocalStorage();
+    const als2 = getAsyncLocalStorage<object>();
     assert.equal(als1, als2);
     assert.equal(als1, Reflect.get(globalThis, kGALS));
     assert.equal(als2, Reflect.get(global, kGALS));
+    assert.equal(typeof als2.getStore, 'function');
+    assert.equal(als2.getStore(), undefined);
   });
 });