有条件地添加多个值的Java流

Question

class Sale {
   String country;
   BigDecimal amount;
}

country | amount
nl      | 9
nl      | -3
be      | 7.9
be      | -7

是否有一种方法可以使用单个流来实现映射 > ？

用两个独立的流很容易做到这一点，但我不能只用一个。

Answer 1

它应该使用collectors.tomap和合并函数来求和对。

假设对是不可变的，并且只有第一个和第二个元素的getter，代码可能如下所示：

static Map<String, Pair<BigDecimal, BigDecimal>> sumUp(List<Sale> list) {
    return list.stream()
               .collect(Collectors.toMap(
                   Sale::getCountry,
                   sale -> sale.getAmount().signum() >= 0 
                       ? new Pair<>(sale.getAmount(), BigDecimal.ZERO)
                       : new Pair<>(BigDecimal.ZERO, sale.getAmount()),
                   (pair1, pair2) -> new Pair<>(
                       pair1.getFirst().add(pair2.getFirst()),
                       pair1.getSecond().add(pair2.getSecond())
                   )
                   // , LinkedHashMap::new // optional parameter to keep insertion order
               ));
}

测试

List<Sale> list = Arrays.asList(
    new Sale("us", new BigDecimal(100)),
    new Sale("uk", new BigDecimal(-10)),
    new Sale("us", new BigDecimal(-50)),
    new Sale("us", new BigDecimal(200)),
    new Sale("uk", new BigDecimal(333)),
    new Sale("uk", new BigDecimal(-70))
);

Map<String, Pair<BigDecimal, BigDecimal>> map = sumUp(list);

map.forEach((country, pair) -> 
    System.out.printf("%-4s|%s%n%-4s|%s%n", 
        country, pair.getFirst(), country, pair.getSecond()
));

uk  |333
uk  |-80
us  |300
us  |-50