move layout to only trigger during react lifecycle

bobthekingofegypt · Mar 12, 2021 · f2b1d89 · f2b1d89
1 parent 721e6c6
commit f2b1d89
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/DagreReact.tsx b/src/DagreReact.tsx
@@ -162,6 +162,12 @@ export default class DagreReact extends React.Component<
 
   checkRender() {
     if (this.state.graph.layoutIfSized()) {
+      this.forceUpdate();
+    }
+  }
+
+  componentDidUpdate() {
+    if (this.state.graph.dirty) {
       // console.log("Forcing an update");
       this.state.graph.layout();
       this.adjustIntersections();