return offsetHeight on updateOffsetHeight

jsonkao · Jul 13, 2020 · 24f36a3 · 24f36a3
1 parent bed9ef9
commit 24f36a3
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/Step.js b/src/Step.js
@@ -14,10 +14,13 @@ class Step extends Component {
 
   getData = () => this.props.data;
 
-  updateOffsetHeight = () =>
+  updateOffsetHeight = () =>{
+    const offsetHeight = this.getDOMNode().offsetHeight;
     this.setState({
-      offsetHeight: this.getDOMNode().offsetHeight,
+      offsetHeight,
     });
+    return offsetHeight;
+  }
 
   enter = direction => this.setState({ state: 'enter', direction });
   exit = direction => this.setState({ state: 'exit', direction });