java - How would you make a java.util.stream.Collector that collects to com.google.common.collect.ImmutableSet?

Question

That seems too tricky for me since ImmutableSet instances are only built with ImmutableSet.Builder instances, which don't implement Collection so you can't just use Collectors.toCollection(ImmutableSet::new) or Collectors.toCollection(ImmutableSet.Builder::new).

Answer 1

这是现在内置在番石榴中的，

不可变集#toImmutableSet

使用喜欢，

something.stream().collect(ImmutableSet.toImmutableSet())

Answer 2

事实上，在 3 个月后 :-)，您可以使用Collector.of并将其包装在一个简单的实用方法中，而不是为此定义一个完整的类：

public static <T> Collector<T, Builder<T>, ImmutableSet<T>> immutableSetCollector() {
    return Collector.of(Builder<T>::new, Builder<T>::add, (s, r) -> s.addAll(r.build()), Builder<T>::build);
}

接着：

import com.google.common.collect.ImmutableSet;
import com.google.common.collect.ImmutableSet.Builder;
....

ImmutableSet<Point2D> set = 
    Stream.of(new Point2D(1, 2), ...).collect(immutableSetCollector());

Answer 3

Collector我们可以通过扩展接口来编写自定义收集器。

这是一个可靠的参考，详细地讨论了它：http: //java.dzone.com/articles/introduction-writing-custom

Answer 4

这是我发现的最接近的：

Set<String> set = list.stream().collect(ImmutableSet.Builder<String>::new, ImmutableSet.Builder<String>::add, (builder1, builder2) -> builder1.addAll(builder2.build())).build();

没有将一个构建器的元素附加到另一个构建器的方法，只有 Iterable 或 Iterator，因此您不能为此使用方法引用。

Answer 5

因此，在这种情况下编写自定义收集器似乎并不像我想象的那么困难：

package org.tendiwa.collections;

import com.google.common.collect.ImmutableSet;

import java.util.Set;
import java.util.function.BiConsumer;
import java.util.function.BinaryOperator;
import java.util.function.Function;
import java.util.function.Supplier;
import java.util.stream.Collector;

public class ImmutableSetCollector<T> implements Collector<T, ImmutableSet.Builder<T>, ImmutableSet<T>> {

    @Override
    public Supplier<ImmutableSet.Builder<T>> supplier() {
        return ImmutableSet.Builder::new;
    }

    @Override
    public BiConsumer<ImmutableSet.Builder<T>, T> accumulator() {
        return (builder, element) -> builder.add(element);
    }

    @Override
    public BinaryOperator<ImmutableSet.Builder<T>> combiner() {
        return (b1, b2) -> b1.addAll(b2.build());
    }

    @Override
    public Function<ImmutableSet.Builder<T>, ImmutableSet<T>> finisher() {
        return ImmutableSet.Builder::build;
    }

    @Override
    public Set<Characteristics> characteristics() {
        return ImmutableSet.of();
    }
}

用法：

public static void main(String[] args) {
    ImmutableSet<Point2D> set = Arrays.asList(
        new Point2D(1, 2),
        new Point2D(2, 4),
        new Point2D(3, 5),
        new Point2D(4, 4),
        new Point2D(5, 6),
        new Point2D(8, 6)
    ).stream().collect(new ImmutableSetCollector<>());
    System.out.println(set);
}

输出：

[{1.0:2.0}, {2.0:4.0}, {3.0:5.0}, {4.0:4.0}, {5.0:6.0}, {8.0:6.0}]

Answer 6

这个功能在guava-jdk8库（我编写的）中是开箱即用的。