From c6116b434e1f97df34146bf2352b5493630c60da Mon Sep 17 00:00:00 2001
From: Ruslan Sennov <ruslan.sennov@gmail.com>
Date: Tue, 25 Aug 2015 07:31:53 +0300
Subject: [PATCH] Iterator.takeWhile() #487

---
 .../java/javaslang/collection/HashSet.java    |  4 +--
 .../java/javaslang/collection/Iterator.java   | 30 ++++++++++++++++++-
 2 files changed, 31 insertions(+), 3 deletions(-)

diff --git a/src/main/java/javaslang/collection/HashSet.java b/src/main/java/javaslang/collection/HashSet.java
index fd11468399..91880b8d25 100644
--- a/src/main/java/javaslang/collection/HashSet.java
+++ b/src/main/java/javaslang/collection/HashSet.java
@@ -446,8 +446,8 @@ public HashSet<T> takeRight(int n) {
     @Override
     public HashSet<T> takeWhile(Predicate<? super T> predicate) {
         Objects.requireNonNull(predicate, "predicate is null");
-        List<T> taken = list.get().takeWhile(predicate);
-        return taken.length() == list.get().length() ? this : HashSet.ofAll(taken);
+        HashSet<T> taken = HashSet.ofAll(iterator().takeWhile(predicate));
+        return taken.length() == length() ? this : taken;
     }
 
     @Override
diff --git a/src/main/java/javaslang/collection/Iterator.java b/src/main/java/javaslang/collection/Iterator.java
index 259ee92d60..01c436bdbf 100644
--- a/src/main/java/javaslang/collection/Iterator.java
+++ b/src/main/java/javaslang/collection/Iterator.java
@@ -727,7 +727,35 @@ default Iterator<T> takeRight(int n) {
     @Override
     default Iterator<T> takeWhile(Predicate<? super T> predicate) {
         Objects.requireNonNull(predicate, "predicate is null");
-        return null;
+        final Iterator<T> that = this;
+        return new Iterator<T>() {
+
+            private T next = null;
+            private boolean finished = false;
+
+            @Override
+            public boolean hasNext() {
+                while (!finished && next == null && that.hasNext()) {
+                    final T value = that.next();
+                    if (predicate.test(value)) {
+                        next = value;
+                    } else {
+                        finished = true;
+                    }
+                }
+                return next != null;
+            }
+
+            @Override
+            public T next() {
+                if (!hasNext()) {
+                    EMPTY.next();
+                }
+                final T result = next;
+                next = null;
+                return result;
+            }
+        };
     }
 
     default <U> Iterator<Tuple2<T, U>> zip(Iterable<U> that) {