使用 Java Stream 计算包含在另一个集合中的集合中包含的字段的总和

Question

我想使用 java stream() 计算包含在另一个集合中的集合中包含的字段的总和。 如果计算字段直接在第一个集合中，我只知道该怎么做，像这样：

return collectionOfObjectA.stream() 
             .mapToDouble(c1 -> c1.getSalary()).sum();

但是现在“collectionOfObjectA”包含另一个“collectionOfObjectB”，而最后一个是包含“salary”字段的那个。

对象A：

{
 id:1,
Collection<ObjectB>
}

和对象B：

{
 id:1,
 salary: 4000
}

我现在的计算方式如下：

     Double result=0.0;
        for (ObjectA obj: collectionOfObjectA())
        {
            for(ObjectB objB : objA.getcollectionOfObjectB()){
                result+= objB.getSalary();
            }
        }

        return result;

Answer 1

在mapToDouble 之前，您必须像这样使用flatMap：

return collection1.stream()
        .flatMap(cA -> cA.getCollectionB().stream())
        .mapToDouble(ObjectB::getSalary)
        .sum();

或者你可以在flatMap之前使用map：

return collection1.stream()
        .map(ObjectA::getCollectionB)
        .flatMap(List::stream)
        .mapToDouble(ObjectB::getSalary)
        .sum();

你可以像@Naman 提到的那样直接使用flatMapToDouble：

return collection1.stream()
        .flatMapToDouble(
                cA -> cA.getCollectionB().stream().mapToDouble(ObjectB::getSalary)
        ).sum();