From d9cc4e7060e81a583016b7b69203ab397f3d3478 Mon Sep 17 00:00:00 2001 From: Vitaly Takmazov Date: Wed, 13 Jun 2018 11:58:48 +0300 Subject: fix processing message with empty * tag --- .../src/main/java/com/juick/util/StreamUtils.java | 34 ++++++++++++++++++++++ 1 file changed, 34 insertions(+) create mode 100644 juick-common/src/main/java/com/juick/util/StreamUtils.java (limited to 'juick-common/src/main/java/com/juick/util') diff --git a/juick-common/src/main/java/com/juick/util/StreamUtils.java b/juick-common/src/main/java/com/juick/util/StreamUtils.java new file mode 100644 index 00000000..de19a43e --- /dev/null +++ b/juick-common/src/main/java/com/juick/util/StreamUtils.java @@ -0,0 +1,34 @@ +package com.juick.util; + +import java.util.Spliterator; +import java.util.Spliterators; +import java.util.function.Consumer; +import java.util.function.Predicate; +import java.util.stream.Stream; +import java.util.stream.StreamSupport; + +public class StreamUtils { + private static Spliterator takeWhile( + Spliterator splitr, Predicate predicate) { + return new Spliterators.AbstractSpliterator(splitr.estimateSize(), 0) { + boolean stillGoing = true; + @Override public boolean tryAdvance(Consumer consumer) { + if (stillGoing) { + boolean hadNext = splitr.tryAdvance(elem -> { + if (predicate.test(elem)) { + consumer.accept(elem); + } else { + stillGoing = false; + } + }); + return hadNext && stillGoing; + } + return false; + } + }; + } + + public static Stream takeWhile(Stream stream, Predicate predicate) { + return StreamSupport.stream(takeWhile(stream.spliterator(), predicate), false); + } +} -- cgit v1.2.3