树形节点的过滤处理

// nodes就是树形的最原始数据,query就是关键字,最后会返回满足条件的节点数组
function filter(nodes, query) {
    // 条件就是节点的title过滤关键字
    let predicate = function (node) {
        if (node.title.indexOf(query) > -1) {
            return true;
        } else {
            return false;
        }
    };
    if (!(nodes && nodes.length)) {
        return [];
    }
    let newChildren = [];
    for (let node of nodes) {
        // 以下两个条件任何一个成立,当前节点都应该加入到新子节点集中
        // 1. 子孙节点中存在符合条件的,即 subs 数组中有值
        // 2. 自己本身符合条件
        // let subs = this.filter(node.children, query);
        // if (predicate(node)) {
        //   newChildren.push(node);
        // } else if (subs && subs.length) {
        //   node.children = subs;
        //   newChildren.push(node);
        // }

        // 以下只需要考虑自身的节点满足条件即可,不用带上父节点
        if (predicate(node)) {
            newChildren.push(node);
            node.children = this.filter(node.children, query);
        } else {
            newChildren.push(...this.filter(node.children, query));
        }
    }
    return newChildren.length ? newChildren : [];
}

初始状态的树形数据结构

 

(初始状态的树形结构)

 

 

过滤后不仅展示满足条件的节点,还展示其父节点

 

 (过滤后不仅展示满足条件的节点,还展示其父节点)

过滤后只展示满足条件的节点

 

 (过滤后只展示满足条件的节点)

 

posted @ 2020-06-30 17:52  蛙仔  阅读(4717)  评论(0编辑  收藏  举报