java - 如何将嵌套 JSONObject/JSONArray 转换为 flatJSON 并生成多条记录

Question

我想将嵌套的 JSONObject 转换为 flatJSON，如果 JSONAArray 存在，它必须生成多条记录。因此，我使用递归编写了代码，它将嵌套的 JSON 对象转换为 flatJSON。

public class Test {


    public static Map<String,Object> flatJSON(Object obj,Map<String,Object> flatmap) {
        Iterator<?> keys = ((JSONObject) obj).keys();
        while (keys.hasNext()) {
            String key = (String) keys.next();
            if(((JSONObject) obj).get(key) instanceof JSONObject) {
                flatJSON(((JSONObject) obj).get(key), flatmap);
            }
            else if(((JSONObject) obj).get(key) instanceof JSONArray) {
                if(((JSONObject) obj).getJSONArray(key).optJSONObject(0)!=null) {
                    JSONObject jsonObject = ((JSONObject) obj).getJSONArray(key).getJSONObject(0);
                    flatJSON(jsonObject, flatmap);
                }
                else {
                    flatmap.put(key, ((JSONObject) obj).get(key));
                }
            }
            else {
                flatmap.put(key, ((JSONObject) obj).get(key));
            }
        }
        return flatmap;
    }

    public static void main(String[] args) {
        String s="{\r\n" + 
                "  \"root\": {\r\n" + 
                "    \"data\": {\r\n" + 
                "      \"a\": \"1\",\r\n" + 
                "      \"b\": {\r\n" + 
                "        \"c\": \"2\"\r\n" + 
                "      },\r\n" + 
                "      \"test\": [\r\n" + 
                "        {\r\n" + 
                "          \"e\": \"3\"\r\n" + 
                "        },\r\n" + 
                "        {\r\n" + 
                "          \"f\": \"4\"\r\n" + 
                "        }\r\n" + 
                "      ],\r\n" + 
                "      \"test1\": [\r\n" + 
                "        {\r\n" + 
                "          \"g\": \"5\"\r\n" + 
                "        }\r\n" + 
                "      ],\r\n" + 
                "      \"test2\": [\r\n" + 
                "        1,\r\n" + 
                "        2,\r\n" + 
                "        3\r\n" + 
                "      ]\r\n" + 
                "    }\r\n" + 
                "  }\r\n" + 
                "}";
        System.out.println(s);
        System.out.println(flatJSON(new JSONObject(s), new HashMap<String, Object>()));
    }

但是上面的代码仅适用于 JSONObject，对于 JSONArray 作为输入，我们如何转换为 flatJSON。以下是输入和预期输出的示例。

示例 1：

输入：

{
  "root": {
    "data": {
      "a": "1",
      "b": {
        "c": "2"
      },
      "test": [
        {
          "e": "3",
          "f": "5"
        },
        {
          "e": "4",
          "f": "6"
        }
      ]
    }
  }
}

预期输出：

{
  "a": "1",
  "c": "2",
  "e": "3",
  "f": "5"
}

{
  "a": "1",
  "c": "2",
  "e": "4",
  "f": "6"
}

例 2：

输入：

{
  "root": {
    "data": {
      "a": "1",
      "b": {
        "c": "2"
      },
      "test": [
        {
          "e": [{"m":1},{"m":2}]
        },
        {
          "e": [{"m":3}]
        }
      ]
    }
  }
}

预期输出：

{
  "a": "1",
  "c": "2",
  "m": "1"
}

{
  "a": "1",
  "c": "2",
  "m": "2"
}
{
  "a": "1",
  "c": "2",
  "m": "3"
}

例 3：

输入：

{
  "root": {
    "data": {
      "a": "1",
      "b": {
        "c": "2"
      },
      "test": [1,2,3]
    }
  }
}

预期输出：

{
  "a": "1",
  "c": "2",
  "test": "1"
}

{
  "a": "1",
  "c": "2",
  "test": "2"
}
{
  "a": "1",
  "c": "2",
  "test": "3"
}

如何修改上述代码以获得预期结果。

标签： java