fix: hmr should rebuild cache (#137)

ant-design · Jul 26, 2023 · d907912 · d907912
1 parent c8a9393
commit d907912
Showing 1 changed file with 14 additions and 2 deletions.
diff --git a/src/hooks/useGlobalCache.tsx b/src/hooks/useGlobalCache.tsx
@@ -42,13 +42,25 @@ export default function useGlobalCache<CacheType>(
 
   // Create cache
   React.useMemo(
-    () => buildCache(),
+    () => {
+      buildCache();
+    },
     /* eslint-disable react-hooks/exhaustive-deps */
     [deps],
     /* eslint-enable */
   );
 
-  const cacheContent = globalCache.get(fullPath)![1];
+  let cacheEntity = globalCache.get(fullPath);
+
+  // HMR clean the cache but not trigger `useMemo` again
+  // Let's fallback of this
+  // ref https://github.com/ant-design/cssinjs/issues/127
+  if (process.env.NODE_ENV !== 'production' && !cacheEntity) {
+    buildCache();
+    cacheEntity = globalCache.get(fullPath);
+  }
+
+  const cacheContent = cacheEntity![1];
 
   // Remove if no need anymore
   useCompatibleInsertionEffect(