如何根据React中点击的按钮切换div类？

import React, { Component } from 'react';

class MyComponent extends Component {
  constructor(props) {
    super(props);
    this.state = {
      currentDivClass: 'div1', // 初始div类名
    };
  }

  handleButtonClick = () => {
    // 根据当前div的类名切换为另一个类名
    this.setState(prevState => ({
      currentDivClass: prevState.currentDivClass === 'div1' ? 'div2' : 'div1',
    }));
  };

  render() {
    const { currentDivClass } = this.state;
    return (
      <div>
        <button onClick={this.handleButtonClick}>切换div</button>
        <div className={currentDivClass}>我是{currentDivClass}</div>
      </div>
    );
  }
}

export default MyComponent;