如何在继续流式传输的同时获取第一个元素？

Question

我有一个通用项目流。我想打印第一个项目的类名，即所有项目的toString（）。

如果我有一个Iterable，它应该是这样的:

Iterable<E> itemIter = ...;
boolean first = true;
for (E e : itemIter) {
    if (first) {
        first = false;
        System.out.println(e.getClass().getSimpleName());
    }
    System.out.println(e);
}

我可以在流上执行此操作吗（流

*请注意，这是一个关于流的问题——不是关于迭代器的问题。我有一个流——不是一个迭代器。




Answer 1

本机解决方案：Java中的Stream不可重用。这意味着，消费流只能完成一次。如果您从流中获取第一个元素，您可以再次迭代它。

解决方法是创建与第一个相同的另一个流，或者获取第一个项目然后创建一个流，类似于这样：

Stream<E> stream = StreamSupport.stream(Spliterators.spliteratorUnknownSize(sourceIterator, Spliterator.ORDERED), false);
E firstElement = itemIter.next();
stream.foreach(...);

编辑

实际上没有任何方法可以“复制”一个流，您需要保留一个迭代器/集合。这里有更多关于它的信息。当涉及到内存时，一旦流耗尽，它可以被垃圾收集器收集，因为它没有任何用处。流本身并不比它起源的迭代器占用更多的空间。请记住，流可能是无限的。当前操作的元素存储在内存中。

Answer 2

如果您的起点是一个< code >流，并且您希望保留它的所有属性和惰性，下面的解决方案就可以了:

public static <E> Stream<E> forFirst(Stream<E> stream, Consumer<? super E> c) {
    boolean parallel = stream.isParallel();
    Spliterator<E> sp = stream.spliterator();
    return StreamSupport.stream(() -> {
        if(sp.getExactSizeIfKnown() == 0) return sp;
        Stream.Builder<E> b = Stream.builder();
        if(!sp.tryAdvance(b.andThen(c))) return sp;
        return Stream.concat(b.build(), StreamSupport.stream(sp, parallel)).spliterator();
    }, sp.characteristics(), parallel);
}

当你用它与

List<String> list = new ArrayList<>(List.of("foo", "bar", "baz"));
Stream<String> stream = forFirst(
        list.stream().filter(s -> s.startsWith("b")),
        s -> System.out.println(s+" ("+s.getClass().getSimpleName()+')')
    ).map(String::toUpperCase);
list.add(1, "blah");
System.out.println(stream.collect(Collectors.joining(" | ")));

它会打印出来

blah (String)
BLAH | BAR | BAZ

表明在开始终端操作(< code>collect)之前不会开始处理，因此反映了之前对source List的更新。

Answer 3

匿名用户

有一个< code>StreamEx库，它扩展了标准Java的流API。使用< code > streamex . of(Iterator)和< code>peekFirst:

StreamEx.of(itemIter.iterator())
        .peekFirst(e -> System.out.println(e.getClass().getSimpleName()))
        .forEach(System.out::println);