树形数据处理

为树形结构增加层级(level)、路径(path)、全路径(allPath)及自定义属性

function addAttrToTree(treeData, path = "", allPath = "", level = 1) {
  treeData.forEach((node, idx) => {
    let nPath = path ? path + "-" + (idx + 1) : idx + 1 + "",
      allNPath = allPath ? allPath + "," + nPath : idx + 1 + "";
    node.path = nPath; // 路径
    node.allPath = allNPath; // 全路径
    node.level = level; // 层级
    // 自定义属性也可在此处增加
    if (node.children && node.children.length) {
      addAttrToTree(node.children, nPath, allNPath, level + 1);
    }
  });
  if (level === 1) {
    return treeData;
  }
}

tips：增加了全路径，可以快速的查找指定节点的父级及直系祖级

树形结构平铺为list

function setTreeDataToList(treeData, treeList = []) {
  treeData.forEach((node) => {
    treeList.push({
      ...node,
      children: [], // 如果需要节点携带直系子集，请注释此行代码
    });
    if (node.children && node.children.length) {
      setTreeDataToList(node.children, treeList);
    }
  });
  return treeList;
}

查找指定树节点的直系祖级

/**
 * 1、将指定节点的“全节点”属性分割为数组
 * tips：(“全节点”的添加请参考本文章的 - addAttrToTree - 方法)
 * 2、然后遍历平铺树，判断遍历节点的路径是否存在于全节点数组
 * tips：(平铺树方法请参考本文章的 - setTreeDataToList - 方法)
 */
// 示例代码：
let treeData = addAttrToTree(treeData); // 为树结构增加路径与全路径
let treeList = setTreeDataToList(treeData); // 获取树的平铺树
let pathList = node.allPath.split(","); // 指定节点的全路径数组
// 直系祖级：
let parentList = treeList.filter((node) => {
  return pathList.indexOf(node.path) > -1;
});