React:如何获得元素的高度，这些元素消失并以新的大小出现

constructor(props) {
  super(props);
  this.elementRef = React.createRef();
}

render() {
  return (
    <div ref={this.elementRef}>
      {/* 元素内容 */}
    </div>
  );
}

componentDidMount() {
  const element = this.elementRef.current;
  const height = element.offsetHeight;
  // 处理获取到的高度
}

constructor(props) {
  super(props);
  this.state = {
    showElement: false
  };
}

render() {
  return (
    <div>
      {this.state.showElement && (
        <div ref={this.elementRef}>
          {/* 元素内容 */}
        </div>
      )}
      {/* 其他内容 */}
    </div>
  );
}

handleShowElement() {
  this.setState({ showElement: true });
}