let data = [
{
id: 1
},
{
id: 2,
child: [
{
id: 3
},
{
id: 4,
child: [
{
id: 5
}
]
},
{
id: 6,
child: [
{
id: 7
}
]
},
]
},
];
//法一、递归
const print = (item, num) => console.log(item, num)
function traversalData (data, num, callback) {
data.forEach((item)=>{
callback(item, num)
if (item.child) {
traversalData(item.child, num + 1, callback)
}
})
}
traversalData(data, 1, print)
//法二、es6生成器
function* traversalData (data, n = 1) {
for(let i = 0; i< data.length; i++) {
data[i].level = n;
yield data[i]
if (data[i].child) {
yield* traversalData(data[i].child, n + 1)
}
}
}
for ( let item of traversalData(data)) {
console.log(item, 'item')
}
//层级数组
[...traversalData(data)].map(item => item.level)
Array.from(traversalData(data)).map(item => item.level)