java stream分组之后求和

68 阅读 0 评论 45 点赞

我是靠谱客的博主朴实石头，最近开发中收集的这篇文章主要介绍java stream分组之后求和，觉得挺不错的，现在分享给大家，希望可以做个参考。

概述

注：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();
        })));