fix: tree replaceFields error #4298

pull/4301/head
tangjinzhou 2021-06-30 15:44:08 +08:00
parent e79a930e59
commit aa05d369e1
1 changed files with 7 additions and 8 deletions

View File

@ -99,13 +99,12 @@ export function convertDirectoryKeysToNodes(rootChildren: VNode[], keys: TreeKey
export function getFullKeyListByTreeData(treeData: TreeDataItem[], replaceFields: any = {}) {
let keys = [];
const { key = 'key', children = 'children' } = replaceFields(treeData || []).forEach(
(item: TreeDataItem) => {
keys.push(item[key]);
if (item[children]) {
keys = [...keys, ...getFullKeyListByTreeData(item[children], replaceFields)];
}
},
);
const { key = 'key', children = 'children' } = replaceFields;
(treeData || []).forEach((item: TreeDataItem) => {
keys.push(item[key]);
if (item[children]) {
keys = [...keys, ...getFullKeyListByTreeData(item[children], replaceFields)];
}
});
return keys;
}