(12/05/21) 封装处理树形结构的工具函数

1. 目标

:把后端传过来的平铺数组数据处理为tree组件需要的树形数据,我们把它封装成工具，单独放在utils下

2. 补充工具函数

在src/utils/index.js文件中，补充一个函数tranListToTreeData

/**
 * 把平铺的数组结构转成树形结构
 *
 * [
 *  {id:"01", pid:"",   "name":"老王" },
 *  {id:"02", pid:"01", "name":"小张" }
 * ]
 * 上面的结构说明： 老王是小张的上级
 */
export function tranListToTreeData(list) {
  // 最终要产出的树状数据的数组
  const treeList = []
  // 所有项都使用对象存储起来
  const map = {}

  // 建立一个映射关系：通过id快速找到对应的元素
  list.forEach(item => {
    if (!item.children) {
      item.children = []
    }
    map[item.id] = item
  })
  // map:
  // {
  //   "312c": { 'id': '312c', 'pid': '',     'name': '财务部',    children: [{ 'id': '312d', 'pid': '312c', 'name': '财务核算部',children: []}] },
  //   "312d": { 'id': '312d', 'pid': '312c', 'name': '财务核算部',children: []}
  // }

  list.forEach(item => {
    // 对于每一个元素来说，先找它的上级
    //    如果能找到，说明它有上级，则要把它添加到上级的children中去
    //    如果找不到，说明它没有上级，直接添加到 treeList
    const parent = map[item.pid]
    // 如果存在上级则表示item不是最顶层的数据
    if (parent) {
      parent.children.push(item)
    } else {
      // 如果不存在上级 则是顶层数据,直接添加
      treeList.push(item)
    }
  })
  // 返回出去
  return treeList
}

3.使用

在vue文件中，引入函数tranListToTreeData

import { tranListToTreeData } from '@/utils'

async loadDepartments() {
      try {
        const res = await getDepartments() // 1.发送请求,获取平铺的数据
        // 业务中pid是-1，它是最高级的节点
        // 2.取出最高级别的节点
        const root = res.data.depts.shift()
        // 3.用工具函数把 平铺的数组转成 树型数据
        root.children = tranListToTreeData(res.data.depts)
        this.list = [root]
      } catch (err) {
        console.log(err)
      }
    }