chore(useReactFlow): cleanup

2024-04-17 13:50:52 +02:00
parent 6bab47fcb7
commit 12c081a62e
2 changed files with 4 additions and 6 deletions
@@ -47,10 +47,7 @@ export function useReactFlow<NodeType extends Node = Node, EdgeType extends Edge
    return edges.map((e) => ({ ...e })) as EdgeType[];
  }, []);

-  const getEdge = useCallback<Instance.GetEdge<EdgeType>>((id) => {
-    const { edges = [] } = store.getState();
-    return edges.find((e) => e.id === id) as EdgeType;
-  }, []);
+  const getEdge = useCallback<Instance.GetEdge<EdgeType>>((id) => store.getState().edgeLookup.get(id) as EdgeType, []);

  type SetElementsQueue = {
    nodes: (NodeType[] | ((nodes: NodeType[]) => NodeType[]))[];