Collectors收集器的高级用法
场景1：获取关联的班级名称
原先如果需要通过关联字段拿到其他表的某个字段，只能遍历List匹配获取
for (Student student : studentList) {
Long clazzId = student.getClazzId();
// 遍历班级列表，获取学生对应班级名称
for (Clazz clazz : clazzList) {
if(ObjectUtil.equal(clazzId, clazz.getClazzId())) {
student.setClazzName(clazz.getClazzName());
break;
}
}
}
但是如果使用 toMap 方法，可以一步到位，代码的可读性是不是要比上面那种方式好很多？
Map<Long, String> clazzMap = clazzList.stream()
.collect(Collectors.toMap(Clazz::getClazzId, Clazz::getClazzName));
// 使用Map进行匹配，直接通过get方法获取班级名称
studentList.forEach(i -> i.setClazzName(clazzMap.get(i.getClazzId())));
如果你想关联整个对象而不是某个字段，也没关系，可以像下面这样
Map<Long, Clazz> clazzMap = clazzList.stream()
.collect(Collectors.toMap(Clazz::getClazzId, Function.identity()));
// 使用Map关联班级对象
studentList.forEach(i -> i.setClazz(clazzMap.get(i.getClazzId())));
场景2：统计图书的借阅次数
有一些时候，我们需要对列表的数据做一些统计，比如图书的借阅次数，活动的参与人数等，这些统计可以通过循环手动遍历进行计算
for (BookInfo bookInfo : bookList) {
long borrowNumber = 0L;
Long bookId = bookInfo.getBookId();
for (BorrowRecord record : recordList) {
// 次数+1
if(ObjectUtil.equal(bookId, record.getBookId())) {
borrowNumber ++;
}
}
bookInfo.setBorrowNumber(borrowNumber);
}
上面的这种写法其实还好，但是业务如果复杂一些的话，写起来就比较麻烦了，但是如果使用 toMap 方法，可以一步到位
Map<Long, Long> numberMap = recordList.stream()
.collect(Collectors.toMap(BorrowRecord::getBookId, e -> 1L, Long::sum));
// 统计图书的借阅次数
bookList.forEach(i -> i.setBorrowNumber(numberMap.get(i.getBookId())));
场景3：根据城市分组，查看城市都有哪些姓氏的人
有些业务场景比较复杂，需要对我们对数据进行分组统计，这些情况下，就需要用到 groupingBy() 方法了
Map<City, Set<String>> lastNamesByCity
= people.stream().collect(
Collectors.groupingBy(Person::getCity,
Collectors.mapping(Person::getLastName, Collectors.toSet())));
上面的方法，其实不是唯一的写法，我们还可以这样写
Map<City, Set<String>> lastNamesByCity
= people.stream().collect(
Collectors.groupingBy(Person::getCity,
Collectors.flatMapping((i) -> Stream.of(i.getLastName()), Collectors.toSet())));
Map 也可以间接实现分组的效果，让我们换成 toMap() 方法试试
Map<City, Set<String>> lastNamesByCity
= people.stream().collect(
Collectors.toMap(Person::getCity, (v) -> {
Set<String> set = Sets.newHashSet();
set.add(v.getLastName());
return set;
},
(a, b) -> {
(a).addAll(b);
return a;
}, HashMap::new));
另外，如果最后的统计不要求姓氏写成集合，那么我们还可以写成这样
Map<City, String> lastNamesByCity
= people.stream().collect(
Collectors.toMap(Person::getCity,
Person::getLastName, (a, b) -> a + "," + b));
能看得出来，在分组这件事上，groupingBy() 方法更适合，使用 toMap() 方法虽然也能实现，但是写出的代码会更复杂。
场景4：根据客户名称分组，查看订单都有哪些商品
如果需要做购物商城一类的项目，那么必然就离不开订单、用户和商品，根据客户统计订单商品，也是一个非常复杂的业务逻辑，因此这里我们可以用到 groupingBy() 方法
Map<String, Set<Product>> productsByCustomerName
= orders.stream().collect(
Collectors.groupingBy(Order::getCustomName,
Collectors.flatMapping(order -> order.getProducts().stream(),
Collectors.toSet())));
让我们换一种写法试试
Map<String, Set<Product>> productsByCustomerName
= orders.stream().collect(
Collectors.groupingBy(Order::getCustomName,
Collectors.collectingAndThen(Collectors.toSet(),
i -> i.stream().flatMap(p -> p.getProducts().stream()).collect(Collectors.toSet()))));
换成 toMap() 如何
Map<String, Set<Product>> productsByCustomerName
= orders.stream().collect(
Collectors.toMap(Order::getCustomName,
(v) -> Sets.newHashSet(v.getProducts()),
(a, b) -> {
(a).addAll(b);
return a;
}
)
);
场景5：根据部门分组，查看工资大于2000的员工
说到部门，那就不得不提到人员，对人员的各种信息进行统计，我们依然可以使用 groupingBy() 方法
Map<Department, Set<Employee>> wellPaidEmployeesByDepartment
= employees.stream().collect(
Collectors.groupingBy(Employee::getDepartment,
Collectors.filtering(e -> e.getSalary() > 2000,
Collectors.toCollection(HashSet::new))));
同样地，如果你喜欢使用 toMap()，也可以像下面这样写
Map<Department, Set<Employee>> wellPaidEmployeesByDepartment
= employees.stream()
.filter(e -> e.getSalary() > 2000)
.collect(Collectors.toMap(Employee::getDepartment,
Sets::newHashSet,
(a, b) -> {
(a).addAll(b);
return a;
}));
场景6：区分及格和不及格的学生
统计学生的成绩是一大麻烦事，很多数据需要统计到，但如果只是区分及格和不及格的学生，这里我们可以使用 partitioningBy() 方法，将学生分为两个区间
Map<Boolean, List<Student>> passingFailing = students.stream()
.collect(Collectors.partitioningBy(s -> s.getGrade() >= DataUtils.PASS_THRESHOLD));
partitioningBy 相当于另一种版本的 groupingBy，但是数据最多只有两组，因为 partitioningBy 是通过指定的条件进行分组的，满足的在一边，不满足的在另一边。
场景7：同时统计订单数以及综合评分
从 JDK12 开始，Collectors 新增了一个方法 teeing，直译为发球，这样翻译可能听不太懂，但我给你看一下怎么用，你就知道了
Pair<Map<Long, Long>, Map<Long, Double>> pair = orderList.stream().collect(
Collectors.teeing(
Collectors.groupingBy(BookOrder::getHotelId, Collectors.counting()),
Collectors.groupingBy(BookOrder::getHotelId, Collectors.averagingDouble(BookOrder::getOrderRating)),
Pair::of));
Collectors.groupingBy(BookOrder::getHotelId, Collectors.counting()),
Collectors.groupingBy(BookOrder::getHotelId, Collectors.averagingDouble(BookOrder::getOrderRating)),
Pair::of));
是的，teeing 方法接收两个参数，这两个参数分别会得出一个结果，把两个结果再进行处理，就是第三个参数，就这么简单。利用这个方法我们可以分别统计两个结果（比如最值，平均数）并对这两个结果再进行处理，非常好用