From aa873d6ddc69011c49a355f9b0c640d03dde3cfc Mon Sep 17 00:00:00 2001
From: peterkogo <peter@xyflow.com>
Date: Mon, 31 Mar 2025 13:47:15 +0200
Subject: [PATCH] enhance fitView scheduling

---
 packages/react/src/components/BatchProvider/index.tsx | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/packages/react/src/components/BatchProvider/index.tsx b/packages/react/src/components/BatchProvider/index.tsx
index 8a7d289a..382324da 100644
--- a/packages/react/src/components/BatchProvider/index.tsx
+++ b/packages/react/src/components/BatchProvider/index.tsx
@@ -52,7 +52,12 @@ export function BatchProvider<NodeType extends Node = Node, EdgeType extends Edg
       } else if (fitViewQueued) {
         // If there are no changes to the nodes, we still need to call setNodes
         // to trigger a re-render and fitView.
-        setNodes(next);
+        window.requestAnimationFrame(() => {
+          const { fitViewQueued, nodes, setNodes } = store.getState();
+          if (fitViewQueued) {
+            setNodes(nodes);
+          }
+        });
       }
     }
   }, []);