From 61ca0aede839a9dda4fedf3a61d83d86de517af8 Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Sat, 21 Nov 2015 18:01:59 +0000
Subject: Allow single compilation unit to generate multiple classes

---
 src/main/java/org/zwobble/couscous/util/ExtraLists.java | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

(limited to 'src/main/java/org/zwobble/couscous/util')

diff --git a/src/main/java/org/zwobble/couscous/util/ExtraLists.java b/src/main/java/org/zwobble/couscous/util/ExtraLists.java
index 8c282e8..1951868 100644
--- a/src/main/java/org/zwobble/couscous/util/ExtraLists.java
+++ b/src/main/java/org/zwobble/couscous/util/ExtraLists.java
@@ -20,14 +20,14 @@ public class ExtraLists {
     }
 
     @SuppressWarnings("unchecked")
-    public static <T, R, E extends Exception> List<R> map(
+    public static <T, R, E extends Exception> List<R> flatMap(
             Stream<T> stream,
-            CheckedFunction<T, R, E> function) throws E {
+            CheckedFunction<T, Iterable<R>, E> function) throws E {
         try {
             return stream
-                .map(value -> {
+                .flatMap(value -> {
                     try {
-                        return function.apply(value);
+                        return StreamSupport.stream(function.apply(value).spliterator(), false);
                     } catch (Exception exception) {
                         throw new StashedException(exception);
                     }
-- 
cgit v1.2.3