Java流组和多个字段相加

Question

我有一个List fooList

class Foo {
    private String category;
    private int amount;
    private int price;

    ... constructor, getters & setters
}

我想按类别分组,然后总和金额以及价格.

结果将存储在地图中:

Map<Foo, List<Foo>> map = new HashMap<>();

关键是Foo持有汇总的金额和价格,列表作为具有相同类别的所有对象的值.

到目前为止,我已经尝试了以下内容:

Map<String, List<Foo>> map = fooList.stream().collect(groupingBy(Foo::getCategory()));

现在我只需要用保存汇总数量和价格的Foo对象替换String键.这是我被困的地方.我似乎无法找到任何办法.

Answer 1

有点难看,但它应该工作:

list.stream().collect(Collectors.groupingBy(Foo::getCategory))
    .entrySet().stream()
    .collect(Collectors.toMap(x -> {
        int sumAmount = x.getValue().stream().mapToInt(Foo::getAmount).sum();
        int sumPrice= x.getValue().stream().mapToInt(Foo::getPrice).sum();
        return new Foo(x.getKey(), sumAmount, sumPrice);
    }, Map.Entry::getValue));

Answer 2

我对Sweepers答案的变化形式是使用减少型收集器，而不是两次流式传输来汇总各个字段：

        Map<Foo, List<Foo>> map = fooList.stream()
                .collect(Collectors.groupingBy(Foo::getCategory))
                .entrySet().stream()
                .collect(Collectors.toMap(e -> e.getValue().stream().collect(
                            Collectors.reducing((l, r) -> new Foo(l.getCategory(),
                                        l.getAmount() + r.getAmount(),
                                        l.getPrice() + r.getPrice())))
                            .get(), 
                            e -> e.getValue()));

不过，它并不是真的更好，因为它会造成很多短暂的后果Foos。

但是请注意，Foo提供hashCode和equals实现只需要考虑category到结果map正确工作所必需。Foo通常，这可能不是您想要的。我希望定义一个单独的FooSummary类来包含聚合数据。