数组转为tree，parengId转为children格式

PairsNightRain

已于 2024-08-26 16:49:15 修改

阅读量216

点赞数 2

文章标签：前端 typescript

于 2024-08-26 16:45:59 首次发布

本文链接：https://blog.csdn.net/qq_37161741/article/details/141568027

版权

export const generateTreeFromArray = (groups: ITreeNode[]): DataNode[] => {
  if (!groups.length) return [];
  const nodes: DataNode[] = [];
  const map = new Map();

  groups.forEach((group) => {
    map.set(group.id, {
      key: `${group.id}`,
      title: group.name,
      data: group,
      icon: group.icon,
      disabled: group.disabled,
    } as DataNode);
  });
  groups.forEach((group) => {
    const parent = map.get(group.parent_id);
    if (parent) {
      (parent.children || (parent.children = [])).push(map.get(group.id));
    } else {
      nodes.push(map.get(group.id));
    }
  });

  return nodes;
};