1

2026-05-25 17:07:28 +08:00
parent 30cba5ee96
commit df501f6151
10000 changed files with 276005 additions and 0 deletions
@@ -0,0 +1,77 @@
+import * as React from 'react';
+import { useInsertionEffect } from 'react';
+import { pathKey } from "../Cache";
+import StyleContext from "../StyleContext";
+import useHMR from "./useHMR";
+const effectMap = new Map();
+export default function useGlobalCache(prefix, keyPath, cacheFn, onCacheRemove,
+// Add additional effect trigger by `useInsertionEffect`
+onCacheEffect) {
+  const {
+    cache: globalCache
+  } = React.useContext(StyleContext);
+  const fullPath = [prefix, ...keyPath];
+  const fullPathStr = pathKey(fullPath);
+  const HMRUpdate = useHMR();
+  const buildCache = updater => {
+    globalCache.opUpdate(fullPathStr, prevCache => {
+      const [times = 0, cache] = prevCache || [undefined, undefined];
+
+      // HMR should always ignore cache since developer may change it
+      let tmpCache = cache;
+      if (process.env.NODE_ENV !== 'production' && cache && HMRUpdate) {
+        onCacheRemove?.(tmpCache, HMRUpdate);
+        tmpCache = null;
+      }
+      const mergedCache = tmpCache || cacheFn();
+      const data = [times, mergedCache];
+
+      // Call updater if need additional logic
+      return updater ? updater(data) : data;
+    });
+  };
+
+  // Create cache
+  React.useMemo(() => {
+    buildCache();
+  }, /* eslint-disable react-hooks/exhaustive-deps */
+  [fullPathStr]
+  /* eslint-enable */);
+  let cacheEntity = globalCache.opGet(fullPathStr);
+
+  // 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.opGet(fullPathStr);
+  }
+  const cacheContent = cacheEntity[1];
+
+  // Remove if no need anymore
+  useInsertionEffect(() => {
+    buildCache(([times, cache]) => [times + 1, cache]);
+    if (!effectMap.has(fullPathStr)) {
+      onCacheEffect?.(cacheContent);
+      effectMap.set(fullPathStr, true);
+
+      // 微任务清理缓存，可以认为是单次 batch render 中只触发一次 effect
+      Promise.resolve().then(() => {
+        effectMap.delete(fullPathStr);
+      });
+    }
+    return () => {
+      globalCache.opUpdate(fullPathStr, prevCache => {
+        const [times = 0, cache] = prevCache || [];
+        const nextCount = times - 1;
+        if (nextCount === 0) {
+          onCacheRemove?.(cache, false);
+          effectMap.delete(fullPathStr);
+          return null;
+        }
+        return [times - 1, cache];
+      });
+    };
+  }, [fullPathStr]);
+  return cacheContent;
+}