Update Task.swift

fmo91 · Feb 8, 2017 · dcf0e94 · dcf0e94
1 parent de8f85c
commit dcf0e94
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/FOTask/Classes/Task.swift b/FOTask/Classes/Task.swift
@@ -109,3 +109,20 @@ public extension Task {
         }
     }
 }
+
+public extension Task {
+    public func map<C>(_ f: @escaping (B) -> C) -> Task<A, C> {
+        return BasicTask<A, C> { (input: A) in
+            return { (onSuccess: @escaping (C) -> Void, onError: @escaping (Error) -> Void) in
+                self.perform(input,
+                    onSuccess: { (output: B) in
+                        onSuccess(f(output))
+                    },
+                    onError: { (error: Error) in
+                        onError(error)
+                    }
+                )
+            }
+        }
+    }
+}