如何将antd表排序器从升序或降序重置为正常？

import React, { useState } from 'react';
import { Table } from 'antd';

const dataSource = [
  { key: '1', name: 'John', age: 32 },
  { key: '2', name: 'Mike', age: 28 },
  { key: '3', name: 'Tom', age: 36 },
];

const columns = [
  {
    title: 'Name',
    dataIndex: 'name',
    key: 'name',
    sorter: true, // 开启排序功能
  },
  {
    title: 'Age',
    dataIndex: 'age',
    key: 'age',
    sorter: true, // 开启排序功能
  },
];

const MyTable = () => {
  const [tableData, setTableData] = useState(dataSource);

  const handleResetSort = (column) => {
    const { field, order } = column;

    // 重置排序状态
    setTableData((prevData) => {
      const newData = [...prevData];
      newData.sort((a, b) => a.key - b.key); // 恢复原始顺序
      return newData;
    });
  };

  return (
    <Table
      dataSource={tableData}
      columns={columns.map((column) => ({
        ...column,
        sortOrder: null, // 重置排序状态
        onHeaderCell: () => ({
          onClick: () => handleResetSort(column),
        }),
      }))}
    />
  );
};

export default MyTable;