扁平化数据转化成树形数据

const data = [
    {
        "area_id": 5,
        "name": "广东省",
        "parent_id": 0,
    },  
    {
        "area_id": 6,
        "name": "广州市",
        "parent_id": 5,
    },
    {
        "area_id": 7,
        "name": "深圳市",
        "parent_id": 5,
    },
    {
        "area_id": 4,
        "name": "北京市",
        "parent_id": 3,
    },
    {
        "area_id": 3,
        "name": "北京",
        "parent_id": 0,
    },
    {
        "area_id": 2,
        "name": "测试子地区",
        "parent_id": 1,
    },
    {
        "area_id": 1,
        "name": "测试地区",
        "parent_id": 0,
    }
]

 

这是一个扁平化的数据

但是这种数据返回给前端展示成树形的话，会比较不方便，因此需要进行一个转换

转换的方式有两种，

方法一，递归调用

function toTreeData(data,pid){
 
    function tree(id) {
        let arr = []
        data.filter(item => {
            return item.parent_id === id;
        }).forEach(item => {
            arr.push({
                area_id: item.area_id,
                label: item.name,
                children: tree(item.area_id)
            })
        })
        return arr
    }
    return tree(pid)  // 第一级节点的父id，是null或者0，视情况传入
}

方法二，

function setTreeData(arr) {
    //  删除所有 children,以防止多次调用
    arr.forEach(function (item) {
            delete item.children;
    });
    let map = {}; // 构建map
    arr.forEach(i => {
        map[i.area_id] = i; // 构建以area_id为键 当前数据为值
    });

    let treeData = [];
    arr.forEach(child => {
        const mapItem = map[child.parent_id]; // 判断当前数据的parent_id是否存在map中

        if (mapItem) { // 存在则表示当前数据不是最顶层数据
        
            // 注意: 这里的map中的数据是引用了arr的它的指向还是arr，当mapItem改变时arr也会改变,踩坑点
            (mapItem.children || ( mapItem.children = [] )).push(child); // 这里判断mapItem中是否存在children, 存在则插入当前数据, 不存在则赋值children为[]然后再插入当前数据
        } else { // 不存在则是组顶层数据
            treeData.push(child);
        }
    });

    return treeData;
};

console.log(setTreeData(data)); // 输出整理后的数据

 
 输出结果：

[
  {
    "area_id": 5,
    "name": "广东省",
    "parent_id": 0,
    "children": [
      {
        "area_id": 6,
        "name": "广州市",
        "parent_id": 5,
        "children": [
          {
            "area_id": 99,
            "name": "sss",
            "parent_id": 6
          }
        ]
      },
      {
        "area_id": 7,
        "name": "深圳市",
        "parent_id": 5
      }
    ]
  },
  {
    "area_id": 3,
    "name": "北京",
    "parent_id": 0,
    "children": [
      {
        "area_id": 4,
        "name": "北京市",
        "parent_id": 3
      }
    ]
  },
  {
    "area_id": 1,
    "name": "测试地区",
    "parent_id": 0,
    "children": [
      {
        "area_id": 2,
        "name": "测试子地区",
        "parent_id": 1
      }
    ]
  }
]