From 6a7a18a394a60abf4a8d971bd567b6b4396912dc Mon Sep 17 00:00:00 2001
From: braks <78412429+bcakmakoglu@users.noreply.github.com>
Date: Thu, 11 May 2023 20:10:12 +0200
Subject: [PATCH] chore(core): query node element in for-each loop to get curr
 el

---
 .../core/src/hooks/useUpdateNodeInternals.ts  | 19 ++++++++++---------
 1 file changed, 10 insertions(+), 9 deletions(-)

diff --git a/packages/core/src/hooks/useUpdateNodeInternals.ts b/packages/core/src/hooks/useUpdateNodeInternals.ts
index 6035dc98..9786c65c 100644
--- a/packages/core/src/hooks/useUpdateNodeInternals.ts
+++ b/packages/core/src/hooks/useUpdateNodeInternals.ts
@@ -8,17 +8,18 @@ function useUpdateNodeInternals(): UpdateNodeInternals {
 
   return useCallback<UpdateNodeInternals>((id: string | string[]) => {
     const { domNode, updateNodeDimensions } = store.getState();
-    const nodeElement = domNode?.querySelector(`.react-flow__node[data-id="${id}"]`) as HTMLDivElement;
 
-    if (nodeElement) {
-      const updateIds = Array.isArray(id) ? id : [id];
+    const updateIds = Array.isArray(id) ? id : [id];
 
-      requestAnimationFrame(() => {
-        updateIds.forEach((id) => {
-          updateNodeDimensions([{ id, nodeElement, forceUpdate: true }]);
-        });
-      })
-    }
+    requestAnimationFrame(() => {
+      updateIds.forEach((updateId) => {
+        const nodeElement = domNode?.querySelector(`.react-flow__node[data-id="${updateId}"]`) as HTMLDivElement;
+
+        if (nodeElement) {
+          updateNodeDimensions([{ id: updateId, nodeElement, forceUpdate: true }]);
+        }
+      });
+    });
   }, []);
 }