java 按条件过滤树工具类,不保留子节点,或者保留子节点

1 篇文章 0 订阅

按条件过滤树工具类,不保留子节点。
要保留子节点的话把注释的(// 去除子节点)这段去掉即可。


    /**
     * 按条件过滤树,不保留子节点
     *
     * @param tree 树型集合
     * @param key 搜索的字段名称
     * @param value 搜索的值
     * @param childNodeName 子节点名称
     * @param <T> 数据具体对象
     * @return tree
     */
    public static <T> List<T> filterTree(List<T> tree, String key, String value, String childNodeName) {
        if (CollectionUtils.isEmpty(tree)) {
            return new ArrayList<>();
        }
        JSONArray arr = JSONArray.parseArray(JSON.toJSONString(tree));
        JSONArray result = filterTree(arr, key, value, childNodeName);
        Type listType = new TypeReference<List<T>>() {}.getType();
        return JSON.parseObject(result.toJSONString(), listType);
    }

    private static JSONArray filterTree(JSONArray tree, String key, String value, String childNodeName) {
        Iterator<Object> it = tree.iterator();
        while (it.hasNext()) {
            JSONObject current = (JSONObject) it.next();
            String currentValue = current.getString(key);
            if(StringUtils.containsIgnoreCase(currentValue, value)) {
                // 去除子节点 start
                JSONArray childNodes = current.getJSONArray(childNodeName);
                if(!CollectionUtils.isEmpty(childNodes)) {
                    JSONArray filterTree = filterTree(childNodes, key, value, childNodeName);
                    if(CollectionUtils.isEmpty(filterTree)) {
                        current.put(childNodeName, new JSONArray());
                    }
                }
                // 去除子节点 end
                continue;
            }
            JSONArray childNodes = current.getJSONArray(childNodeName);
            if(!CollectionUtils.isEmpty(childNodes)) {
                filterTree(childNodes, key, value, childNodeName);
            }
            if(CollectionUtils.isEmpty(childNodes)) {
                it.remove();
            }
        }
        return tree;
    }

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值