feat(asyncMemoize): 支持外部管理缓存

src/utils/asyncMemoize.test.ts

@@ -88,4 +88,19 @@ describe('asyncMemoize', () => {
     expect(await fnMemoized(2)).toBe(2)
     expect(i).toBe(2)
   })
+
+  test('支持外部管理缓存', async () => {
+    const fn = jest.fn().mockImplementation(async () => 1)
+    const fnMemoized = asyncMemoize(fn)
+
+    await fnMemoized()
+    expect(fn).toBeCalled().toBeCalledTimes(1)
+
+    await fnMemoized()
+    expect(fn).toBeCalled().toBeCalledTimes(1)
+
+    fnMemoized.cache.clear()
+    await fnMemoized()
+    expect(fn).toBeCalled().toBeCalledTimes(2)
+  })
 })

src/utils/asyncMemoize.ts

@@ -16,6 +16,14 @@ export interface AsyncMemoizeOptions<
   cacheTTL?: number
 }
 
+export type AsyncMemoizeCacheMap = Map<
+  string,
+  {
+    value: any
+    expiredAt: number
+  }
+>
+
 /**
  * 异步函数执行缓存。
  *
@@ -25,15 +33,11 @@ export interface AsyncMemoizeOptions<
 export function asyncMemoize<T extends (...args: any[]) => Promise<any>>(
   asyncFn: T,
   options: AsyncMemoizeOptions<T> = {},
-): T {
+): T & {
+  cache: AsyncMemoizeCacheMap
+} {
   const { cacheKey = (...args: any[]) => args[0], cacheTTL } = options
-  const cache = new Map<
-    string,
-    {
-      value: any
-      expiredAt: number
-    }
-  >()
+  const cache: AsyncMemoizeCacheMap = new Map()
 
   const call = (...args: any[]) => {
     const _cacheKey = cacheKey(...args)
@@ -53,6 +57,7 @@ export function asyncMemoize<T extends (...args: any[]) => Promise<any>>(
     })
     return cacheValueNew
   }
+  call.cache = cache
 
   return call as any
 }