目标
理解数组转树的代码,能照着注释写出来
/**
* 把平铺的数组结构转成树形结构
*
* [
* {id:"01", pid:"", "name":"老王" },
* {id:"02", pid:"01", "name":"小张" }
* ]
* 上面的结构说明: 老王是小张的上级
* @param {*} list
* @returns
*/
// [
// { 'id': '3129', 'pid': '', 'name': '总裁办' },
// { 'id': '312a', 'pid': '', 'name': '行政部' },
// { 'id': '312b', 'pid': '', 'name': '人事部' },
// { 'id': '312c', 'pid': '', 'name': '财务部' },
// { 'id': '312d', 'pid': '312c', 'name': '财务核算部'},
// { 'id': '312e', 'pid': '312c', 'name': '税务管理部'},
// { 'id': '312f', 'pid': '312c', 'name': '薪资管理部'},
// { 'id': '00d2', 'pid': '', 'name': '技术部'},
// { 'id': '00d3', 'pid': '00d2', 'name': 'Java研发部' },
// { 'id': '00d4', 'pid': '00d2', 'name': 'Python研发部'},
// { 'id': '00d5', 'pid': '00d2', 'name': 'Php研发部'},
// { 'id': '00d6', 'pid': '', 'name': '运营部' },
// { 'id': '00d7', 'pid': '', 'name': '市场部'},
// { 'id': '00d8', 'pid': '00d7', 'name': '北京事业部'},
// { 'id': '00d9', 'pid': '00d7', 'name': '上海事业部'}
// ]
export function tranListToTreeData(list) {
// 最终要产出的树状数据的数组
const treeList = []
// 所有项都使用对象存储起来
const map = {}
// 建立一个映射关系:通过id快速找到对应的元素
list.forEach(item => {
if (!item.children) {
item.children = []
}
map[item.id] = item
})
// map:
// {
// "312c": { 'id': '312c', 'pid': '', 'name': '财务部', children: [{ 'id': '312d', 'pid': '312c', 'name': '财务核算部',children: []}] },
// "312d": { 'id': '312d', 'pid': '312c', 'name': '财务核算部',children: []}
// }
list.forEach(item => {
// 对于每一个元素来说,先找它的上级
// 如果能找到,说明它有上级,则要把它添加到上级的children中去
// 如果找不到,说明它没有上级,直接添加到 treeList
const parent = map[item.pid]
// 如果存在则表示item不是最顶层的数据
if (parent) {
parent.children.push(item)
} else {
// 如果不存在 则是顶层数据
treeList.push(item)
}
})
// 返回出去
return treeList
}