Java 流：转换地图的值

Question

有一个 Map<A, Set<B>> 的实例，其中 B 是一个带有属性 Integer price （以及其他）的 POJO，我如何优雅地将此映射转换为 Map<A, Integer> 其中 Integer 表示给定集合的 prices 的总和，对于每个键 A?

Answer 1

您可以像这样使用流：

Map<A, Integer> result = map.entrySet().stream()
        .collect(Collectors.toMap(
                Map.Entry::getKey, 
                b -> b.getValue().stream()
                        .mapToInt(B::getPrice)
                        .sum()
                ));

Answer 2

这应该能帮到你。使用 Java 8 中的 Streaming-API：（在此示例中，class“PricePojo”是您的 B，“String”是您的 A）

 Map<String, Integer> newMap = map.entrySet().stream()//
.collect(Collectors.toMap(Entry::getKey, entry -> entry.getValue().stream().mapToInt(PricePojo::getPrice).sum()));

您也可以自己实现 Map.Entry，如下例所示：

    Map<String, Integer> newMap = map.entrySet().stream()//
    .map(entry -> new Map.Entry<String, Integer>() {
    private int sum = entry.getValue().stream().mapToInt(PricePojo::getPrice).sum();
    @Override
    public String getKey() {
        return entry.getKey();
    }

    @Override
    public Integer getValue() {
        return sum;
    }

    @Override
    public Integer setValue(Integer value) {
        return sum = value.intValue();
    }
    }).collect(Collectors.toMap(Entry::getKey, Entry::getValue));

Java 流：转换地图的值

Java streams: transforming map's values

java

collections

dictionary

stream

java-8