fix(Canvas): DnD related logs

KaotoIO · Jan 17, 2025 · cdb4548 · cdb4548
1 parent 8c32011
commit cdb4548
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 3 deletions.
diff --git a/packages/ui/src/components/Visualization/Canvas/Canvas.tsx b/packages/ui/src/components/Visualization/Canvas/Canvas.tsx
@@ -90,8 +90,16 @@ export const Canvas: FunctionComponent<PropsWithChildren<CanvasProps>> = ({ enti
       },
     };
 
-    controller.fromModel(model, false);
-    setInitialized(true);
+    if (!initialized) {
+      controller.fromModel(model, false);
+      setInitialized(true);
+      return;
+    }
+
+    requestAnimationFrame(() => {
+      controller.fromModel(model, true);
+      controller.getGraph().layout();
+    });
   }, [controller, entities, visibleFlows]);
 
   const handleSelection = useCallback((selectedIds: string[]) => {

diff --git a/packages/ui/src/components/Visualization/Custom/Node/CustomNode.tsx b/packages/ui/src/components/Visualization/Custom/Node/CustomNode.tsx
@@ -114,7 +114,10 @@ const CustomNodeInner: FunctionComponent<CustomNodeProps> = observer(
             nodes: [],
             edges: [],
           });
-          entitiesContext.updateEntitiesFromCamelResource();
+
+          requestAnimationFrame(() => {
+            entitiesContext.updateEntitiesFromCamelResource();
+          });
         } else {
           // Show all edges after dropping
           element