构造树型结构数据并过滤层级

遇到需求:
只展示有三层的数据,需要过滤掉不满足三层的。
在这里插入图片描述
实现代码:
使用的若伊框架,在封装好的 handleTree 方法里,加了个 level 的参数

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

  var childrenListMap = {};
  var nodeIds = {};
  var tree = [];
  // 构建初步的childrenListMap
  for (let d of data) {
    let parentId = d[config.parentId];
    if (childrenListMap[parentId] == null) {
      childrenListMap[parentId] = [];
    }
    nodeIds[d[config.id]] = d;
    childrenListMap[parentId].push(d);
  }
  // 找到根节点
  for (let d of data) {
    let parentId = d[config.parentId];
    if (nodeIds[parentId] == null) {
      tree.push(d);
    }
  }
  // 递归适应childrenList,同时计算层级
  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);
      }
    }
  }
  // 如果传入level,则表示只返回达到level层级的tree
  function hasMinimumDepth(node, depth, level) {
    if (level) {
      if (!node[config.childrenList] || !node[config.childrenList].length) {
        return false;
      } else {
        depth++
        if (depth === level) {
          return true;
        } else {
          return node[config.childrenList].every(child => hasMinimumDepth(child, depth, level));
        }
      }
    } else { // 考虑到有其他地方使用该方法,所以不改动原先树形结构数据,直接返回true
      return true;
    }
  }

  return tree.filter(node => hasMinimumDepth(node, 1, level));
}
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值