Java 8流-合并地图集合

时间:2019-03-06 23:22:10

标签: java java-8 java-stream

让我们说我有一个带有以下签名的方法:

Map<String, String> merge (String[][] ... maps)

它需要(可变)数量的表示映射的二维数组,例如:

        String[][] m1 = new String[][] {
            {"k1", "a"},
            {"k2", "b"}
        };

        String[][] m2 = new String[][] {
            {"k1", "a"},
            {"k2", "b"},
            {"k3", "c"}   
        };

        String[][] m3 = new String[][] {
            {"k1", "x"},
            {"k2", "b"}
        };

现在,我可以像这样从左到右合并任意两个地图:

 Map<String, String> m1 = Stream.of(map1).collect(Collectors.toMap(k -> k[0], v -> v[1]));
 Map<String, String> m2 = Stream.of(map2).collect(Collectors.toMap(k -> k[0], v -> Optional.ofNullable(v[1]).orElse("null")));
 m1.forEach((key, value) -> m2.merge(key, value, (v1, v2) -> v1));

但是我该如何合并各种各样的这种数组映射,以便在通过上面定义的m1, m2, m3之后,结果是:

        String[][] m3 = new String[][] {
            {"k1", "x"},
            {"k2", "b"}
            {"k3", "c"}   
        };

2 个答案:

答案 0 :(得分:6)

如果我正确理解了您的问题,这很简单。您只需flatMap所有条目并收集到地图,然后使用合并函数选择后面的值即可:

template <auto* Ptr>

如果要将其转换回数组,可以执行

Map<String, String> merge (String[][] ... maps) {
    return Arrays.stream(maps)
            .flatMap(Arrays::stream)
            .collect(Collectors.toMap(a -> a[0], a -> a[1], (a, b) -> b));
}

答案 1 :(得分:3)

Map<String, String> merge(String[][]... maps) {
    return Stream.of(maps)
            .flatMap(Arrays::stream)          //flatten all maps into a single stream
            .map(s -> Map.entry(s[0], s[1]))  //convert array to Entry
            .collect(Collectors.toMap(Map.Entry::getKey,
                    Map.Entry::getValue,
                    (o1, o2) -> o2));
}

编辑:无需转换为条目。我们可以直接收集地图,如@shmosel所示。