aQute.bnd.stream.TakeWhile Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of biz.aQute.bnd.util Show documentation

Util classes

The newest version!

package aQute.bnd.stream;

import java.util.Spliterator;
import java.util.function.Consumer;
import java.util.function.Predicate;
import java.util.stream.Stream;
import java.util.stream.StreamSupport;

public final class TakeWhile extends AbstractWhile {
	public static  Stream takeWhile(Stream stream, Predicate predicate) {
		return StreamSupport.stream(new TakeWhile<>(stream.spliterator(), predicate), stream.isParallel())
			.onClose(stream::close);
	}

	private boolean take = true;

	private TakeWhile(Spliterator spliterator, Predicate predicate) {
		super(spliterator, predicate);
	}

	@Override
	public void forEachRemaining(Consumer action) {
		if (take) {
			while (spliterator.tryAdvance(this) && predicate.test(item)) {
				action.accept(item);
			}
			take = false;
		}
	}

	@Override
	public boolean tryAdvance(Consumer action) {
		if (take) {
			if (spliterator.tryAdvance(this) && predicate.test(item)) {
				action.accept(item);
				return true;
			}
			take = false;
		}
		return false;
	}
}