Merge pull request #33 from qhua948/master

Added a new `recover` method to allow return of promises
freshOS · Feb 1, 2018 · ca7648d · ca7648d
2 parents 52373d7 + 24708da
commit ca7648d
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/Source/Promise+Recover.swift b/Source/Promise+Recover.swift
@@ -80,6 +80,26 @@ extension Promise {
         }, progress: p.setProgress)
         return p
     }
+
+    public func recover(_ block:@escaping (Error) throws -> Promise<T>) -> Promise<T> {
+        let p = newLinkedPromise()
+        syncStateWithCallBacks(
+            success: p.fulfill,
+            failure: { e in
+                do {
+                    let promise = try block(e)
+                    promise.then { t in
+                        p.fulfill(t)
+                    }.onError { error in
+                        p.reject(error)
+                    }
+                } catch {
+                    p.reject(error)
+                }
+        }, progress: p.setProgress)
+        return p
+    }
+
 }
 
 // Credits to Quick/Nimble for how to compare Errors