85 lines
1.9 KiB
Java
85 lines
1.9 KiB
Java
var setupDefaults = require('../setupDefaults')
|
|
|
|
var map = require('./map')
|
|
var sortBy = require('./sortBy')
|
|
|
|
var clone = require('../base/clone')
|
|
var includes = require('./includes')
|
|
var each = require('../base/each')
|
|
var remove = require('../base/remove')
|
|
|
|
var assign = require('../object/assign')
|
|
|
|
function strictTree (array, optChildren) {
|
|
each(array, function (item) {
|
|
if (item.children && !item.children.length) {
|
|
remove(item, optChildren)
|
|
}
|
|
})
|
|
}
|
|
|
|
/**
|
|
* 将一个带层级的数据列表转成树结构
|
|
*
|
|
* @param {Array} array 数组
|
|
* @param {Object} options {strict: false, parentKey: 'parentId', key: 'id', children: 'children', data: 'data'}
|
|
* @return {Array}
|
|
*/
|
|
function toArrayTree (array, options) {
|
|
var opts = assign({}, setupDefaults.treeOptions, options)
|
|
var optStrict = opts.strict
|
|
var optKey = opts.key
|
|
var optParentKey = opts.parentKey
|
|
var optChildren = opts.children
|
|
var optSortKey = opts.sortKey
|
|
var optReverse = opts.reverse
|
|
var optData = opts.data
|
|
var result = []
|
|
var treeMap = {}
|
|
var idList, id, treeData, parentId
|
|
|
|
if (optSortKey) {
|
|
array = sortBy(clone(array), optSortKey)
|
|
if (optReverse) {
|
|
array = array.reverse()
|
|
}
|
|
}
|
|
|
|
idList = map(array, function (item) {
|
|
return item[optKey]
|
|
})
|
|
|
|
each(array, function (item) {
|
|
id = item[optKey]
|
|
|
|
if (optData) {
|
|
treeData = {}
|
|
treeData[optData] = item
|
|
} else {
|
|
treeData = item
|
|
}
|
|
|
|
parentId = item[optParentKey]
|
|
treeMap[id] = treeMap[id] || []
|
|
treeMap[parentId] = treeMap[parentId] || []
|
|
treeMap[parentId].push(treeData)
|
|
treeData[optKey] = id
|
|
treeData[optParentKey] = parentId
|
|
treeData[optChildren] = treeMap[id]
|
|
|
|
if (!optStrict || (optStrict && !parentId)) {
|
|
if (!includes(idList, parentId)) {
|
|
result.push(treeData)
|
|
}
|
|
}
|
|
})
|
|
|
|
if (optStrict) {
|
|
strictTree(array, optChildren)
|
|
}
|
|
|
|
return result
|
|
}
|
|
|
|
module.exports = toArrayTree
|