handleTree树状结构化数据

1,933 阅读1分钟

适用场景: 将后端传过来的数据转化为树状结构。 例如:将后端传过来的部门转化为树状结构的部门。

转化前部门数据: 转换前部门数据.png

转化后部门数据:

转化后的部门数据.png

从这两组数据中我们可以看出,转化之前数组是扁平的数据,所有的部门都在同一个一维数组中。转化之后,所有的子部门都在父部门的children属性里。

/**
 * 构造树型结构数据
 * @param {*} data 数据源
 * @param {*} id id字段 默认 'id'
 * @param {*} parentId 父节点字段 默认 'parentId'
 * @param {*} children 孩子节点字段 默认 'children'
 */
export function handleTree(data, id, parentId, children) {
  let config = {
    id: id || 'id',
    parentId: parentId || 'parentId',
    childrenList: children || 'children'
  };

  var childrenListMap = {};
  var nodeIds = {};
  var tree = [];


  //将同一父元素的所有元素放在同一个数组中
  for (let d of data) {
    console.log('d',d[config.parentId]);
    let parentId = d[config.parentId];
    if (childrenListMap[parentId] == null) {
      childrenListMap[parentId] = [];
    }
    nodeIds[d[config.id]] = d;
    childrenListMap[parentId].push(d);
  }


  //将顶部元素放入tree数组中
  for (let d of data) {
    let parentId = d[config.parentId];
    if (nodeIds[parentId] == null) {
      tree.push(d);
    }
  }

  //开始通过父元素寻找子元素并放入children属性中
  for (let t of tree) {
    adaptToChildrenList(t);
  }

  function adaptToChildrenList(o) {
    if (childrenListMap[o[config.id]] !== null) {
      o[config.childrenList] = childrenListMap[o[config.id]];
    }
    if (o[config.childrenList]) {
      for (let c of o[config.childrenList]) {
        adaptToChildrenList(c);
      }
    }
  }
  return tree;
}