您好,欢迎来到花图问答。
搜索
您的当前位置:首页java8-07-自定义Collector-ToList

java8-07-自定义Collector-ToList

来源:花图问答

[TOC]

声明

这一系列文章旨在帮助大家理解 Collector 的执行流程,至于实现的是否高效、是否优雅、是否合理等暂且不论。


现在来实现一个和 java.util.stream.Collectors#toList 一样效果的 ToListCollector

简单实现

static class ToListCollector<T> implements Collector<T, List<T>, List<T>> {

    @Override
    public Supplier<List<T>> supplier() {
        return () -> Lists.newArrayList();
    }

    @Override
    public BiConsumer<List<T>, T> accumulator() {
        return (list, e) -> list.add(e);
    }

    @Override
    public BinaryOperator<List<T>> combiner() {
        return (acc1, acc2) -> {
            acc1.addAll(acc2);
            return acc1;
        };
    }

    @Override
    public Function<List<T>, List<T>> finisher() {
        return (acc) -> acc;
    }

    @Override
    public Set<Characteristics> characteristics() {
        Set<Characteristics> ret = new HashSet<>();
        ret.add(Characteristics.CONCURRENT);
        ret.add(Characteristics.IDENTITY_FINISH);
        return Collections.unmodifiableSet(ret);
    }
}

简化代码

static class ToListCollector2<T> implements Collector<T, List<T>, List<T>> {

    @Override
    public Supplier<List<T>> supplier() {
        return ArrayList::new;
    }

    @Override
    public BiConsumer<List<T>, T> accumulator() {
        return List::add;
    }

    @Override
    public BinaryOperator<List<T>> combiner() {
        return (acc1, acc2) -> {
            acc1.addAll(acc2);
            return acc1;
        };
    }

    @Override
    public Function<List<T>, List<T>> finisher() {
        return Function.identity();
    }

    @Override
    public Set<Characteristics> characteristics() {
        return Collections.unmodifiableSet(
                EnumSet.of(
                        Characteristics.CONCURRENT,
                        Characteristics.IDENTITY_FINISH
                )
        );
    }
}

使用自定义Collector

@Test
public void testListCollector() {
    List<Integer> list = IntStream.rangeClosed(1, 5).boxed()
      .collect(new ToListCollector<>());
    System.out.println(list);

    List<Integer> list2 = Stream.iterate(1, i -> i + 1).limit(5)
      .collect(new ToListCollector2<>());
    System.out.println(list2);
}

博客源码

Copyright © 2019- huatuowenda.com 版权所有 湘ICP备2023022495号-1

违法及侵权请联系:TEL:199 1889 7713 E-MAIL:2724546146@qq.com

本站由北京市万商天勤律师事务所王兴未律师提供法律服务