From 06ad02b5ad356e1d3d8862fc627fa1686843aad2 Mon Sep 17 00:00:00 2001
From: Mikhailov Nikita <mikhaylovnv.dev@gmail.com>
Date: Tue, 23 Aug 2022 16:40:33 +0300
Subject: [PATCH] fix(useMemoize): allow number in `getKey` (#2074)

---
 packages/core/useMemoize/index.test.ts | 2 +-
 packages/core/useMemoize/index.ts      | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/packages/core/useMemoize/index.test.ts b/packages/core/useMemoize/index.test.ts
index 8acd03e8a47..b6e4ca56191 100644
--- 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
index a414ec53fe3..083e19fc979 100644
--- 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
   }