JAVA 8 StreamAPI 和 lambda表达式 总结(四)--stream的一些聚合操作

3/8/2017来源:ASP.NET技巧人气:3005

在sql里,一些聚合操作max,count,或者只是进行一些分组,这些我们都很熟悉,在Stream API中,我们也可以做一些聚合操作.

我们定义一个类Group,里面有班级号,分数和学生的名字

package JDK.StreamAndLambda.groupToMap; /** * Created by panqian on 2017/2/28. */ public class Group { String classes; String num; String name; public Group() { } public Group(String classes, String num, String name) { this.classes = classes; this.num = num; this.name = name; } public String getClasses() { return classes; } public void setClasses(String classes) { this.classes = classes; } public String getNum() { return num; } public void setNum(String num) { this.num = num; } public String getName() { return name; } public void setName(String name) { this.name = name; } }

然后创建一个ArrayList集合,分别添加四个学生,他们可能是同班同学。

ArrayList<Group> groups = new ArrayList<>(); groups.add(new Group("1321813", "88", "panqian01")); groups.add(new Group("1321813", "100", "panqian02")); groups.add(new Group("1321814", "99", "panqian03")); groups.add(new Group("1321815", "70", "panqian04")); //所有人按班级分组 Map<String, List<Group>> collect = groups.stream().collect(Collectors.groupingBy(Group::getClasses)); Set<String> strings = collect.keySet(); for (String s : strings) { System.out.PRintln(s + ":"); collect.get(s).forEach(x -> System.out.println(x.getName())); }

程序执行后,我们可以看到如下结果。它往collect方法里面传了一个groupingBy(班级)静态方法,跟sql 的 group by 很像,得到了是一个map结果,我们可以知道每个班里面有哪些人:

1321813: panqian01 panqian02 1321814: panqian03 1321815: panqian04

如果我们想把学生的名字用”,”连接,程序可以稍作修改,groupingBy方法的第二个参数可以对每个结果进行map处理。

//所有人按班级分组(名字用,连接) Map<String, String> collect4 = groups.stream().collect(Collectors.groupingBy(Group::getClasses, Collectors.mapping(Group::getName, Collectors.joining(",")))); Set<String> strings4 = collect4.keySet(); for (String s : strings4) { System.out.println(s + ":"); System.out.println(collect4.get(s)); }

得出结果如下:

1321813: panqian01,panqian02 1321814: panqian03 1321815: panqian04

第三个需求是我们以90分为界限,分别找出90以上的人和90分以下的人,partitioningBy静态方法返回的是带布尔值的Collector,可以对元素处理之后返回的true或false为依据进行分组。

//按成绩90分为界限 进行分组 Map<Boolean, List<Group>> collect1 = groups.stream().collect(Collectors.partitioningBy(x -> Integer.valueOf(x.getNum()) > 90)); Set<Boolean> strings1 = collect1.keySet(); for (Boolean s : strings1) { System.out.println(s + ":"); collect1.get(s).forEach(x -> System.out.println(x.getName())); } //90分以下的人 false: panqian01 panqian04 //90分以上的人 true: panqian02 panqian03

第四个需求,计算每个班的班级人数,先用groupingBy静态方法进行班级分组,然后第二个参数传入counting静态方法进行聚合操作,从而对每个组计算总人数

//计算每个班的班级人数 Map<String, Long> collect2 = groups.stream().collect(Collectors.groupingBy(Group::getClasses, Collectors.counting())); Set<String> strings2 = collect2.keySet(); for (String s : strings2) { System.out.println(s + ":"); System.out.println(collect2.get(s)); } 1321813: 2 1321814: 1 1321815: 1

最后一个需求,计算每个班的最高分,在groupingBy的第二个参数,我们可以传入maxBy方法,求最大值,在maxBy方法里,需要我们传入自定义的Comparator比较器,从而对分数进行排序,最后得出每个班得分最高的人

//计算每个班的最高分 Map<String, Optional<Group>> collect3 = groups.stream().collect(Collectors.groupingBy(Group::getClasses, Collectors.maxBy(new Comparator<Group>() { @Override public int compare(Group o1, Group o2) { if (Integer.valueOf(o1.getNum()) > Integer.valueOf(o2.getNum())) return 1; return -1; } }))); Set<String> strings3 = collect3.keySet(); for (String s : strings3) { System.out.println(s + ":"); Optional<Group> group = collect3.get(s); group.ifPresent(x -> System.out.println(x.getName())); } 1321813: panqian02 1321814: panqian03 1321815: panqian04