mirror of
https://gitee.com/ant-design/ant-design.git
synced 2024-12-15 17:31:25 +08:00
527c71d207
* docs: table demo in dark mode * chore: update snaptshot * chore: update
141 lines
3.8 KiB
TypeScript
141 lines
3.8 KiB
TypeScript
import React, { useEffect, useRef, useState } from 'react';
|
|
import { Table, theme } from 'antd';
|
|
import type { TableProps } from 'antd';
|
|
import classNames from 'classnames';
|
|
import ResizeObserver from 'rc-resize-observer';
|
|
import { VariableSizeGrid as Grid } from 'react-window';
|
|
|
|
const VirtualTable = <RecordType extends object>(props: TableProps<RecordType>) => {
|
|
const { columns, scroll } = props;
|
|
const [tableWidth, setTableWidth] = useState(0);
|
|
const { token } = theme.useToken();
|
|
|
|
const widthColumnCount = columns!.filter(({ width }) => !width).length;
|
|
const mergedColumns = columns!.map((column) => {
|
|
if (column.width) {
|
|
return column;
|
|
}
|
|
|
|
return {
|
|
...column,
|
|
width: Math.floor(tableWidth / widthColumnCount),
|
|
};
|
|
});
|
|
|
|
const gridRef = useRef<any>();
|
|
const [connectObject] = useState<any>(() => {
|
|
const obj = {};
|
|
Object.defineProperty(obj, 'scrollLeft', {
|
|
get: () => {
|
|
if (gridRef.current) {
|
|
return gridRef.current?.state?.scrollLeft;
|
|
}
|
|
return null;
|
|
},
|
|
set: (scrollLeft: number) => {
|
|
if (gridRef.current) {
|
|
gridRef.current.scrollTo({ scrollLeft });
|
|
}
|
|
},
|
|
});
|
|
|
|
return obj;
|
|
});
|
|
|
|
const resetVirtualGrid = () => {
|
|
gridRef.current?.resetAfterIndices({
|
|
columnIndex: 0,
|
|
shouldForceUpdate: true,
|
|
});
|
|
};
|
|
|
|
useEffect(() => resetVirtualGrid, [tableWidth]);
|
|
|
|
const renderVirtualList = (rawData: object[], { scrollbarSize, ref, onScroll }: any) => {
|
|
ref.current = connectObject;
|
|
const totalHeight = rawData.length * 54;
|
|
|
|
return (
|
|
<Grid
|
|
ref={gridRef}
|
|
className="virtual-grid"
|
|
columnCount={mergedColumns.length}
|
|
columnWidth={(index: number) => {
|
|
const { width } = mergedColumns[index];
|
|
return totalHeight > scroll!.y! && index === mergedColumns.length - 1
|
|
? (width as number) - scrollbarSize - 1
|
|
: (width as number);
|
|
}}
|
|
height={scroll!.y as number}
|
|
rowCount={rawData.length}
|
|
rowHeight={() => 54}
|
|
width={tableWidth}
|
|
onScroll={({ scrollLeft }: { scrollLeft: number }) => {
|
|
onScroll({ scrollLeft });
|
|
}}
|
|
>
|
|
{({
|
|
columnIndex,
|
|
rowIndex,
|
|
style,
|
|
}: {
|
|
columnIndex: number;
|
|
rowIndex: number;
|
|
style: React.CSSProperties;
|
|
}) => (
|
|
<div
|
|
className={classNames('virtual-table-cell', {
|
|
'virtual-table-cell-last': columnIndex === mergedColumns.length - 1,
|
|
})}
|
|
style={{
|
|
...style,
|
|
boxSizing: 'border-box',
|
|
padding: token.padding,
|
|
borderBottom: `${token.lineWidth}px ${token.lineType} ${token.colorSplit}`,
|
|
background: token.colorBgContainer,
|
|
}}
|
|
>
|
|
{(rawData[rowIndex] as any)[(mergedColumns as any)[columnIndex].dataIndex]}
|
|
</div>
|
|
)}
|
|
</Grid>
|
|
);
|
|
};
|
|
|
|
return (
|
|
<ResizeObserver
|
|
onResize={({ width }) => {
|
|
setTableWidth(width);
|
|
}}
|
|
>
|
|
<Table
|
|
{...props}
|
|
className="virtual-table"
|
|
columns={mergedColumns}
|
|
pagination={false}
|
|
components={{
|
|
body: renderVirtualList,
|
|
}}
|
|
/>
|
|
</ResizeObserver>
|
|
);
|
|
};
|
|
|
|
// Usage
|
|
const columns = [
|
|
{ title: 'A', dataIndex: 'key', width: 150 },
|
|
{ title: 'B', dataIndex: 'key' },
|
|
{ title: 'C', dataIndex: 'key' },
|
|
{ title: 'D', dataIndex: 'key' },
|
|
{ title: 'E', dataIndex: 'key', width: 200 },
|
|
{ title: 'F', dataIndex: 'key', width: 100 },
|
|
];
|
|
|
|
const data = Array.from({ length: 100000 }, (_, key) => ({ key }));
|
|
|
|
const App: React.FC = () => (
|
|
<VirtualTable columns={columns} dataSource={data} scroll={{ y: 300, x: '100vw' }} />
|
|
);
|
|
|
|
export default App;
|