js扁平结构转树状结构

/**
 * @description 扁平数据转树状
 * @param {Array} arr 扁平数据
 * @param {String} id id字段名
 * @param {String} pid 父id字段名
 * @returns {Array} tree 树状结构
*/
function arrayToTree(arr, id, pid) {
  let tree = []
  const lv1 = getLv1(arr) // 1. 先将一级菜单获取到，放到一个数组里边
  tree = addSon(lv1)
  function addSon (lv1) {
    arr.forEach((item) => {
      if(!item.children) item.children = [] // 给所有菜单加上children节点
      // console.log('执行了');
      if (item[pid]) {
        lv1.forEach((lv1Item) => {
          if (item[pid] === lv1Item[id]) {
            if(!lv1Item.children) lv1Item.children = []
            lv1Item.children.push(item)
            const newarr = []
            newarr.push(item)
            return addSon(newarr)
          }
        })
      }
    })
    return lv1
  }
  function getLv1 (arr) {
    const lv1 = []
    arr.forEach((item) => {
      if (item[pid] === 0) {
        lv1.push(item)
      }
    })
    return lv1
  }
  return tree
}

{/tabs-pane}
{tabs-pane label="id映射引用"}

/**
 * @description 扁平数据转树状
 * @param {Array} arr 扁平数据
 * @param {String} id id字段名
 * @param {String} pid 父id字段名
 * @returns {Array} tree 树状结构
*/
function arrayToTree(arr = [], id = 'id', pid = 'pid') {
    const data = JSON.parse(JSON.stringify(arr))
    const idMap = new Map()
    const tree = []
    // 转换为id映射
    data.forEach(item => idMap.set(item[id], item))
    // 遍历数据，找到根节点， 并追加children节点
    data.forEach(item => {
        if (!item[pid]) return tree.push(item)
        const parent = idMap.get(item[pid])
        if (!parent.children) parent.children = []
        parent.children.push(item)
    })
    return tree
}

{/tabs-pane}