fix: bug

ant-design · Feb 29, 2024 · 4235ff4 · 4235ff4
1 parent e0f8ffe
commit 4235ff4
Showing 1 changed file with 32 additions and 22 deletions.
diff --git a/src/hooks/useStyleRegister.tsx b/src/hooks/useStyleRegister.tsx
@@ -491,30 +491,40 @@ export default function useStyleRegister(
       },
     );
 
-  return useCallback((node: React.ReactElement) => {
-    let styleNode: React.ReactElement;
+  return React.useCallback(
+    (node: React.ReactElement) => {
+      let styleNode: React.ReactElement;
+
+      if (!ssrInline || isMergedClientSide || !defaultCache) {
+        styleNode = <Empty />;
+      } else {
+        styleNode = (
+          <style
+            {...{
+              [ATTR_TOKEN]: cachedTokenKey,
+              [ATTR_MARK]: cachedStyleId,
+            }}
+            dangerouslySetInnerHTML={{ __html: cachedStyleStr }}
+          />
+        );
+      }
 
-    if (!ssrInline || isMergedClientSide || !defaultCache) {
-      styleNode = <Empty />;
-    } else {
-      styleNode = (
-        <style
-          {...{
-            [ATTR_TOKEN]: cachedTokenKey,
-            [ATTR_MARK]: cachedStyleId,
-          }}
-          dangerouslySetInnerHTML={{ __html: cachedStyleStr }}
-        />
+      return (
+        <>
+          {styleNode}
+          {node}
+        </>
       );
-    }
-
-    return (
-      <>
-        {styleNode}
-        {node}
-      </>
-    );
-  }, [ssrInline, isMergedClientSide, defaultCache, cachedStyleStr, cachedTokenKey, cachedStyleId]);
+    },
+    [
+      ssrInline,
+      isMergedClientSide,
+      defaultCache,
+      cachedStyleStr,
+      cachedTokenKey,
+      cachedStyleId,
+    ],
+  );
 }
 
 export const extract: ExtractStyle<StyleCacheValue> = (