React+Hook+ts+antDesign实现table行编辑功能（1）

<Table
                     onChange={onTableChange}
                        rowKey="id"
                        components={components}
                        rowClassName={() => 'editable-row'}
                        bordered
                        dataSource={menuData.list}
                        pagination={{
                            pageSizeOptions: ['5', '10', '20', '50'],
                            total: menuData?.page.dataTotal,
                            showTotal: () => {
                                return '共 ' + menuData?.page.dataTotal + ' 条记录';
                            },
                        }}
                        columns={columns as ColumnTypes}
                        rowSelection={rowSelection}
                    />

const components = {
        body: {
            row: EditableRow,
            cell: EditableCell,
        },
    };

const EditableContext = React.createContext<FormInstance<any> | null>(null);
const EditableRow: React.FC<EditableRowProps> = ({ index, ...props }) => {
    const [form] = Form.useForm();
    return (
        <Form form={form} component={false}>
            <EditableContext.Provider value={form}>
                <tr {...props} />
            </EditableContext.Provider>
        </Form>
    );
};

const EditableCell: React.FC<EditableCellProps> = ({
    title,
    editable,
    children,
    dataIndex,
    record,
    handleSave,
    ...restProps
}) => {
    const [editing, setEditing] = useState(false);
    const inputRef = useRef<InputRef>(null);
    const form = useContext(EditableContext)!;
    useEffect(() => {
        if (editing) {
            inputRef.current!.focus();
        }
    }, [editing]);
    const toggleEdit = () => {
        setEditing(!editing);
        form.setFieldsValue({ [dataIndex]: record[dataIndex] });
    };
    const save = async () => {
        try {
            const values = await form.validateFields();
            toggleEdit();
            handleSave({ ...record, ...values });
        } catch (errInfo) {
            console.log('Save failed:', errInfo);
        }
    };
    let childNode = children;
    if (editable) {
        childNode = editing ? (
            <Form.Item
                style={{ margin: 0 }}
                name={dataIndex}
                rules={[
                    {
                        required: true,
                        message: `${title}不能为空`,
                    },
                ]}
            >
                <Input ref={inputRef} onPressEnter={save} onBlur={save} />
            </Form.Item>
        ) : (
            <div className="editable-cell-value-wrap" style={{ paddingRight: 24 }} onClick={toggleEdit}>
                {children}
            </div>
        );
    }
    return <td {...restProps}>{childNode}</td>;
};

const handleSave = (row: any) => {
        // console.log(batchUpdateSizeList)
        batchUpdateSizeList([{
            id: row.id,
            size: row.sizeName
        }])
            .unwrap()
            .then((response: any) => {
                if (response.code === 200) {
                    message.success("操作成功")
                }
            });
        const newData: any = [...menuData.list];
        const index = newData.findIndex((item: any) => row?.id == item.id);
        const item = newData[index];
        newData.splice(index, 1, {
            ...item,
            ...row,
        });
        setMenuData({
            list: newData, page: {
                dataTotal: menuData.page.dataTotal,
                size: menuData.page.size,
            }
        });
    };

const columns = defaultColumns.map(col => {
        if (!col.editable) {
            return col;
        }
        return {
            ...col,
            onCell: (record: any) => ({
                record,
                editable: col.editable,
                dataIndex: col.dataIndex,
                title: col.title,
                handleSave,
            }),
        };
    });