rsuite-table
Version:
A React table component
14 lines (13 loc) • 385 B
text/typescript
export default function findRowKeys(rows: any[], rowKey: string | number, expanded?: boolean) {
let keys = [];
for (let i = 0; i < rows.length; i++) {
const item = rows[i];
if (item.children) {
keys.push(item[rowKey]);
keys = [...keys, ...findRowKeys(item.children, rowKey)];
} else if (expanded) {
keys.push(item[rowKey]);
}
}
return keys;
}