fix(provider): handle changed nodes and edges
This commit is contained in:
+3
-33
@@ -14,44 +14,14 @@ const GraphWrapper = styled.div`
|
||||
`;
|
||||
|
||||
class ReactGraph extends PureComponent {
|
||||
constructor(props) {
|
||||
super();
|
||||
|
||||
const { elements } = props;
|
||||
|
||||
this.state = {
|
||||
...separateElements(elements)
|
||||
};
|
||||
}
|
||||
|
||||
componentDidUpdate(prevProps, prevState) {
|
||||
const { elements } = this.props;
|
||||
const { nodes, edges } = separateElements(elements);
|
||||
const nodesChanged = !isEqual(nodes, prevState.nodes);
|
||||
const edgesChanged = !isEqual(edges, prevState.edges);
|
||||
|
||||
if (!nodesChanged && !edgesChanged) {
|
||||
return false;
|
||||
}
|
||||
|
||||
if (nodesChanged) {
|
||||
this.setState({ nodes });
|
||||
}
|
||||
|
||||
if (edgesChanged) {
|
||||
this.setState({ edges });
|
||||
}
|
||||
}
|
||||
|
||||
render() {
|
||||
const {
|
||||
style, onNodeClick, children, onLoad, onMove
|
||||
} = this.props;
|
||||
const { nodes, edges } = this.state;
|
||||
style, onNodeClick, children, onLoad, onMove, elements
|
||||
} = this.props;
|
||||
|
||||
return (
|
||||
<GraphWrapper style={style}>
|
||||
<Provider nodes={nodes} edges={edges} onNodeClick={onNodeClick}>
|
||||
<Provider {...separateElements(elements)} onNodeClick={onNodeClick}>
|
||||
<GraphView onLoad={onLoad} onMove={onMove} />
|
||||
{children}
|
||||
</Provider>
|
||||
|
||||
Reference in New Issue
Block a user