You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
93 lines
2.8 KiB
93 lines
2.8 KiB
|
|
import deepClone from './deepClone'
|
|
var util = {};
|
|
/**
|
|
* 将树形结构转成数组结构
|
|
* @param {Array} treeData 数据源
|
|
* @param {Object} treeProps 树结构配置 { id : "menu_id", children : "children" }
|
|
* vk.pubfn.treeToArray(treeData);
|
|
*/
|
|
util.treeToArray = function(treeData, treeProps) {
|
|
let newTreeData = deepClone(treeData);
|
|
return util.treeToArrayFn(newTreeData, treeProps);
|
|
};
|
|
util.treeToArrayFn = function(treeData, treeProps = {}, arr=[], current_parent_id) {
|
|
let { id="_id", parent_id="parent_id", children = "children", deleteChildren = true } = treeProps;
|
|
for(let i in treeData){
|
|
let item = treeData[i];
|
|
if(current_parent_id) item[parent_id] = current_parent_id;
|
|
arr.push(item);
|
|
if(item[children] && item[children].length>0){
|
|
arr = util.treeToArrayFn(item[children], treeProps, arr, item[id]);
|
|
}
|
|
if(deleteChildren){
|
|
delete item[children];
|
|
}
|
|
}
|
|
return arr;
|
|
};
|
|
/**
|
|
* 数组结构转树形结构
|
|
let tree = vk.pubfn.arrayToTree(arrayData,{
|
|
id:"code",
|
|
parent_id:"parent_code",
|
|
});
|
|
*/
|
|
|
|
util.arrayToTree = function(originalArrayData,treeProps) {
|
|
let arrayData = deepClone(originalArrayData);
|
|
let {
|
|
id="_id",
|
|
parent_id="parent_id",
|
|
children = "children",
|
|
deleteParentId = false,
|
|
need_field
|
|
} = treeProps;
|
|
let result = [];
|
|
let temp = {};
|
|
for (let i = 0; i < arrayData.length; i++) {
|
|
temp[arrayData[i][id]] = arrayData[i]; // 以id作为索引存储元素，可以无需遍历直接定位元素
|
|
}
|
|
for (let j = 0; j < arrayData.length; j++) {
|
|
let currentElement = arrayData[j];
|
|
let newCurrentElement = {};
|
|
if(need_field){
|
|
need_field = uniqueArr(need_field.concat([id,parent_id,children]));
|
|
for(let keyName in currentElement){
|
|
if(need_field.indexOf(keyName) === -1){
|
|
delete currentElement[keyName];
|
|
}
|
|
}
|
|
}
|
|
let tempCurrentElementParent = temp[currentElement[parent_id]]; // 临时变量里面的当前元素的父元素
|
|
if (tempCurrentElementParent) {
|
|
// 如果存在父元素
|
|
if (!tempCurrentElementParent[children]) {
|
|
// 如果父元素没有chindren键
|
|
tempCurrentElementParent[children] = []; // 设上父元素的children键
|
|
}
|
|
if(deleteParentId){
|
|
delete currentElement[parent_id];
|
|
}
|
|
tempCurrentElementParent[children].push(currentElement); // 给父元素加上当前元素作为子元素
|
|
} else {
|
|
// 不存在父元素，意味着当前元素是一级元素
|
|
result.push(currentElement);
|
|
}
|
|
}
|
|
return result;
|
|
};
|
|
|
|
// 最简单数组去重法
|
|
function uniqueArr(array) {
|
|
let n = []; //一个新的临时数组
|
|
//遍历当前数组
|
|
for (let i = 0; i < array.length; i++) {
|
|
//如果当前数组的第i已经保存进了临时数组，那么跳过，
|
|
//否则把当前项push到临时数组里面
|
|
if (n.indexOf(array[i]) == -1) n.push(array[i]);
|
|
}
|
|
return n;
|
|
}
|
|
|
|
export default util;
|