java stream分组之后求和
癞蛤蟆吃了小天鹅
于 2022-08-26 09:37:42 发布
6023
收藏 4
文章标签： java
版权
注：elementComponentDtos.stream().mapToDouble(ElementComponentDto::getAmount).sum();为求和可以根据返回类型的不同去改变相对应的求和函数（mapToDouble）
注 BigDecimal为了保留double类型两位小数（方法二和三没做处理）
注：collect1为原数据 collect为返回结果集
方法一：
Map<String, List<ElementComponentDto>> collect2 = collect1.stream().collect(Collectors.groupingBy(o -> o.getName()));
for (Map.Entry entry2 : collect2.entrySet()) {
List<ElementComponentDto> elementComponentDtos = collect2.get(entry2.getKey());
double sum = elementComponentDtos.stream().mapToDouble(ElementComponentDto::getAmount).sum();
BigDecimal bigDecimal = new BigDecimal(sum).setScale(2, RoundingMode.HALF_UP);
double newDouble = bigDecimal.doubleValue();
List<ElementComponentDto> collect3 = elementComponentDtos.stream().filter(Predicate).
map(e -> new ElementComponentDto(null, null, null, e.getName(), e.getCompany(), newDouble)).collect(Collectors.toList());
collect3.forEach(e -> {
collect.add(e);
});
}
方法二：
注：collect4 （ElementComponentDto）即为返回数据已经包装好的数据，你只用把ElementComponentDto放入你想要的具体返回结果集里就好。Map<String, ElementComponentDto>String 即为想要按照那个字段分组的字段。ElementComponentDto.builder()为你要返回类型去创建对象
Map<String, ElementComponentDto> collect4 = collect1.stream().collect(Collectors.groupingBy(ElementComponentDto::getName)).entrySet()
.stream().collect(Collectors.toMap(o->o.getKey(), o -> {
List<ElementComponentDto> l = o.getValue();
double sum = l.stream().mapToDouble(ElementComponentDto::getAmount).sum();
return ElementComponentDto.builder().company(l.get(0).getCompany()).amount(sum).name(l.get(0).getName()).build();
}));
方法三： 同方法二
HashMap<String, ElementComponentDto> collect3 = collect1.stream().collect(Collectors.groupingBy(o -> o.getName(),
HashMap::new, Collectors.collectingAndThen(Collectors.toList(), l -> {
double sum = l.stream().mapToDouble(ElementComponentDto::getAmount).sum();
return ElementComponentDto.builder().company(l.get(0).getCompany()).amount(sum).name(l.get(0).getName())
.build();
})));
————————————————