在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:panqian01panqian021321814:panqian031321815: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,panqian021321814:panqian031321815: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:panqian01panqian04//90分以上的人true:panqian02panqian03第四个需求,计算每个班的班级人数,先用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:21321814:11321815: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:panqian021321814:panqian031321815:panqian04新闻热点
疑难解答