如何在React中向所单击的按钮添加类，以及向先前所单击的按钮移除类

import React, { useState } from 'react';

function MyComponent() {
  const [activeButton, setActiveButton] = useState(null);

  // 其他组件代码...

  return (
    <div>
      <button
        className={activeButton === 'button1' ? 'active' : ''}
        onClick={() => setActiveButton('button1')}
      >
        Button 1
      </button>
      <button
        className={activeButton === 'button2' ? 'active' : ''}
        onClick={() => setActiveButton('button2')}
      >
        Button 2
      </button>
    </div>
  );
}