使用Java 8流的每种元素类型的特定使用者

Question

我有一个Java 8数字流:

Stream<Number> numbers = ...;

我想迭代流并根据每个元素的类型调用特定的使用者.即,对于Integer元素,我想调用Consumer<Integer>,对Long一个Consumer<Long>等

有forEach()方法,但这需要一个Consumer<? super Number>,要求实现(通常是一个Lambda表达式)instanceof-切换精确类型本身.

就API而言,我基本上是在寻找这样的东西:

numbers.forEach( callbacks -> {
    callbacks.on( Integer.class, i -> { /* handle Integer */ } );
    callbacks.on( Long.class, l -> { /* handle Long */ } )
} );

是否有任何现有的API允许我以类似于此的方式为每个流元素子类型注册特定的消费者？

Answer 1

你确定你不想只运行两次流吗？它会更具可读性.

但是如果你愿意,你可以像这样定义一个类型检查用户:

public static<T> Consumer<Object> acceptType(Class<T> clazz, Consumer<? super T> cons) {
    return t -> {
        if (clazz.isInstance(t)) {
            cons.accept(clazz.cast(t));
        }
    };
}

然后,您可以使用andThen以下方法组合多个

Consumer<Object> combined = acceptType(Integer.class, i -> ...)
                   .andThen(acceptType(Long.class, lng -> ...))

如果要隐藏andThen,可以定义

static<T> Consumer<T> doAll(Consumer<T>... consumers) {
    return Arrays.stream(consumers)
            .reduce(Consumer::andThen)
            .orElse(t -> {});
}

然后它变成了

nums.forEach(doAll(
            acceptType(Integer.class, i -> ...),
            acceptType(Long.class, lng -> ..),
            ...
));