java stream 分组求和_Java stream List 求和、分组操作

Java stream List 求和、分组操作

前言

项目中经常会使用Stream操作一些集合数据，今天记录一下我经常使用的Stream操作

求和操作public static void main(String[] args) {

List users = generateUserList();int reduce = users.stream().mapToInt(x -> x.getSalary().intValue()).reduce(0, Integer::sum);

log.info("Int sum - {}", reduce);

BigDecimal reduce1 = users.stream().map(User::getSalary).reduce(BigDecimal.ZERO, BigDecimal::add);

log.info("BigDecimal sum - {}", reduce);

}private static List generateUserList() {

List users = new ArrayList<>();for (int i = 0; i 

users.add(new User((long) (i + 1), StrFormatter.format("用户-{}", i), RandomUtil.randomBigDecimal(new BigDecimal(1000), new BigDecimal(100000))));

}return users;

}复制代码

结果：16:03:32.966 [main] INFO com.springboot.example.stream.StreamTest - Int sum - 31128616:03:32.977 [main] INFO com.springboot.example.stream.StreamTest - BigDecimal sum - 311286复制代码

这里使用了Integer 及BigDecimal类型的求和，其他的例如Long Double类型的自行尝试，操作都是一样的。

分组操作Map> groups = users.stream().collect(Collectors.groupingBy(User::getName, Collectors.toList()));复制代码

结果16:19:24.018 [main] INFO com.springboot.example.stream.StreamTest - groups - {"用户-0":[{"name":"用户-0","id":1,"salary":44133.974869422615}],"用户-1":[{"name":"用户-1","id":2,"salary":32433.52878974182}],"用户-4":[{"name":"用户-4","id":5,"salary":26726.1799749091}],"用户-2":[{"name":"用户-2","id":3,"salary":83274.99312309967}],"用户-3":[{"name":"用户-3","id":4,"salary":1579.9248811900475}]}复制代码

自定义分组的数据结构List groupCustomize = users.stream().collect(Collectors.groupingBy(User::getName, Collectors.toList())).entrySet().stream().map(e -> {

Map newObj = new HashMap<>();

newObj.put("name", e.getKey());

newObj.put("items", e.getValue());return newObj;

}).collect(Collectors.toList());复制代码

结果16:24:23.554 [main] INFO com.springboot.example.stream.StreamTest - groups customize - [{"name":"用户-0","items":[{"name":"用户-0","id":1,"salary":69329.43639757435}]},{"name":"用户-1","items":[{"name":"用户-1","id":2,"salary":14632.879595342842}]},{"name":"用户-4","items":[{"name":"用户-4","id":5,"salary":52953.301688394306}]},{"name":"用户-2","items":[{"name":"用户-2","id":3,"salary":29500.813638173757}]},{"name":"用户-3","items":[{"name":"用户-3","id":4,"salary":30750.353923258994}]}]复制代码