如何将标题图标的排序功能链接到React中要排序的列

import React, { useState } from 'react';

const Table = () => {
  const [sortColumn, setSortColumn] = useState(null);
  const [sortOrder, setSortOrder] = useState('asc');
  const data = [
    { id: 1, name: 'John', age: 25 },
    { id: 2, name: 'Jane', age: 30 },
    { id: 3, name: 'Bob', age: 20 }
  ];

  const handleSort = (column) => {
    if (sortColumn === column) {
      // 切换排序方式
      setSortOrder(sortOrder === 'asc' ? 'desc' : 'asc');
    } else {
      // 设置新的排序列和默认排序方式
      setSortColumn(column);
      setSortOrder('asc');
    }
  };

  const sortedData = data.sort((a, b) => {
    if (sortColumn) {
      // 根据排序列和排序方式进行排序
      const aValue = a[sortColumn];
      const bValue = b[sortColumn];
      if (sortOrder === 'asc') {
        return aValue > bValue ? 1 : -1;
      } else {
        return aValue < bValue ? 1 : -1;
      }
    }
    return 0;
  });

  return (
    <table>
      <thead>
        <tr>
          <th onClick={() => handleSort('id')}>
            ID {sortColumn === 'id' && (sortOrder === 'asc' ? '↑' : '↓')}
          </th>
          <th onClick={() => handleSort('name')}>
            Name {sortColumn === 'name' && (sortOrder === 'asc' ? '↑' : '↓')}
          </th>
          <th onClick={() => handleSort('age')}>
            Age {sortColumn === 'age' && (sortOrder === 'asc' ? '↑' : '↓')}
          </th>
        </tr>
      </thead>
      <tbody>
        {sortedData.map((item) => (
          <tr key={item.id}>
            <td>{item.id}</td>
            <td>{item.name}</td>
            <td>{item.age}</td>
          </tr>
        ))}
      </tbody>
    </table>
  );
};

export default Table;