Added flatMap for Source too.

skylab-inc · Nov 21, 2016 · 1cc5552 · 1cc5552
1 parent 1866fcb
commit 1cc5552
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/Sources/Source.swift b/Sources/Source.swift
@@ -262,6 +262,11 @@ public extension SourceType {
     public func flatMap<U>(_ transform: @escaping (Value) -> U?) -> Source<U, Error> {
         return lift { $0.flatMap(transform) }
     }
-
+
+    public func flatMap<U>(_ transform: @escaping (Value) -> Source<U, Error>) -> Source<U, Error> {
+        return lift { $0.map(transform).joined() }
+    }
+
+
 }