fix: restore state works in react router

Fixes #1076
petyosi · Apr 21, 2024 · bbc0967 · bbc0967
1 parent 461a5c9
commit bbc0967
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/examples/state.tsx b/examples/state.tsx
@@ -6,7 +6,6 @@ export function Example() {
   const state = React.useRef<StateSnapshot | undefined>(undefined)
   const [key, setKey] = React.useState(0)
 
-  console.log('Rendering with key', key)
   return (
     <div>
       <button
@@ -17,8 +16,9 @@ export function Example() {
           setKey((value) => value + 1)
         }}
       >
-        Log State
+        Save state and reload
       </button>
+      <button onClick={() => setKey((value) => value + 1)}>Reload</button>
 
       <Virtuoso
         key={key}

diff --git a/src/initialTopMostItemIndexSystem.ts b/src/initialTopMostItemIndexSystem.ts
@@ -52,7 +52,7 @@ export const initialTopMostItemIndexSystem = u.system(
           u.publish(initialItemFinalLocationReached, true)
         })
 
-        skipFrames(3, () => {
+        skipFrames(4, () => {
           u.handleNext(scrollTop, () => {
             u.publish(scrolledToInitialItem, true)
           })