Allow single compilation unit to generate multiple classes

1 files changed, 4 insertions, 4 deletions

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);
                     }