fix(useMemoize): allow number in getKey (#2074)

vueuse · Aug 23, 2022 · 06ad02b · 06ad02b
1 parent 201eb82
commit 06ad02b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/packages/core/useMemoize/index.test.ts b/packages/core/useMemoize/index.test.ts
@@ -159,7 +159,7 @@ describe('useMemoize', () => {
 
         resolver.mockClear()
         expect(memo(3)).toBe('result-1')
-        expect(memo(4)).toBe('result-2')
+        expect(memo('4')).toBe('result-2')
         expect(resolver).not.toHaveBeenCalled()
       })
     })

diff --git a/packages/core/useMemoize/index.ts b/packages/core/useMemoize/index.ts
@@ -77,7 +77,7 @@ export interface UseMemoizeReturn <Result, Args extends unknown[]> {
 }
 
 export interface UseMemoizeOptions<Result, Args extends unknown[]> {
-  getKey?: (...args: Args) => string
+  getKey?: (...args: Args) => string | number
   cache?: UseMemoizeCache<CacheKey, Result>
 }
 
@@ -109,7 +109,7 @@ export function useMemoize<Result, Args extends unknown[]>(
   /**
    * Load data and save in cache
    */
-  const _loadData = (key: string, ...args: Args): Result => {
+  const _loadData = (key: string | number, ...args: Args): Result => {
     cache.set(key, resolver(...args))
     return cache.get(key) as Result
   }