Java中高效合并JSONArray：基于共享ID整合复杂键值对

[{"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"}]

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

public class JsonArrayMerger {

    public static void main(String[] args) {
        // 第一个JSONArray：用户基本信息
        String jsonString1 = "[{\"name\": \"John\", \"id\": \"1\"}, {\"name\": \"Adam\", \"id\": \"2\"}]";
        JSONArray jsonArray1 = new JSONArray(jsonString1);

        // 第二个JSONArray：用户详细属性
        String jsonString2 = "[{\"color\": \"red\", \"id\": \"1\", \"country\": \"Poland\"}, {\"color\": \"green\", \"id\": \"2\", \"country\": \"Germany\"}, {\"color\": \"red\", \"id\": \"3\", \"country\": \"England\"}]";
        JSONArray jsonArray2 = new JSONArray(jsonString2);

        // 将所有待合并的JSONArray放入一个列表中
        List<JSONArray> arraysToMerge = new ArrayList<>();
        arraysToMerge.add(jsonArray1);
        arraysToMerge.add(jsonArray2);

        // 调用合并方法
        JSONArray mergedResult = mergeJsonArraysById(arraysToMerge, "id");
        System.out.println("合并后的JSONArray:\n" + mergedResult.toString(2));
    }

    /**
     * 根据共享ID合并多个JSONArray中的JSONObject。
     *
     * @param arraysToMerge 包含待合并JSONArray的列表。
     * @param idKey 用于匹配和作为HashMap键的ID字段名称。
     * @return 合并后的JSONArray，其中每个JSONObject都包含了所有匹配ID的键值对，且不包含ID字段。
     */
    public static JSONArray mergeJsonArraysById(List<JSONArray> arraysToMerge, String idKey) {
        // 使用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); // 使用optJSONObject避免空指针异常

                if (currentObj != null) {
                    String id = currentObj.optString(idKey); // 获取ID字段的值

                    // 只有当ID存在且非空时才进行处理
                    if (id != null && !id.isEmpty()) {
                        // 使用computeIfAbsent方法：
                        // 如果map中不存在该ID对应的JSONObject，则创建一个新的JSONObject作为基础对象。
                        // 注意：在创建新对象时，我们将从currentObj中移除ID字段，
                        // 因为最终结果中我们不希望包含ID。
                        JSONObject existingObj = mergedObjectsMap.computeIfAbsent(id, k -> {
                            // 创建一个新的JSONObject作为该ID的合并基础，并复制当前对象（不包含ID）
                            JSONObject baseObj = new JSONObject();
                            for (String key : currentObj.keySet()) {
                                if (!key.equals(idKey)) {
                                    baseObj.put(key, currentObj.get(key));
                                }
                            }
                            return baseObj;
                        });

                        // 将当前JSONObject中的所有键值对（除了ID字段）添加到existingObj中
                        // 如果存在同名键，新值将覆盖旧值。
                        for (String key : currentObj.keySet()) {
                            if (!key.equals(idKey)) { // 排除ID字段
                                existingObj.put(key, currentObj.get(key));
                            }
                        }
                    }
                }
            }
        }

        // 将HashMap中的所有合并后的JSONObject值收集到一个新的JSONArray中
        return new JSONArray(mergedObjectsMap.values());
    }
}