主要用到了reduce()函数
for循环可以做到的事情reduce()都可以做到,甚至操作起来更加简单方便和高雅。reduce 为数组中的每一个元素依次执行回调函数,不包括数组中被删除或从未被赋值的元素
reduce语法:
array.reduce(function(pre, currentValue, currentIndex, arr), initialValue)
- recude函数接收两个参数,一个为回调函数,一个为初始赋值
- 第一个参数的回调函数又接收四个参数,分别为(初始值或计算结束后的返回值,当前元素,当前元素的索引,当前元素所属的数组对象)
- 第二个参数是传给函数的初始值,非必传
代码:
test() {
const list = [
{ id: 1, name: 'a' },
{ id: 11, name: 'aa', pid: 1 },
{ id: 12, name: 'aaa', pid: 1 },
{ id: 13, name: 'aaaa', pid: 1 },
{ id: 2, name: 'b' },
{ id: 22, name: 'bb', pid: 2 },
{ id: 23, name: 'bbb', pid: 2 },
{ id: 24, name: 'bbbb', pid: 2 },
{ id: 3, name: 'c' },
{ id: 31, name: 'cc', pid: 3 },
{ id: 32, name: 'ccc', pid: 3 },
{ id: 33, name: 'cccc', pid: 3 },
{ id: 111, name: 'a11', pid: 11 },
{ id: 112, name: 'aa11', pid: 11 },
{ id: 1121, name: 'aa112', pid: 112 }
];
//prev 是我们最终要生成的树型机构
//curr 为当前的数据
//i为索引
//arr为我们整个list
const result = list.reduce(function (prev, curr, i, arr) {
curr.children = arr.filter((v) => v.pid === curr.id);
if (!curr.pid) {
prev.push(curr);
}
return prev;
}, []);
console.log('list', list);
console.log('tree', result);
},