From e0f8ffe49cf722b729428a77b37021e12e1106a2 Mon Sep 17 00:00:00 2001
From: dream2023 <1098626505@qq.com>
Date: Thu, 29 Feb 2024 12:05:57 +0800
Subject: [PATCH 1/2] fix:  `useStyleRegister` return function uses useCallback
 wrapping

---
 src/hooks/useStyleRegister.tsx | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/hooks/useStyleRegister.tsx b/src/hooks/useStyleRegister.tsx
index d2474a9..a936e3b 100644
--- a/src/hooks/useStyleRegister.tsx
+++ b/src/hooks/useStyleRegister.tsx
@@ -491,7 +491,7 @@ export default function useStyleRegister(
       },
     );
 
-  return (node: React.ReactElement) => {
+  return useCallback((node: React.ReactElement) => {
     let styleNode: React.ReactElement;
 
     if (!ssrInline || isMergedClientSide || !defaultCache) {
@@ -514,7 +514,7 @@ export default function useStyleRegister(
         {node}
       </>
     );
-  };
+  }, [ssrInline, isMergedClientSide, defaultCache, cachedStyleStr, cachedTokenKey, cachedStyleId]);
 }
 
 export const extract: ExtractStyle<StyleCacheValue> = (

From 4235ff43fb1d1bcb3b1c32540f2c4cdd8cdac583 Mon Sep 17 00:00:00 2001
From: dream2023 <1098626505@qq.com>
Date: Thu, 29 Feb 2024 16:40:06 +0800
Subject: [PATCH 2/2] fix: bug

---
 src/hooks/useStyleRegister.tsx | 54 ++++++++++++++++++++--------------
 1 file changed, 32 insertions(+), 22 deletions(-)

diff --git a/src/hooks/useStyleRegister.tsx b/src/hooks/useStyleRegister.tsx
index a936e3b..00decaf 100644
--- 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> = (