fix(router): use startTransition (#742)

follow up #737
dai-shi · Jun 5, 2024 · 5edb781 · 5edb781
1 parent 97f5bc8
commit 5edb781
Showing 1 changed file with 13 additions and 8 deletions.
diff --git a/packages/waku/src/router/client.ts b/packages/waku/src/router/client.ts
@@ -4,6 +4,7 @@ import {
   Component,
   createContext,
   createElement,
+  startTransition,
   useCallback,
   useContext,
   useEffect,
@@ -333,7 +334,9 @@ const InnerRouter = ({ routerData }: { routerData: RouterData }) => {
   const changeRoute: ChangeRoute = useCallback(
     (route, options) => {
       const { checkCache, skipRefetch } = options || {};
-      setRoute(route);
+      startTransition(() => {
+        setRoute(route);
+      });
       const componentIds = getComponentIds(route.path);
       if (
         checkCache &&
@@ -364,14 +367,16 @@ const InnerRouter = ({ routerData }: { routerData: RouterData }) => {
           ]),
         );
       }
-      setCached((prev) => ({
-        ...prev,
-        ...Object.fromEntries(
-          componentIds.flatMap((id) =>
-            skip.includes(id) ? [] : [[id, route]],
+      startTransition(() => {
+        setCached((prev) => ({
+          ...prev,
+          ...Object.fromEntries(
+            componentIds.flatMap((id) =>
+              skip.includes(id) ? [] : [[id, route]],
+            ),
           ),
-        ),
-      }));
+        }));
+      });
     },
     [refetch, routerData],
   );