如何使用Java 8 Stream/Lambda遍历指定键列表的JSONObject

给定键列表：

List<String> traverses = Arrays.asList("json1", "json2", "json3", "json4", "json5");

以及如下嵌套结构的JSONObject：

{
  "json1": {
    "json2": {
      "json3": {
        "json4": {
          "json5": {
            "data": [
              {
                "key": "9941",
                "doc_count": 14
              },
              {
                "key": "9920",
                "doc_count": 11
              },
              {
                "key": "9933",
                "doc_count": 11
              },
              {
                "key": "9931",
                "doc_count": 6
              },
              {
                "key": "9932",
                "doc_count": 4
              }
            ]
          }
        }
      }
    }
  }
}

当前已通过传统for循环实现逐层获取子JSONObject：

JSONObject jsonObject = getJSONObject();
for (String key : traverses) {
    jsonObject = jsonObject.optJSONObject(key);
}

Stream/Lambda实现方案 #

可以利用Stream.reduce()方法实现累积式逐层获取，完美匹配"从初始对象开始，依次用键获取子对象"的逻辑：

JSONObject jsonObject = getJSONObject();
JSONObject result = traverses.stream()
    .reduce(jsonObject, (currentJson, key) -> 
        currentJson != null ? currentJson.optJSONObject(key) : null,
        (json1, json2) -> json1); // 并行流场景下的合并逻辑，串行流不会触发

细节说明：

• 第一个参数jsonObject是初始值，即最外层的JSON对象
• 第二个参数为累积器：如果当前JSON对象不为null，则继续获取下一层子对象；若已为null则直接返回null，避免空指针
• 第三个参数是并行流专用的合并函数，这里因为是串行遍历键列表，直接返回第一个参数即可

如果想在中途遇到null时提前终止遍历（类似for循环里的break），可使用Java 9+支持的takeWhile优化：

JSONObject result = traverses.stream()
    .takeWhile(key -> jsonObject != null)
    .reduce(jsonObject, (currentJson, key) -> currentJson.optJSONObject(key), (a,b) -> a);

内容的提问来源于stack exchange，提问作者sushil