feat(hooks): add useNodesState and useEdgesState

This commit is contained in:
moklick
2021-11-26 14:39:22 +01:00
parent 713e7e1bd4
commit a032306408
8 changed files with 66 additions and 76 deletions
+4 -14
View File
@@ -3,17 +3,15 @@ import { useState, MouseEvent, useCallback } from 'react';
import ReactFlow, {
addEdge,
Background,
applyNodeChanges,
applyEdgeChanges,
MiniMap,
Controls,
Node,
Edge,
NodeChange,
EdgeChange,
OnLoadParams,
Connection,
MarkerType,
useNodesState,
useEdgesState,
} from 'react-flow-renderer';
import DebugNode from './DebugNode';
@@ -108,8 +106,8 @@ const nodeTypes = {
const BasicFlow = () => {
const [rfInstance, setRfInstance] = useState<OnLoadParams | null>(null);
const [nodes, setNodes] = useState<Node[]>(initialNodes);
const [edges, setEdges] = useState<Edge[]>(initialEdges);
const [nodes, setNodes, onNodesChange] = useNodesState(initialNodes);
const [edges, setEdges, onEdgesChange] = useEdgesState(initialEdges);
const onConnect = useCallback((connection: Connection) => {
setEdges((eds) => {
@@ -152,14 +150,6 @@ const BasicFlow = () => {
});
};
const onNodesChange = useCallback((changes: NodeChange[]) => {
setNodes((ns) => applyNodeChanges(changes, ns));
}, []);
const onEdgesChange = useCallback((changes: EdgeChange[]) => {
setEdges((es) => applyEdgeChanges(changes, es));
}, []);
return (
<ReactFlow
nodes={nodes}