如何在componentDidMount中设置按钮焦点

import React, { Component } from 'react';

class MyComponent extends Component {
  componentDidMount() {
    // 假设按钮的id是'myButton'
    const button = document.getElementById('myButton');
    if (button) {
      button.focus();
    }
  }

  render() {
    return (
      <div>
        <button id="myButton">Click Me</button>
      </div>
    );
  }
}

export default MyComponent;

import React, { Component } from 'react';

class MyComponent extends Component {
  constructor(props) {
    super(props);
    this.myButtonRef = React.createRef();
  }

  componentDidMount() {
    if (this.myButtonRef.current) {
      this.myButtonRef.current.focus();
    }
  }

  render() {
    return (
      <div>
        <button ref={this.myButtonRef}>Click Me</button>
      </div>
    );
  }
}

export default MyComponent;