alighasami/xyflow
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

chore(BatchProvider): cleanup

Browse Source
This commit is contained in:
moklick
2024-04-25 15:08:10 +02:00
parent a7b6789ce0
commit 95fd22d25b
4 changed files with 81 additions and 45 deletions
Download Patch File Download Diff File Expand all files Collapse all files
packages/react/src/hooks/useReactFlow.ts
+14 -30
View File
@@ -50,39 +50,23 @@ export function useReactFlow<NodeType extends Node = Node, EdgeType extends Edge
const getEdge = useCallback<Instance.GetEdge<EdgeType>>((id) => store.getState().edgeLookup.get(id) as EdgeType, []);
const setNodes = useCallback<Instance.SetNodes<NodeType>>(
(payload) => {
batchContext?.nodeQueue.push(payload as NodeType[]);
},
[batchContext]
);
const setNodes = useCallback<Instance.SetNodes<NodeType>>((payload) => {
batchContext.nodeQueue.push(payload as NodeType[]);
}, []);
const setEdges = useCallback<Instance.SetEdges<EdgeType>>(
(payload) => {
batchContext?.edgeQueue.push(payload as EdgeType[]);
},
[batchContext]
);
const setEdges = useCallback<Instance.SetEdges<EdgeType>>((payload) => {
batchContext.edgeQueue.push(payload as EdgeType[]);
}, []);
const addNodes = useCallback<Instance.AddNodes<NodeType>>(
(payload) => {
const newNodes = Array.isArray(payload) ? payload : [payload];
const addNodes = useCallback<Instance.AddNodes<NodeType>>((payload) => {
const newNodes = Array.isArray(payload) ? payload : [payload];
batchContext.nodeQueue.push((nodes) => [...nodes, ...newNodes]);
}, []);
// Queueing a functional update means that we won't worry about other calls
// to `setNodes` that might happen elsewhere.
batchContext?.nodeQueue.push((nodes) => [...nodes, ...newNodes]);
},
[batchContext]
);
const addEdges = useCallback<Instance.AddEdges<EdgeType>>(
(payload) => {
const newEdges = Array.isArray(payload) ? payload : [payload];
batchContext?.edgeQueue.push((edges) => [...edges, ...newEdges]);
},
[batchContext]
);
const addEdges = useCallback<Instance.AddEdges<EdgeType>>((payload) => {
const newEdges = Array.isArray(payload) ? payload : [payload];
batchContext.edgeQueue.push((edges) => [...edges, ...newEdges]);
}, []);
const toObject = useCallback<Instance.ToObject<NodeType, EdgeType>>(() => {
const { nodes = [], edges = [], transform } = store.getState();