把带有父子关系的数组转换成tree

1,115 阅读2分钟

先来看一下数据格式

content: [
    { 'name': '拉祜族', 'code': '24', 'parentCode': '2'},
    {'name': '哈萨克族', 'code': '17', 'parentCode': '1'},
    {'name': '水族', 'code': '25', 'parentCode': '2'},
    {'name': '汉族', 'code': '1' },
    {'name': '郾城', 'code': '5', 'parentCode': '4' },
    {'name': '朝鲜族', 'code': '10', 'parentCode': '1'},
    {'name': '河南', 'code': '3', 'parentCode': '25' },
    {'name': '漯河', 'code': '4', 'parentCode': '3'},
    { 'name': '蒙古族', 'code': '2'}
  ]

妈呀累死了,为了让看到的人能有更好的理解,格式化这样的数据也是费老鼻子劲了,首先 code = 1 和 code = 2 是 root 和 个有两个children code = 5 的 parent 是 code = 4 , code = 4 的 parent code = 3 , code = 3 的 parent code = 25,关系就是这样的了。

封装公共方法需要的考虑几个点

  • 需要 list 数据 content
  • 通过 parentCode 来找关系
  • code 是来做唯一的值如果用id呢
  • 找到叶节点用什么作为key呢

所以在调用方法的时候需要传递content 和 配置项,没明白下边结合代码会更容易理解

    // 第一个参数数据, 第二个参数就是上边总结的 3个点 怎么通过配置形式
    buildTree(content, {
      code: 'code',
      parentCode: 'parentCode',
      children: 'children'
    })

想到这个地方很多人会想到用递归的形式,但是递归会很小号内存的,比如有有1000条数据,如果是递归的话 有1000条数据,需要循环1000 次 那就是 执行 1000 * 1000 , 在设计场景中 1000 条数据并不算多, 就会造成浏览器卡顿很长时间甚至,卡死的,直接上代码。

buildTree (data, config) {
    // 设置 三个变量就是为了 动态的查找 data 的数据
    let code = config.code
    let parentCode = config.parentCode
    let children = config.children
    // idMap 利用 code的值 作为key, 因为 parentCode 的值指向的 key
    let idMap = {}
    // 最终要返回的数据
    let jsonTree = []
    // 如果有1000 条数据,最多才循环了 2000 次,浏览器执行 2000 次的速度是非常快的
    data.forEach(item => {
        // 这个地方就是为了做一张地图, 能通过 parentCode 不通过循环快速查找
        idMap[item[code]] = item
    })
    data.forEach(item => {
        // item 的 parentCode 其实 就是 parent 的 code  所以 在 idMap 中能快速定位
        let parent = idMap[item[parentCode]]
        if (parent) {
            // 如果 parent[children] 不存在就 等于 空数组
            !parent[children] && (parent[children] = [])
            parent[children].push(item)
        } else {
            // 走到这里说明没有 parent
            jsonTree.push(item)
        }
    })
}

看到这里可能还有同学会疑问,上边明明就查找了一级为啥会形成一棵树呢, 这里就运用到了 javascript 的 引用类型的 特性, 第二个循环list 中每个值都会循环一遍的,所以 code = 5 的 对象 是 push 到了code = 4 对象中, 当 循环到 code = 4 的 时候 code = 4 的 对象中已经多了一个 children 属性,最重要的理解就在这里