场景: 在我们日常需求中,肯定各种各样的狗产品都有,要求一颗 tree 树,点击的时候判断当前层级,根据不同层级,展示不同的ui,或者说,一棵tree树,产品最多只需要3层, 而后端却返给你了4层或则更多… 这时候后端人员可能比较懒,数据没有给你处理后返回,哎,苦逼的页面仔,只有自己去想办法咯~,多的不说,直接贴代码;
1.代码:
/**
* 设置tree层级信息
* @param {Object} data
* @param {Object} maxLevel
* @param {Object} currLevel
*/
const setTreeHierarchyIndex = function(data, maxLevel, currLevel) {
if (!(data instanceof Array)) throw new TypeError('The data should be an array!');
let result = [];
for (let k = 0; k < data.length; k++) {
//console.log('当前层级:', currLevel);
//console.log('是否在指定层级范围:', currLevel < maxLevel);
let temp = data[k];
let newNode = {
...temp,
hierarchyIndex: currLevel + 1,
children: null
};
delete newNode.children;
//是否在指定层级范围
if (currLevel >= maxLevel) continue;
if (temp.children && temp.children.length > 0) {
currLevel++
newNode.children = setTreeHierarchyIndex(temp.children, maxLevel, currLevel);
currLevel--
}
result.push(newNode);
}
return result;
}
小提示: 我这个主要是做比如tree数据总共有 5 层, 而我最多只需要 3 层, 如果有特殊需求,把上面的判断简单的调整一下即可;