【Java】Java8 stream实现Collector接口来自定义收集器

一、前言

• 自从学习了Java 8函数是编程后，彻底的被Java 8所以吸引。其中最重要的一个优点就是Stream API 。stream和lambda表达式让我们平时对集合的操作更加的便捷，省去复杂的for循环嵌套，和繁琐的容器创建,让代码更加美观。
• 在stream api 的终止操作中，Java已经为我们提供了许多方便好用的收集器，使我们能够从流中收集到我们想要的结果。但是有的时候内置的收集器也可能不能满足大家的需求，这个时候我们就可以利用Java预留的接口来定制自己的收集器了。

二、代码

2.1 从最简单的开始

• 怎么用stream把一个Person类的集合按照年龄分组，得到结果Map<Integer,List< Person >>？
• 其实这个在平时的开发中用到的挺多的，用Collectors.groupingBy()收集器很好实现实。
• 这个方法需要传入一个Function，也就是获取key的Function

2.1.1 源码

public static <T, K> Collector<T, ?, Map<K, List<T>>> groupingBy(Function<? super T, ? extends K> classifier) {
   
   
    return groupingBy(classifier, toList());
}

2.1.2 使用示例

@Slf4j
public class NfzdTest {
   
   

	@Test
	public void test02() {
   
   
	    List<Person> data = this.getPersonList();
	    Map<Integer, List<Person>> collect = data.stream().collect(Collectors.groupingBy(Person::getAge));
	    collect.entrySet().forEach(a -> System.out.println("key:  " + a.getKey() + " value: " + a.getValue()));
	}
	
    private List<Person> getPersonList() {
   
   
        return Arrays.asList(
                new Person("小花", 12),
                new Person("小明", 14),
                new Person("小马", 14),
                new Person("小坏", 13),
                new Person("小红", 13),
                new Person("小赵", 14)
        );
    }
}

2.1.3 Person类

package com.lh.nfzd.model;

import com.lh.nfzd.common.annotation.CheckAge;
import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.NoArgsConstructor;

@Data
@AllArgsConstructor
@NoArgsConstructor
public class Person {
   
   

    private String name;

    @CheckAge(min = 5, max = 20)
    private Integer age;

}

2.1.4 运行结果

2.2 接下来

• 怎么从Person类的集合里按照年龄分组，获取姓名集合，得到结果Map<Integer,List< String >>。
• 刚开始看到这个其实我也不知道怎们办，后来查了资料才有的结果。可以看到groupingBy方法的第二个参数是一个收集器