Add flatMap method to __core__.iter

bamless · Jun 25, 2024 · ea2be7d · ea2be7d
1 parent 270933f
commit ea2be7d
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 1 deletion.
diff --git a/include/jstar/conf.h b/include/jstar/conf.h
@@ -12,7 +12,7 @@
     (JSTAR_VERSION_MAJOR * 100000 + JSTAR_VERSION_MINOR * 1000 + JSTAR_VERSION_PATCH)
 
 // compiler and platform on which this J* binary was compiled
-#define JSTAR_COMPILER "GNU 13.2.1"
+#define JSTAR_COMPILER "GNU 14.1.1"
 #define JSTAR_PLATFORM "Linux"
 
 // Options

diff --git a/src/lib/core/iter.jsc b/src/lib/core/iter.jsc
diff --git a/src/lib/core/iter.jsr b/src/lib/core/iter.jsr
@@ -77,6 +77,10 @@ class Iterable
         return map(this, fn)
     end
 
+    fun flatMap(fn)
+        return flatMap(this, fn)
+    end
+
     fun filter(predicate)
         return filter(this, predicate)
     end
@@ -490,6 +494,19 @@ fun map(iterable, fn)
     end
 end
 
+fun flatMap(iterable, fn)
+    for var e in iterable
+        var res = fn(e)
+        if res is Iterable
+            for var e in res
+                yield e
+            end
+        else
+            yield e
+        end
+    end
+end
+
 fun filter(iterable, predicate)
     for var e in iterable
         if predicate(e)