Java中基于共享键合并JSONArray：构建统一数据视图

[{"name": "John", "id": "1"}, {"name": "Adam", "id": "2"}]

[{"color": "red", "id": "1", "country": "Poland"}, {"color": "green", "id": "2", "country": "Germany"}, {"color": "red", "id": "3", "country": "England"}]

[{"color": "red", "name": "John", "country": "Poland"}, {"color": "green", "name": "Adam", "country": "Germany"}]

<dependency>
    <groupId>org.json</groupId>
    <artifactId>json</artifactId>
    <version>20231013</version> <!-- 使用最新版本 -->
</dependency>

import org.json.JSONArray;
import org.json.JSONObject;
import java.util.HashMap;
import java.util.Map;

public class JsonArrayMerger {

    /**
     * 合并多个JSONArray中的JSONObject，基于共享的'id'字段。
     * 最终的JSONObject将不包含'id'字段。
     *
     * @param arraysToMerge 包含待合并JSONArray的数组。
     * @return 合并后的JSONArray。
     */
    public static JSONArray mergeJsonArraysById(JSONArray... arraysToMerge) {
        // 使用HashMap作为中间存储，键为id，值为合并后的JSONObject
        Map<String, JSONObject> mergedObjectsMap = new HashMap<>();

        // 遍历所有待合并的JSONArray
        for (JSONArray currentArray : arraysToMerge) {
            // 遍历当前JSONArray中的每个JSONObject
            for (int i = 0; i < currentArray.length(); i++) {
                JSONObject currentObj = currentArray.optJSONObject(i);

                if (currentObj != null) {
                    String id = currentObj.optString("id");

                    // 确保id存在且不为空
                    if (id != null && !id.isEmpty()) {
                        // 使用computeIfAbsent确保每个id只有一个合并后的JSONObject
                        // 如果id不存在，则创建一个新的JSONObject并移除其id字段
                        JSONObject existingOrNewObj = mergedObjectsMap.computeIfAbsent(id, k -> {
                            // 为了不修改原始对象，这里进行一次深拷贝
                            JSONObject newObj = new JSONObject(currentObj.toString());
                            newObj.remove("id"); // 移除id字段
                            return newObj;
                        });

                        // 将当前JSONObject中除id外的所有字段合并到existingOrNewObj中
                        // 注意：如果存在同名键，新值将覆盖旧值
                        for (String key : JSONObject.getNames(currentObj)) {
                            if (!key.equals("id")) {
                                existingOrNewObj.put(key, currentObj.get(key));
                            }
                        }
                    }
                }
            }
        }

        // 从Map的值中构建最终的JSONArray
        return new JSONArray(mergedObjectsMap.values());
    }

    public static void main(String[] args) {
        // 示例数据
        JSONArray array1 = new JSONArray("[{\"name\": \"John\", \"id\": \"1\"}, {\"name\": \"Adam\", \"id\": \"2\"}]");
        JSONArray array2 = new JSONArray("[{\"color\": \"red\", \"id\": \"1\", \"country\": \"Poland\"}, {\"color\": \"green\", \"id\": \"2\", \"country\": \"Germany\"}, {\"color\": \"red\", \"id\": \"3\", \"country\": \"England\"}]");

        // 执行合并
        JSONArray mergedResult = mergeJsonArraysById(array1, array2);
        System.out.println("合并后的JSONArray:\n" + mergedResult.toString(2)); // 使用2格缩进美化输出
    }
}