mapIfDuplicatedKeyError

asensei · Jan 30, 2019 · 4b48a4e · 4b48a4e
1 parent 3c71d2a
commit 4b48a4e
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/Sources/FluentMongo/Model+Index.swift b/Sources/FluentMongo/Model+Index.swift
@@ -18,3 +18,20 @@ extension Model where Database == MongoDatabase {
         return IndexBuilder(on: conn.databaseConnection(to: Self.defaultDatabase))
     }
 }
+
+extension Future where T: Model, T.Database == MongoDatabase {
+
+    @discardableResult
+    public func mapIfDuplicatedKeyError(_ callback: @escaping (MongoConnection.Error) throws -> T) -> Future<T> {
+        return self.catchMap { error in
+            guard
+                let mongoConnectionError = error as? MongoConnection.Error,
+                case .duplicatedKey = mongoConnectionError
+                else {
+                    throw error
+            }
+
+            return try callback(mongoConnectionError)
+        }
+    }
+}