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/StreamUtils.java')

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 <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;
+            }
+        };
+    }
+
+    public static <T> Stream<T> takeWhile(Stream<T> stream, Predicate<? super T> predicate) {
+        return StreamSupport.stream(takeWhile(stream.spliterator(), predicate), false);
+    }
+}
-- 
cgit v1.2.3