通过 merge() 函数合并 Map

merge()函数原型：

default V merge(K key, V value,
            BiFunction<? super V, ? super V, ? extends V> remappingFunction) {
        Objects.requireNonNull(remappingFunction);
        Objects.requireNonNull(value);
        V oldValue = get(key);
        V newValue = (oldValue == null) ? value :
                   remappingFunction.apply(oldValue, value);
        if (newValue == null) {
            remove(key);
        } else {
            put(key, newValue);
        }
        return newValue;
    }

该函数是在 JDK8 中为 java.util.Map 新增的功能函数，其功能如果 Map 中不存在给定的 key 或其值为 null，则将 key 设置为 value；若 key 存在值，则设置 remappingFunction 的计算结果为 key 的 value。

使用以下代码初始化两个 Map：

    private static Map<String, Integer> map1 = new HashMap<>();
    private static Map<String, Integer> map2 = new HashMap<>();
    static {
        map1.put("1", 10);
        map1.put("2", 10);
        map1.put("3", 10);
        map1.put("4", 10);
        map1.put("5", 10);

        map2.put("1", 100);
        map2.put("5", 100);
        map2.put("6", 100);
        map2.put("7", 100);
    }

先尝试单独执行以下代码：

        map2.merge("1", 200, (v1, v2)->v1+v2);
        map2.merge("2", 10, (v1, v2)->v1+v2);
        System.out.println("map2=" + map2);

输出信息是：

map2={1=300, 2=10, 5=100, 6=100, 7=100}

可以看到 key 为 "1” 的 value 已被替换为了 v1+v2​ 的结果，即 300；key 为 “2” 时，因为 map2 不存在其值，所以 10 直接被授予了 key=“2”。

合并 map1 和 map2 则可以这样来操作：

        map2.forEach((k, v) -> {
            map1.merge(k, v, (v1, v2) -> {
                System.out.println(v1 + ":" + v2);
                return v2 + v1;
            });
        });
        System.out.println("map1=" + map1);
        System.out.println("map2=" + map2);

遍历 map2 的值，将 map2 合并到 map1 中。以上代码执行会输出如下：

10:100
10:100
map1={1=110, 2=10, 3=10, 4=10, 5=110, 6=100, 7=100}
map2={1=100, 5=100, 6=100, 7=100}

map1 是合并了 map2 之后的结果。