mirror of
https://github.com/ant-design/ant-design.git
synced 2024-12-05 17:59:57 +08:00
08f25f0c8f
* chore: bump rc-table ver * chore: Fix ts definition * docs: fix doc
55 lines
1.4 KiB
TypeScript
55 lines
1.4 KiB
TypeScript
import * as React from 'react';
|
|
import { Key, GetRowKey } from '../interface';
|
|
|
|
interface MapCache<RecordType> {
|
|
data?: readonly RecordType[];
|
|
childrenColumnName?: string;
|
|
kvMap?: Map<Key, RecordType>;
|
|
getRowKey?: Function;
|
|
}
|
|
|
|
export default function useLazyKVMap<RecordType>(
|
|
data: readonly RecordType[],
|
|
childrenColumnName: string,
|
|
getRowKey: GetRowKey<RecordType>,
|
|
) {
|
|
const mapCacheRef = React.useRef<MapCache<RecordType>>({});
|
|
|
|
function getRecordByKey(key: Key): RecordType {
|
|
if (
|
|
!mapCacheRef.current ||
|
|
mapCacheRef.current.data !== data ||
|
|
mapCacheRef.current.childrenColumnName !== childrenColumnName ||
|
|
mapCacheRef.current.getRowKey !== getRowKey
|
|
) {
|
|
const kvMap = new Map<Key, RecordType>();
|
|
|
|
/* eslint-disable no-inner-declarations */
|
|
function dig(records: readonly RecordType[]) {
|
|
records.forEach((record, index) => {
|
|
const rowKey = getRowKey(record, index);
|
|
kvMap.set(rowKey, record);
|
|
|
|
if (record && typeof record === 'object' && childrenColumnName in record) {
|
|
dig((record as any)[childrenColumnName] || []);
|
|
}
|
|
});
|
|
}
|
|
/* eslint-enable */
|
|
|
|
dig(data);
|
|
|
|
mapCacheRef.current = {
|
|
data,
|
|
childrenColumnName,
|
|
kvMap,
|
|
getRowKey,
|
|
};
|
|
}
|
|
|
|
return mapCacheRef.current.kvMap!.get(key)!;
|
|
}
|
|
|
|
return [getRecordByKey];
|
|
}
|