由于项目要做旭日图,由于后端只帮我提取的数据库中扁平化的对象数组,所以需要我来进行数据处理,为了这个我还得处理一下数据,下面是我的数据处理过程,。
// 本算法在不确定顶级节点的情况下,优先过滤出顶级节点,在进行遍历
// 进行数据的大量缓存,用空间换时间,让生成tree的速度像柔丝般顺滑
function generatorTree (data) {
const value = 'value';
const name = 'name';
const id = 'id';
const pid = 'pid';
// 保存所有ID,然后找到pid没有对应节点的就是顶级节点
const hasId = {};
// 顶层父节点数组
const topPIdArr = [];
// 将每个节点对应的子节点对应起来
const nodeToChildren = {};
// 获取所有父ID(包括顶级节点的父ID)
const pidSet = new Set();
data.forEach(e => {
// 之后通过topPIdArr数组可以通过深度遍历生成父子节点tree
if (nodeToChildren[e[pid]]) {
nodeToChildren[e[pid]].push({ name: e[name], value: e[value], id: e[id], children: [] });
} else {
nodeToChildren[e[pid]] = [{ name: e[name], value: e[value], id: e[id], children: [] }];
}
pidSet.add(e[pid]);
hasId[e[id]] = true;
});
// 遍历父ID集合
for (const pid of pidSet) {
if (!hasId[pid]) {
// 如果hasId没有该pid节点,说明它是顶级父节点
topPIdArr.push(pid);
}
}
// 用来存储生成的树状结构数据
const nodeTree = [];
// 找到所有顶级节点,开始递归生成tree,tree,tree,真的脆
data.filter(e => topPIdArr.includes(e[pid])).forEach((e, index) => {
nodeTree.push({ name: e[name], value: e[value], id: e[id], children: [] });
deepCreateTree(nodeTree[index].id, nodeTree[index].children);
});
// 通过深度优先遍历的思路,生成父子Tree
function deepCreateTree (id, children) {
if (nodeToChildren[id]) {
// 如果你不想改变nodeToChildren,建议深拷贝一下,当然里面如果属性有undefined,函数或者symbol那你得自己实现一个深拷贝方法
const copyNodeToChildren = JSON.parse(JSON.stringify(nodeToChildren[id]));
children.push(...copyNodeToChildren);
copyNodeToChildren.forEach(e => {
deepCreateTree(e.id, e.children);
});
}
}
return nodeTree;
}
generatorTree([
{id: 1, value: 2, name: 'A', pid: 0},
{id: 2, value: 2, name:'B', pid: 1},
{id: 3, value: 2, name:'C', pid: 2},
{id: 4, value: 2, name:'D', pid: 3},
{id: 5, value: 2, name:'E', pid: 0}
]);