110 lines
2.7 KiB
JavaScript
110 lines
2.7 KiB
JavaScript
/**
|
||
* pid形式数据转children形式
|
||
* @param data 需要转换的数组
|
||
* @param idKey id字段名
|
||
* @param pidKey pid字段名
|
||
* @param childKey 生成的children字段名
|
||
* @param pid 顶级的pid
|
||
* @param addPIds 是否添加所有父级id的字段
|
||
* @param parentsKey 所有父级id的字段名称，默认parentIds
|
||
* @param parentIds 所有父级id
|
||
* @returns {[]}
|
||
*/
|
||
export function toTreeData(data, idKey, pidKey, childKey, pid, addPIds, parentsKey, parentIds) {
|
||
if (typeof data === 'object' && !Array.isArray(data)) {
|
||
idKey = data.idKey
|
||
pidKey = data.pidKey
|
||
childKey = data.childKey
|
||
pid = data.pid
|
||
addPIds = data.addPIds
|
||
parentsKey = data.parentsKey
|
||
parentIds = data.parentIds
|
||
data = data.data
|
||
}
|
||
if (!childKey) {
|
||
childKey = 'children'
|
||
}
|
||
if (typeof pid === 'undefined') {
|
||
pid = []
|
||
data.forEach((d) => {
|
||
let flag = true
|
||
for (let i = 0; i < data.length; i++) {
|
||
if (d[pidKey] === data[i][idKey]) {
|
||
flag = false
|
||
break
|
||
}
|
||
}
|
||
if (flag) {
|
||
pid.push(d[pidKey])
|
||
}
|
||
})
|
||
}
|
||
const result = []
|
||
data.forEach((d) => {
|
||
if (d[idKey] === d[pidKey]) {
|
||
console.error('data error: ', d)
|
||
return
|
||
}
|
||
if (Array.isArray(pid) ? (pid.indexOf(d[pidKey]) !== -1) : (d[pidKey] === pid)) {
|
||
const children = toTreeData({
|
||
data: data,
|
||
idKey: idKey,
|
||
pidKey: pidKey,
|
||
childKey: childKey,
|
||
pid: d[idKey],
|
||
addPIds: addPIds,
|
||
parentsKey: parentsKey,
|
||
parentIds: (parentIds || []).concat([d[idKey]])
|
||
})
|
||
if (children.length > 0) {
|
||
d[childKey] = children
|
||
}
|
||
if (addPIds) {
|
||
d[parentsKey || 'parentIds'] = parentIds || []
|
||
}
|
||
result.push(d)
|
||
}
|
||
})
|
||
return result
|
||
}
|
||
|
||
/**
|
||
* 遍历children形式数据
|
||
* @param data 需要遍历的数组
|
||
* @param callback 回调
|
||
* @param childKey children字段名
|
||
*/
|
||
export function eachTreeData(data, callback, childKey = 'children') {
|
||
if (!data || !data.length) {
|
||
return
|
||
}
|
||
data.forEach((d) => {
|
||
if (callback(d) !== false && d[childKey]) {
|
||
eachTreeData(d[childKey], callback, childKey)
|
||
}
|
||
})
|
||
}
|
||
|
||
/**
|
||
* 处理树形数据
|
||
* @param data 需要处理的数据
|
||
* @param formatter 处理器
|
||
* @param childKey children字段名
|
||
* @returns {[]} 处理后的数据
|
||
*/
|
||
export function formatTreeData(data, formatter, childKey = 'children') {
|
||
const result = []
|
||
if (data && data.length) {
|
||
data.forEach((d) => {
|
||
const item = formatter(d)
|
||
if (item !== false) {
|
||
if (item[childKey]) {
|
||
item[childKey] = formatTreeData(item[childKey], formatter, childKey)
|
||
}
|
||
result.push(item)
|
||
}
|
||
})
|
||
}
|
||
return result
|
||
}
|