Java 8 Streams - 从流中对类型中的多个值求和

Question

我正在尝试对一个类中从集合中流式传输的两个变量的总和进行求和。考虑以下简单的类。

class Widget {
    String colour;
    Double legs;
    Double arms;
    // ...
}

我想流式传输一组小部件，按颜色对它们进行分组并计算出（腿 + 手臂）的总和，以找到每种颜色的肢体总数。

我想做一些简单的事情，像这样 -

widgets.stream().collect(groupingBy(Widget::getColour, summingDouble(legs + arms)));

但是，没有 Collectors 函数可以让我指定一个自定义值来求和。

我的问题是：我需要编写自定义收集器还是有其他简单的方法来实现我想要的？

Answer 1

你可以使用Collectors.summingDouble():

Map<String, Double> result = widgets.stream()
        .collect(Collectors.groupingBy(
                Widget::getColour,
                Collectors.summingDouble(w -> w.getArms() + w.getLegs())
        ));

您可以只使用函数w -> w.getArms() + w.getLegs() 对手臂和腿求和。

Answer 2

您也可以在这里使用toMap：

Map<String, Double> colorToLimbsCount = widgets.stream()
        .collect(Collectors.toMap(Widget::getColour,
                widget -> widget.getArms() + widget.getLegs(), Double::sum));

在groupingBy 表示中，它可能看起来像：

Map<String, Double> colorToLimbsCount = widgets.stream()
        .collect(Collectors.groupingBy(Widget::getColour,
                Collectors.reducing((double) 0,
                        widget -> widget.getArms() + widget.getLegs(), Double::sum)));