Tree数据设置层级关系

场景： 在我们日常需求中，肯定各种各样的狗产品都有，要求一颗 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 层， 如果有特殊需求，把上面的判断简单的调整一下即可；