// 从后端获取的数据结构
interface BsType {
cid: number;
cname: string;
pid: number;
alias: string;
}
// 转换后的数据结构
interface BussinessTYpe {
value: number;
label: string;
pid: number;
alias: string;
children?: BussinessTYpe[];
}
// 将一维数组转换成树形结构
function arr2tree(data: BsType[]) {
const list: BussinessTYpe[] = []
// 定义setItem方法:递归实现树形结构
const setItem = (arr: BussinessTYpe[], item: BussinessTYpe) => {
const arrItem = arr.find(e => e.value === item.pid)
if (arrItem) {
arrItem.children ? arrItem.children.push(item) : Object.assign(arrItem, { children: [item] })
} else {
for (let i = 0; i < arr.length; i++) {
arr[i].children && setItem(<BussinessTYpe[]>arr[i].children, item)
}
}
}
for (let i = 0; i < data.length; i++) {
const item = { value: data[i].cid, label: data[i].cname, pid: data[i].pid, alias: data[i].alias }
item.pid < 1 ? list.push(item) : setItem(list, item)
}
return list
}
const data: BsType[] = []
console.log(arr2tree(data))
网友评论