Question

我有一个班级价值观：

public class Values {
    private int count;
    private int values;
}

以及Map<String, Values>类型的多个地图的列表

Map<String, Values> map1 = new HashMap<>();
        map1 .put("aaa", new Values(1, 10));
        map1 .put("bbb", new Values(5, 50));
        map1 .put("ccc", new Values(2, 30));

Map<String, Values> map2= new HashMap<>();
        map2.put("aaa", new Values(2, 20));
        map2.put("bbb", new Values(3, 50));
        map2.put("ccc", new Values(3, 10));

List<Map<String, Values>> list = Arrays.asList(map1, map2);

地图中可以有任意数量的地图和任意数量的条目，但是地图的键始终相同，只有值可以不同。为了清楚起见，我的示例仅包含2个映射，每个映射仅包含3个条目。

我想要获得一个具有相同键和Values对象作为对象中每个“计数”和每个“值”之和的单个映射，如下所示：

{
        aaa= {
            count = 3,
            value = 30
        },
        bbb= {
            count = 8,
            value = 100
        },
        ccc= {
            count = 6,
            value = 40
        }
    }

我正在尝试使用Streams API来实现这一目标，但是我被困住了：

public static Map<String, Values> mergeMaps(List<Map<String, Values>> maps) {
        return maps.stream()
                .flatMap(m -> m.entrySet().stream()) ...?
    }

如何将每个条目按其键分组，并将每个计数和每个值加到一个映射图中？

谢谢。

Answer 1

您可以使用合并功能通过Stream收集器收集toMap的条目。

public static Map<String, Values> mergeMaps(List<Map<String, Values>> maps) {
    return maps.stream()
               .flatMap(m -> m.entrySet().stream())
               .collect(Collectors.toMap(Map.Entry::getKey,
                                         Map.Entry::getValue,
                                         (v1,v2) -> new Values(v1,v2)));
}

假设您有一个Values构造函数，该构造函数接受两个Values实例并创建一个具有值之和的实例。

当然，您可以在没有该构造函数的情况下编写合并函数。例如：

(v1,v2) -> new Values(v1.getCount()+v2.getCount(),v1.getValue()+v2.getValue())

Answer 2

使用groupingBy的另一种解决方案：

Map<String, Optional<Values>> collect = list.stream()
            .flatMap(map -> map.entrySet().stream())
            .collect(groupingBy(Map.Entry::getKey, mapping(Map.Entry::getValue,
                    reducing((v1, v2) -> new Values(v1.count + v2.count, v1.values + v2.values)))));

注意：此映射的值为Optional<Values>。如果您在其中一个源地图（例如null）上具有map2.put("ddd", null);值它可以避免NullPointerException并返回Optional.empty

通过汇总值将地图列表合并为一张地图

2 个答案: