Skip to content

数组转树

js
const data = [
  { id: "01", name: "张大大", pid: "", job: "项目经理" },
  { id: "02", name: "小亮", pid: "01", job: "产品leader" },
  { id: "03", name: "小美", pid: "01", job: "UIleader" },
  { id: "04", name: "老马", pid: "01", job: "技术leader" },
  { id: "05", name: "老王", pid: "01", job: "测试leader" },
  { id: "06", name: "老李", pid: "01", job: "运维leader" },
  { id: "07", name: "小丽", pid: "02", job: "产品经理" },
  { id: "08", name: "大光", pid: "02", job: "产品经理" },
  { id: "09", name: "小高", pid: "03", job: "UI设计师" },
  { id: "10", name: "小刘", pid: "04", job: "前端工程师" },
  { id: "11", name: "小华", pid: "04", job: "后端工程师" },
  { id: "12", name: "小李", pid: "04", job: "后端工程师" },
  { id: "13", name: "小赵", pid: "05", job: "测试工程师" },
  { id: "14", name: "小强", pid: "05", job: "测试工程师" },
  { id: "15", name: "小涛", pid: "06", job: "运维工程师" },
];

const tree = arrToTree(data);
console.log(tree);

function arrToTree(data) {
  let tree = [];
  if (!Array.isArray(data)) {
    return tree;
  } // 将数组转换成对象(键值对),将ID作为属性名,原来的数组里的对象作为属性值
  let map = {};
  data.forEach((item) => {
    map[item.id] = item;
  }); // 通过对象的属性名(ID)来找到父级节点,将存到map里的对象取出来放到父级节点的childere数组中
  data.forEach((item) => {
    let parent = map[item.pid]; // 修改对象的属性

    delete item.pid;
    item["label"] = item.name;
    delete item.name;

    if (parent) {
      (parent.children || (parent.children = [])).push(item);
    } else {
      tree.push(item);
    }
  });
  return tree;
}

Released under the MIT License.