1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
package org.springblade.core.mongo.utils;
 
import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.node.ObjectNode;
 
import java.util.*;
 
/**
 * 处理 mongo json 数据结构
 *
 * @author L.cm
 */
public class MongoJsonUtils {
 
    /**
     * 获取所有的叶子节点和路径信息
     *
     * @param jsonNode jsonTree
     * @return tree叶子信息
     */
    public static List<JsonNodeInfo> getLeafNodes(JsonNode jsonNode) {
        if (jsonNode == null || !jsonNode.isObject()) {
            return Collections.emptyList();
        }
        List<JsonNodeInfo> list = new ArrayList<>();
        // 双向的队列 Deque 代替 Stack，Stack 性能不好
        LinkedList<String> deque = new LinkedList<>();
        // 递归获取叶子 🍃🍃🍃 节点
        getLeafNodes(jsonNode, null, deque, list);
        return list;
    }
 
    private static void getLeafNodes(JsonNode jsonNode, JsonNode parentNode, LinkedList<String> deque, List<JsonNodeInfo> list) {
        Iterator<Map.Entry<String, JsonNode>> iterator;
        if (parentNode == null) {
            iterator = jsonNode.fields();
        } else {
            iterator = parentNode.fields();
        }
        // tree 子节点
        while (iterator.hasNext()) {
            Map.Entry<String, JsonNode> entry = iterator.next();
            String fieldName = entry.getKey();
            JsonNode nextNode = entry.getValue();
            // 如果不是值节点
            if (nextNode.isObject()) {
                // 添加到队列尾，先进先出
                deque.addLast(fieldName);
                getLeafNodes(parentNode, nextNode, deque, list);
            }
            // 如果是值节点，也就是到叶子节点了，取叶子节点上级即可
            if (nextNode.isValueNode()) {
                // 封装节点列表
                LinkedList<String> elements = new LinkedList<>(deque);
                // tree 的 叶子节点，此处为引用
                list.add(new JsonNodeInfo(elements, parentNode));
                break;
            }
            // 栈非空时弹出
            if (!deque.isEmpty()) {
                deque.removeLast();
            }
        }
    }
 
    /**
     * 构建树形节点
     *
     * @param jsonNode 父级节点
     * @param elements tree节点列表
     * @return JsonNode 叶子节点，返回用于塞数据
     */
    public static ObjectNode buildNode(ObjectNode jsonNode, List<String> elements) {
        ObjectNode newNode = jsonNode;
        for (String element : elements) {
            // 如果已经存在节点，这不生成新的
            if (newNode.has(element)) {
                newNode = (ObjectNode) newNode.get(element);
            } else {
                newNode = newNode.putObject(element);
            }
        }
        return newNode;
    }
 
    /**
     * 获取所有 🍃🍃🍃 节点的值，并构建成 mongodb update 语句
     * @param prefix 前缀
     * @param nodeKeys mongo keys
     * @param objectNode tree 🍃 节点
     * @return tree 节点信息
     */
    public static Map<String, Object> getAllUpdate(String prefix, String nodeKeys, ObjectNode objectNode) {
        Map<String, Object> values = new HashMap<>(8);
        Iterator<String> iterator = objectNode.fieldNames();
        while (iterator.hasNext()) {
            String fieldName = iterator.next();
            JsonNode valueNode = objectNode.get(fieldName);
            if (valueNode.isValueNode()) {
                Object value;
                if (valueNode.isShort()) {
                    value = valueNode.shortValue();
                } else if (valueNode.isInt()) {
                    value = valueNode.intValue();
                } else if (valueNode.isLong()) {
                    value = valueNode.longValue();
                } else if (valueNode.isBoolean()) {
                    value = valueNode.booleanValue();
                } else if (valueNode.isFloat()) {
                    value = valueNode.floatValue();
                } else if (valueNode.isDouble()) {
                    value = valueNode.doubleValue();
                } else if (valueNode.isMissingNode()) {
                    value = null;
                } else {
                    value = valueNode.textValue();
                }
                if (value != null) {
                    String valueKey = prefix + '.' + nodeKeys + '.' + fieldName;
                    values.put(valueKey, value);
                }
            }
        }
        return values;
    }
}