Java流，如何GROUP BY并查找流的乘积

Question

我有一个流，可以获取流中每个余额的“每日返回”，但我想获取每个 accountId 的所有每日返回的乘积。我知道我需要添加类似的内容:

.collect(Collectors.groupingBy(n -> n.getAccountId(), Collectors.reducing(BigDecimal.ZERO, BigDecimal::multiply)));

但是我不知道如何。这是到目前为止的流:

Map<Long, Double> solution = new HashMap<Long, Double>();

balances.stream()
          .forEach(n -> {
              if(n.getCashflow().compareTo(BigDecimal.ZERO) > 0) {
                  solution.put(n.getAccountId(), n.getEndingBalance().divide((n.getStartingBalance().add(n.getCashflow())), 6, RoundingMode.FLOOR).doubleValue());
              }
              else {
                  solution.put(n.getAccountId(), (n.getEndingBalance().subtract(n.getCashflow())).divide(n.getStartingBalance(), 6, RoundingMode.FLOOR).doubleValue());
              }
              });

请随时提出任何实现此解决方案的方法，谢谢!

Answer 1

为了更容易理解，我们将每日 yield 的计算与流分开:

Function<Balance, BigDecimal> dailyReturn = n -> {
    if (...) {  
        return ...;
    } else {
        return ...;
    }
};   // or use a ternary statement

现在我们可以使用 3 参数 toMap 收集器收集每个帐户的所有每日 yield 的乘积:

balances.stream().collect(Collectors.toMap(
    Balance::getAccountId, 
    dailyReturn, 
    BigDecimal::multiply
));