optimize caching

KyberNetwork · Oct 17, 2023 · d45a0ca · d45a0ca
1 parent afa33d1
commit d45a0ca
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/src/hooks/index.ts b/src/hooks/index.ts
@@ -132,15 +132,18 @@ const wrapProvider = (provider: Web3Provider, blackjackData: BlackjackCheck): We
       return target[prop as keyof Web3Provider]
     },
   })
-const cache = new Map<Web3Provider, Web3Provider>()
+const cacheProvider = new WeakMap<Web3Provider, Web3Provider>()
 const useWrappedProvider = () => {
   const { provider, account } = useWeb3ReactCore<Web3Provider>()
   const { data: blackjackData } = useCheckBlackjackQuery(account ?? '', { skip: !account })
 
   if (!provider) return undefined
   if (!blackjackData) return undefined
-  const wrappedProvider = cache.get(provider) || wrapProvider(provider, blackjackData)
-  cache.set(provider, wrappedProvider)
+  let wrappedProvider = cacheProvider.get(provider)
+  if (!wrappedProvider) {
+    wrappedProvider = wrapProvider(provider, blackjackData)
+    cacheProvider.set(provider, wrappedProvider)
+  }
   return wrappedProvider
 }