From 01c57a7a2e9e0c586696dafbb17d0d6aa716df67 Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Sat, 26 Mar 2016 09:41:47 +0000
Subject: Implement eagerMap and eagerFlatMap using streams

---
 src/main/java/org/zwobble/mammoth/internal/util/MammothLists.java | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

(limited to 'src')

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 aa45ebf..3519cd2 100644
--- a/src/main/java/org/zwobble/mammoth/internal/util/MammothLists.java
+++ b/src/main/java/org/zwobble/mammoth/internal/util/MammothLists.java
@@ -1,11 +1,11 @@
 package org.zwobble.mammoth.internal.util;
 
-import com.google.common.base.Function;
 import com.google.common.collect.ImmutableList;
 import com.google.common.collect.Iterables;
 import com.google.common.collect.Ordering;
 
 import java.util.List;
+import java.util.function.Function;
 import java.util.stream.Collectors;
 
 import static org.zwobble.mammoth.internal.util.MammothIterables.stream;
@@ -36,11 +36,13 @@ public class MammothLists {
     }
 
     public static <T, R> List<R> eagerMap(Iterable<T> iterable, Function<T, R> function) {
-        return ImmutableList.copyOf(Iterables.transform(iterable, function));
+        return stream(iterable).map(function).collect(Collectors.toList());
     }
 
     public static <T, R> List<R> eagerFlatMap(Iterable<T> iterable, Function<T, Iterable<R>> function) {
-        return ImmutableList.copyOf(Iterables.concat(Iterables.transform(iterable, function)));
+        return stream(iterable)
+            .flatMap(element -> stream(function.apply(element)))
+            .collect(Collectors.toList());
     }
 
     public static <T, R extends Comparable<R>> List<T> orderedBy(Iterable<T> iterable, Function<T, R> getKey) {
-- 
cgit v1.2.3