diff --git a/yudao-ui-app/utils/tree.js b/yudao-ui-app/utils/tree.js new file mode 100644 index 000000000..2be7336ce --- /dev/null +++ b/yudao-ui-app/utils/tree.js @@ -0,0 +1,59 @@ +/** + * 构造树型结构数据 + * @param {*} data 数据源 + * @param {*} id id字段 默认 'id' + * @param {*} parentId 父节点字段 默认 'parentId' + * @param {*} children 孩子节点字段 默认 'children' + * @param {*} rootId 根Id 默认 0 + */ +export function handleTree(data, id, parentId, children, rootId) { + id = id || 'id' + parentId = parentId || 'parentId' + children = children || 'children' + rootId = rootId || Math.min.apply(Math, data.map(item => { + return item[parentId] + })) || 0 + //对源数据深度克隆 + const cloneData = JSON.parse(JSON.stringify(data)) + //循环所有项 + const treeData = cloneData.filter(father => { + let branchArr = cloneData.filter(child => { + //返回每一项的子级数组 + return father[id] === child[parentId] + }); + branchArr.length > 0 ? father.children = branchArr : ''; + //返回第一层 + return father[parentId] === rootId; + }); + return treeData !== '' ? treeData : data; +} + +/** + * 树形结构进行删除深度不够的分支 + * 目前只删除了不够三层的分支 + * 对于高于三层的部分目前不做处理 + * todo 暴力遍历,可用递归修改 + * @param {*} data 树形结构 + */ +export function convertTree(data) { + //对源数据深度克隆 + const cloneData = JSON.parse(JSON.stringify(data)) + // 遍历克隆数据,对源数据进行删除操作 + for (let first = 0; first < cloneData.length; first++) { + for (let second = 0; second < cloneData[first].children.length; second++) { + for (let three = 0; three < cloneData[first].children[second].children.length; three++) { + if (data[first].children[second].children[three].children == undefined || + data[first].children[second].children[three].children === 0) { + data[first].children[second].children.splice(second, 1); + } + } + if (data[first].children[second].children == undefined || data[first].children[second].children === 0) { + data[first].children.splice(second, 1); + } + } + if (data[first].children == undefined || data[first].children.length === 0) { + data.splice(first, 1); + } + } + return data; +}