Java 查找树的某子支 并组装Json

public class TreeNodeUtils {

    private static JsonNode result = JsonUtils.objectMapper().nullNode();

    public static JsonNode fetch(JsonNode node, Predicate<JsonNode> predicate) {

        if (!node.isObject()) {
            node = mock(node);
            JsonNode test = test(node, predicate);

            return test.get("children");
        }

        return test(node, predicate);
    }

    private static JsonNode mock(JsonNode node) {

        String mockStr = "{\"id\": \"idTest\", \"name\" : \"nameTest\"}";

        JsonNode mockNode = JsonUtils.toJson(mockStr);
        JsonUtils.writeJsonNode(mockNode, "children", node);

        return mockNode;
    }

    private static JsonNode test(JsonNode node, Predicate<JsonNode> predicate) {
        AtomicReference<JsonNode> rsJson = new AtomicReference<>(JsonUtils.objectMapper().nullNode());
        if (predicate.test(node)) {
            return appendNode(result, node);
        }

        JsonNode children = node.get("children");
        if (children.isArray()) {

            children.iterator().forEachRemaining(child -> {
                JsonNode found = test(child, predicate);
                if (found.isObject()) {
                    rsJson.set(appendNode(result, node));
                    JsonUtils.writeJsonNode(rsJson.get(), "children", JsonUtils.arrayNode().add(found));
                    result = rsJson.get();
                }
            });
        }
        return rsJson.get();
    }

    private static JsonNode appendNode(JsonNode result, JsonNode node) {

        if (result.isNull()) {

            return node.deepCopy();
        }

        JsonNode copyNode = JsonUtils.copyOf(node);

        Iterator<JsonNode> iterator = copyNode.iterator();
        while (iterator.hasNext()) {

            if (iterator.next().isArray()) {
                iterator.remove();
            }
        }

        return copyNode;
    }
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值