[TOC]
聲明
這一系列文章旨在幫助大家理解 Collector 的執(zhí)行流程，至于實現的是否高效、是否優(yōu)雅、是否合理等暫且不論。
現在來實現一個和 java.util.stream.Collectors#toList 一樣效果的 ToListCollector。
若對 Collector 的各個流程有疑問請移駕此處：http://blog.csdn.net/hylexus/article/details/78941843
簡單實現
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);
}
