mirror of
https://github.com/ant-design/ant-design.git
synced 2024-12-13 15:49:10 +08:00
14a1e6bd51
* feat: tsconfig enable strict * feat: add no-explicit-any * feat: strict * feat: as THEME * feat: 优化 keys 类型写法 * feat: demo remove any * feat: as number * feat: this any * feat: add eslint * feat: cascader * feat: props any * feat: remove any * feat: remove any * feat: any 提示错误 * feat: remove any * feat: add eslint * feat: 允许 T = any 存在 * feat: color funciton * feat: 恢复 lint * feat: merge master * feat: as ReactElement * feat: type
44 lines
1.4 KiB
TypeScript
44 lines
1.4 KiB
TypeScript
import * as React from 'react';
|
|
|
|
import type { KeyWise, TransferProps } from '..';
|
|
import { groupKeysMap } from '../../_util/transKeys';
|
|
import type { AnyObject } from '../../_util/type';
|
|
import type { TransferKey } from '../interface';
|
|
|
|
const useData = <RecordType extends AnyObject>(
|
|
dataSource?: RecordType[],
|
|
rowKey?: TransferProps<RecordType>['rowKey'],
|
|
targetKeys?: TransferKey[],
|
|
) => {
|
|
const mergedDataSource = React.useMemo(
|
|
() =>
|
|
(dataSource || []).map((record) => {
|
|
if (rowKey) {
|
|
record = { ...record, key: rowKey(record) };
|
|
}
|
|
return record;
|
|
}),
|
|
[dataSource, rowKey],
|
|
);
|
|
|
|
const [leftDataSource, rightDataSource] = React.useMemo(() => {
|
|
const leftData: KeyWise<RecordType>[] = [];
|
|
const rightData: KeyWise<RecordType>[] = new Array((targetKeys || []).length);
|
|
const targetKeysMap = groupKeysMap(targetKeys || []);
|
|
mergedDataSource.forEach((record) => {
|
|
// rightData should be ordered by targetKeys
|
|
// leftData should be ordered by dataSource
|
|
if (targetKeysMap.has(record.key)) {
|
|
(rightData as any)[targetKeysMap.get(record.key) as any] = record;
|
|
} else {
|
|
leftData.push(record as any);
|
|
}
|
|
});
|
|
return [leftData, rightData] as const;
|
|
}, [mergedDataSource, targetKeys, rowKey]);
|
|
|
|
return [mergedDataSource, leftDataSource, rightDataSource];
|
|
};
|
|
|
|
export default useData;
|