是否存在限制（可能是无限的）Stream的Java 8流操作，直到第一个元素与谓词不匹配为止？
在Java 9中，我们可以使用takeWhile，如下例所示，打印小于10的所有数字。
IntStream
.iterate(1, n -> n + 1)
.takeWhile(n -> n < 10)
.forEach(System.out::println);
由于Java 8中没有这样的操作，以一般方式实现它的最佳方法是什么？
答案 0 :(得分：140)
已将操作takeWhile和dropWhile添加到JDK 9.您的示例代码
IntStream
.iterate(1, n -> n + 1)
.takeWhile(n -> n < 10)
.forEach(System.out::println);
的行为与您期望的完全相同。
JDK 9已经发布。可在此处下载：http://jdk.java.net/9/
答案 1 :(得分：77)
这样的操作应该是可能的与Java 8 Stream，但它不一定能有效地完成 - 例如，你不一定能并行化这样的操作，因为你必须按顺序查看元素。
API并没有提供一种简单的方法，但最简单的方法是采用Stream.iterator()，包裹Iterator以实现“需要”实施，然后返回Spliterator，然后返回Stream。或者 - 也许 - 包装Spliterator，尽管在此实现中它不能再被拆分。
以下是takeWhile上Spliterator的未经测试的实施：
static <T> Spliterator<T> takeWhile(
Spliterator<T> splitr, Predicate<? super T> predicate) {
return new Spliterators.AbstractSpliterator<T>(splitr.estimateSize(), 0) {
boolean stillGoing = true;
@Override public boolean tryAdvance(Consumer<? super T> consumer) {
if (stillGoing) {
boolean hadNext = splitr.tryAdvance(elem -> {
if (predicate.test(elem)) {
consumer.accept(elem);
} else {
stillGoing = false;
}
});
return hadNext && stillGoing;
}
return false;
}
};
}
static <T> Stream<T> takeWhile(Stream<T> stream, Predicate<? super T> predicate) {
return StreamSupport.stream(takeWhile(stream.spliterator(), predicate), false);
}
答案 2 :(得分：50)
allMatch()是一个短路功能，因此您可以使用它来停止处理。主要的缺点是你必须进行两次测试：一次是看你是否应该进行测试，再看看是否继续进行。
IntStream
.iterate(1, n -> n + 1)
.peek(n->{if (n<10) System.out.println(n);})
.allMatch(n->n < 10);
答案 3 :(得分：32)
作为@StuartMarks answer的后续行动。我的StreamEx库具有takeWhile操作，该操作与当前的JDK-9实现兼容。在JDK-9下运行时，它将委托给JDK实现（通过MethodHandle.invokeExact非常快）。在JDK-8下运行时，＆＃34; polyfill＆＃34;将使用实施。所以使用我的库可以解决这个问题：
IntStreamEx.iterate(1, n -> n + 1)
.takeWhile(n -> n < 10)
.forEach(System.out::println);
答案 4 :(得分：13)
takeWhile是protonpack library提供的功能之一。
Stream<Integer> infiniteInts = Stream.iterate(0, i -> i + 1);
Stream<Integer> finiteInts = StreamUtils.takeWhile(infiniteInts, i -> i < 10);
assertThat(finiteInts.collect(Collectors.toList()),
hasSize(10));
答案 5 :(得分：8)
更新：Java 9 Stream现在附带takeWhile方法。
无需黑客或其他解决方案。只需使用它！
我相信这可以大大改善： （有人可能会让它成为线程安全的）
Stream<Integer> stream = Stream.iterate(0, n -> n + 1);
TakeWhile.stream(stream, n -> n < 10000)
.forEach(n -> System.out.print((n == 0 ? "" + n : "," + n)));
class TakeWhile<T> implements Iterator<T> {
private final Iterator<T> iterator;
private final Predicate<T> predicate;
private volatile T next;
private volatile boolean keepGoing = true;
public TakeWhile(Stream<T> s, Predicate<T> p) {
this.iterator = s.iterator();
this.predicate = p;
}
@Override
public boolean hasNext() {
if (!keepGoing) {
return false;
}
if (next != null) {
return true;
}
if (iterator.hasNext()) {
next = iterator.next();
keepGoing = predicate.test(next);
if (!keepGoing) {
next = null;
}
}
return next != null;
}
@Override
public T next() {
if (next == null) {
if (!hasNext()) {
throw new NoSuchElementException("Sorry. Nothing for you.");
}
}
T temp = next;
next = null;
return temp;
}
public static <T> Stream<T> stream(Stream<T> s, Predicate<T> p) {
TakeWhile tw = new TakeWhile(s, p);
Spliterator split = Spliterators.spliterator(tw, Integer.MAX_VALUE, Spliterator.ORDERED);
return StreamSupport.stream(split, false);
}
}
答案 6 :(得分：8)
您可以使用java8 + rxjava。
import java.util.stream.IntStream;
import rx.Observable;
// Example 1)
IntStream intStream = IntStream.iterate(1, n -> n + 1);
Observable.from(() -> intStream.iterator())
.takeWhile(n ->
{
System.out.println(n);
return n < 10;
}
).subscribe() ;
// Example 2
IntStream intStream = IntStream.iterate(1, n -> n + 1);
Observable.from(() -> intStream.iterator())
.takeWhile(n -> n < 10)
.forEach( n -> System.out.println(n));
答案 7 :(得分：5)
实际上，在Java 8中有两种方法可以在没有任何额外库或使用Java 9的情况下完成它。
如果要在控制台上打印2到20的数字，可以执行以下操作：
IntStream.iterate(2, (i) -> i + 2).peek(System.out::println).allMatch(i -> i < 20);
或
IntStream.iterate(2, (i) -> i + 2).peek(System.out::println).anyMatch(i -> i >= 20);
输出在两种情况下都是：
2
4
6
8
10
12
14
16
18
20
还没有人提到 anyMatch 。这就是这篇文章的原因。
答案 8 :(得分：5)
这是从JDK 9 java.util.stream.Stream.takeWhile（Predicate）复制的源代码。与JDK 8一起使用有点不同。
static <T> Stream<T> takeWhile(Stream<T> stream, Predicate<? super T> p) {
class Taking extends Spliterators.AbstractSpliterator<T> implements Consumer<T> {
private static final int CANCEL_CHECK_COUNT = 63;
private final Spliterator<T> s;
private int count;
private T t;
private final AtomicBoolean cancel = new AtomicBoolean();
private boolean takeOrDrop = true;
Taking(Spliterator<T> s) {
super(s.estimateSize(), s.characteristics() & ~(Spliterator.SIZED | Spliterator.SUBSIZED));
this.s = s;
}
@Override
public boolean tryAdvance(Consumer<? super T> action) {
boolean test = true;
if (takeOrDrop && // If can take
(count != 0 || !cancel.get()) && // and if not cancelled
s.tryAdvance(this) && // and if advanced one element
(test = p.test(t))) { // and test on element passes
action.accept(t); // then accept element
return true;
} else {
// Taking is finished
takeOrDrop = false;
// Cancel all further traversal and splitting operations
// only if test of element failed (short-circuited)
if (!test)
cancel.set(true);
return false;
}
}
@Override
public Comparator<? super T> getComparator() {
return s.getComparator();
}
@Override
public void accept(T t) {
count = (count + 1) & CANCEL_CHECK_COUNT;
this.t = t;
}
@Override
public Spliterator<T> trySplit() {
return null;
}
}
return StreamSupport.stream(new Taking(stream.spliterator()), stream.isParallel()).onClose(stream::close);
}
答案 9 :(得分：4)
这是一个关于整数的版本 - 正如问题所述。
用法：
StreamUtil.takeWhile(IntStream.iterate(1, n -> n + 1), n -> n < 10);
这是StreamUtil的代码：
import java.util.PrimitiveIterator;
import java.util.Spliterators;
import java.util.function.IntConsumer;
import java.util.function.IntPredicate;
import java.util.stream.IntStream;
import java.util.stream.StreamSupport;
public class StreamUtil
{
public static IntStream takeWhile(IntStream stream, IntPredicate predicate)
{
return StreamSupport.intStream(new PredicateIntSpliterator(stream, predicate), false);
}
private static class PredicateIntSpliterator extends Spliterators.AbstractIntSpliterator
{
private final PrimitiveIterator.OfInt iterator;
private final IntPredicate predicate;
public PredicateIntSpliterator(IntStream stream, IntPredicate predicate)
{
super(Long.MAX_VALUE, IMMUTABLE);
this.iterator = stream.iterator();
this.predicate = predicate;
}
@Override
public boolean tryAdvance(IntConsumer action)
{
if (iterator.hasNext()) {
int value = iterator.nextInt();
if (predicate.test(value)) {
action.accept(value);
return true;
}
}
return false;
}
}
}
答案 10 :(得分：1)
去图书馆AbacusUtil。它提供了您想要的精确API以及更多：
IntStream.iterate(1, n -> n + 1).takeWhile(n -> n < 10).forEach(System.out::println);
声明：我是AbacusUtil的开发者。
答案 11 :(得分：1)
如果您知道将执行的重新排列的确切数量，则可以进行
IntStream
.iterate(1, n -> n + 1)
.limit(10)
.forEach(System.out::println);
答案 12 :(得分：0)
除了短路终端操作之外，您不能中止流，这会使某些流值未经处理而无论其值如何。但是，如果您只是想避免对流进行操作，可以添加转换并过滤到流：
import java.util.Objects;
class ThingProcessor
{
static Thing returnNullOnCondition(Thing thing)
{ return( (*** is condition met ***)? null : thing); }
void processThings(Collection<Thing> thingsCollection)
{
thingsCollection.stream()
*** regular stream processing ***
.map(ThingProcessor::returnNullOnCondition)
.filter(Objects::nonNull)
*** continue stream processing ***
}
} // class ThingProcessor
当事物满足某些条件时，将事物流转换为空值，然后过滤掉空值。如果你愿意沉迷于副作用，你可以在遇到某些事情时将条件值设置为true，这样所有后续事物都会被过滤掉而不管它们的值如何。但即使不是，您也可以通过从您不想处理的流中过滤值来节省大量（如果不是全部）处理。
答案 13 :(得分：0)
即使我有类似的要求 - 调用Web服务，如果失败，重试3次。如果在经过这么多次试验后仍然失败，请发送电子邮件通知。经过谷歌搜索后，anyMatch()成了救世主。我的示例代码如下。在以下示例中，如果 webServiceCall 方法在第一次迭代中返回true，则流不会进一步迭代，因为我们调用了anyMatch()。我相信，这就是你要找的东西。
import java.util.stream.IntStream;
import io.netty.util.internal.ThreadLocalRandom;
class TrialStreamMatch {
public static void main(String[] args) {
if(!IntStream.range(1,3).anyMatch(integ -> webServiceCall(integ))){
//Code for sending email notifications
}
}
public static boolean webServiceCall(int i){
//For time being, I have written a code for generating boolean randomly
//This whole piece needs to be replaced by actual web-service client code
boolean bool = ThreadLocalRandom.current().nextBoolean();
System.out.println("Iteration index :: "+i+" bool :: "+bool);
//Return success status -- true or false
return bool;
}
答案 14 :(得分：0)
IntStream.iterate(1, n -> n + 1)
.peek(System.out::println) //it will be executed 9 times
.filter(n->n>=9)
.findAny();
您可以使用mapToObj而不是峰值来返回最终对象或消息
IntStream.iterate(1, n -> n + 1)
.mapToObj(n->{ //it will be executed 9 times
if(n<9)
return "";
return "Loop repeats " + n + " times";});
.filter(message->!message.isEmpty())
.findAny()
.ifPresent(System.out::println);
答案 15 :(得分：-1)
如果你有不同的问题，可能需要不同的解决方案，但对于你当前的问题，我会简单地说：
IntStream
.iterate(1, n -> n + 1)
.limit(10)
.forEach(System.out::println);
答案 16 :(得分：-2)
这是我尝试使用Java Stream库。
IntStream.iterate(0, i -> i + 1)
.filter(n -> {
if (n < 10) {
System.out.println(n);
return false;
} else {
return true;
}
})
.findAny();
答案 17 :(得分：-2)
我实施了另一个快速解决方案（实际上这是不干净的，但你明白了）：
public static void main(String[] args) {
System.out.println(StreamUtil.iterate(1, o -> o + 1).terminateOn(15)
.map(o -> o.toString()).collect(Collectors.joining(", ")));
}
static interface TerminatedStream<T> {
Stream<T> terminateOn(T e);
}
static class StreamUtil {
static <T> TerminatedStream<T> iterate(T seed, UnaryOperator<T> op) {
return new TerminatedStream<T>() {
public Stream<T> terminateOn(T e) {
Builder<T> builder = Stream.<T> builder().add(seed);
T current = seed;
while (!current.equals(e)) {
current = op.apply(current);
builder.add(current);
}
return builder.build();
}
};
}
}
答案 18 :(得分：-2)
可能有点偏离主题，但这是我们为List<T>而不是Stream<T>所拥有的。
首先，您需要一个take util方法。此方法采用第一个n元素：
static <T> List<T> take(List<T> l, int n) {
if (n <= 0) {
return newArrayList();
} else {
int takeTo = Math.min(Math.max(n, 0), l.size());
return l.subList(0, takeTo);
}
}
它就像scala.List.take
assertEquals(newArrayList(1, 2, 3), take(newArrayList(1, 2, 3, 4, 5), 3));
assertEquals(newArrayList(1, 2, 3), take(newArrayList(1, 2, 3), 5));
assertEquals(newArrayList(), take(newArrayList(1, 2, 3), -1));
assertEquals(newArrayList(), take(newArrayList(1, 2, 3), 0));
现在，基于takeWhile编写take方法非常简单
static <T> List<T> takeWhile(List<T> l, Predicate<T> p) {
return l.stream().
filter(p.negate()).findFirst(). // find first element when p is false
map(l::indexOf). // find the index of that element
map(i -> take(l, i)). // take up to the index
orElse(l); // return full list if p is true for all elements
}
它是这样的：
assertEquals(newArrayList(1, 2, 3), takeWhile(newArrayList(1, 2, 3, 4, 3, 2, 1), i -> i < 4));
此实现会部分迭代列表几次，但不会添加添加O(n^2)操作。希望可以接受。