Lazily combine warnings in results

2 files changed, 37 insertions, 0 deletions

diff --git a/src/main/java/org/zwobble/mammoth/internal/util/MammothIterables.java b/src/main/java/org/zwobble/mammoth/internal/util/MammothIterables.java
index b9fd029..c0231e0 100644
--- a/src/main/java/org/zwobble/mammoth/internal/util/MammothIterables.java
+++ b/src/main/java/org/zwobble/mammoth/internal/util/MammothIterables.java
@@ -6,6 +6,8 @@ import java.util.Iterator;
 import java.util.Optional;
 import java.util.function.Function;
 import java.util.function.Predicate;
+import java.util.stream.Stream;
+import java.util.stream.StreamSupport;
 
 public class MammothIterables {
     public static <T> Optional<T> tryGetLast(Iterable<T> iterable) {
@@ -39,6 +41,30 @@ public class MammothIterables {
         }
     }
 
+    public static <T, R> Iterable<R> lazyFlatMap(Iterable<T> iterable, Function<T, Iterable<R>> function) {
+        return lazyFlatten(lazyMap(iterable, function));
+    }
+
+    public static <T> Iterable<T> lazyConcat(Iterable<T> iterable1, Iterable<T> iterable2) {
+        return new Iterable<T>() {
+            @Override
+            public Iterator<T> iterator() {
+                return Stream.concat(stream(iterable1), stream(iterable2)).iterator();
+            }
+        };
+    }
+
+    public static <T> Iterable<T> lazyFlatten(Iterable<? extends Iterable<T>> iterables) {
+        return new Iterable<T>() {
+            @Override
+            public Iterator<T> iterator() {
+                return stream(iterables)
+                    .flatMap(iterable -> stream(iterable))
+                    .iterator();
+            }
+        };
+    }
+
     private static <T, R> Iterator<R> map(Iterator<T> iterator, Function<T, R> function) {
         return new Iterator<R>() {
             @Override
@@ -52,4 +78,8 @@ public class MammothIterables {
             }
         };
     }
+
+    public static <T> Stream<T> stream(Iterable<T> iterable) {
+        return StreamSupport.stream(iterable.spliterator(), false);
+    }
 }
diff --git a/src/main/java/org/zwobble/mammoth/internal/util/MammothLists.java b/src/main/java/org/zwobble/mammoth/internal/util/MammothLists.java
index eff5411..aa45ebf 100644
--- a/src/main/java/org/zwobble/mammoth/internal/util/MammothLists.java
+++ b/src/main/java/org/zwobble/mammoth/internal/util/MammothLists.java
@@ -6,6 +6,9 @@ import com.google.common.collect.Iterables;
 import com.google.common.collect.Ordering;
 
 import java.util.List;
+import java.util.stream.Collectors;
+
+import static org.zwobble.mammoth.internal.util.MammothIterables.stream;
 
 public class MammothLists {
     public static <T> List<T> list() {
@@ -47,4 +50,8 @@ public class MammothLists {
     private static <T, R extends Comparable<R>> Ordering<T> orderBy(Function<T, R> getKey) {
         return Ordering.from((first, second) -> getKey.apply(first).compareTo(getKey.apply(second)));
     }
+
+    public static <T> List<T> toList(Iterable<T> iterable) {
+        return stream(iterable).collect(Collectors.toList());
+    }
 }