Java8自定義實現Collector接口

/**
 * Created BY poplar ON 2019/11/21
 * 自定義實現Collector接口
 */
public class CustomCollector<T> implements Collector<T, Set<T>, Set<T>> {

    @Override
    public Supplier<Set<T>> supplier() {
        System.out.println("supplier invoked");
        //直接返回一個HashSet容器
        return HashSet::new;
    }

    @Override
    public BiConsumer<Set<T>, T> accumulator() {
        System.out.println("accumulator invoked");
        //此處不能直接返回具體實現類,因爲supplier中已經指定了容器具體類型
        return Set::add;
    }

    @Override
    public BinaryOperator<Set<T>> combiner() {
        System.out.println("combiner invoked");
        //接收2個參數返回一個
        return (p1, p2) -> {
            p1.addAll(p2);
            return p1;
        };
    }

    @Override
    public Function<Set<T>, Set<T>> finisher() {
        System.out.println("finisher invoked");
        //return Function.identity();
        return t -> t;
    }

    @Override
    public Set<Characteristics> characteristics() {
        System.out.println("characteristics invoked");
        return Collections.unmodifiableSet(EnumSet.of(Characteristics.IDENTITY_FINISH, Characteristics.UNORDERED));
    }

    public static void main(String[] args) {
        List<String> list = Arrays.asList("hello", "hello", "world", "beautiful cat", "severn cat");
        Set<String> set = list.stream().collect(new CustomCollector<>());
        System.out.println(set);
        /*
        執行結果:
        supplier invoked
         accumulator invoked
         combiner invoked
         characteristics invoked
         characteristics invoked
         [world, severn cat, hello, beautiful cat]*/
    }
}

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章