扁平数组与JSON树结构互转

在工作中我们往往可能会遇到无限级别的分类等等的需求，往往后端返回的数据结构可能不是我们想要的数据结构，所以我们来看怎么进行处理

扁平数据结构转换为JSON树型结构

[
  {id: 1, title: "解忧杂货铺1", parent_id: 0},
  {id: 2, title: '解忧杂货铺2', parent_id: 0},
  {id: 3, title: '解忧杂货铺2-1', parent_id: 2},
  {id: 4, title: '解忧杂货铺3-1', parent_id: 3},
  {id: 5, title: '解忧杂货铺4-1', parent_id: 4},
  {id: 6, title: '解忧杂货铺2-2', parent_id: 2},
]
复制代码

代码

function fn(data, pid) {
  let result = [], temp;
  for (let i = 0; i < data.length; i++) {
    if (data[i].pid === pid) {
      let obj = {"title": data[i].title,"pid": data[i].pid,"id": data[i].id};
      temp = fn(data, data[i].id);
      if (temp.length > 0) {
        obj.children = temp;
      }
      result.push(obj);
    }
  }
  return result;
}
let returnTree = fn(flatArr,0);
console.log(returnTree);
复制代码

输出结果

[
{id: 1, title: '解忧杂货铺1', pid: 0},
  {
    id: 2, title: '解忧杂货铺2', pid: 0, children: [
      {id: 6, title: '解忧杂货铺4-2', pid: 2},
      {
        id: 3, title: '解忧杂货铺2-1', pid: 2, children: [
          {
            id: 4, title: '解忧杂货铺3-1', pid: 3, children: [
              {id: 5, title: '解忧杂货铺4-1', pid: 4},
            ]
          },
        ]
      },
    ]
  }
];
复制代码

JSON树型结构转换扁平结构

我们看到上面已经有转换好了的JSON树形结构，那么我们如何把上面的结构反推回来呢

代码

function flatten(data) {
  return data.reduce((arr, {id, title, pid, children = []}) =>
    arr.concat([{id, title, pid}], flatten(children)), []);
}
let flatArr = flatten(JsonTree);
console.log(flatArr)
复制代码

输出结果

[
  {id: 1, title: '解忧杂货铺1', pid: 0},
  {id: 2, title: '解忧杂货铺2', pid: 0},
  {id: 3, title: '解忧杂货铺2-1', pid: 2},
  {id: 4, title: '解忧杂货铺3-1', pid: 3},
  {id: 5, title: '解忧杂货铺4-1', pid: 4},
  {id: 6, title: '解忧杂货铺4-2', pid: 2},
]
复制代码

结语

利用上面的代码，我们就可以轻松的转换这两种数据结构~ 如果大佬们有更好的方法，希望可以发出来一起探讨~