feat: catch the throwing error and resume it to the continuation

firebase · Jul 27, 2024 · af4e2b8 · af4e2b8
1 parent 6005fe9
commit af4e2b8
Showing 1 changed file with 10 additions and 6 deletions.
diff --git a/Firestore/Swift/Source/Codable/CollectionReference+WriteEncodable.swift b/Firestore/Swift/Source/Codable/CollectionReference+WriteEncodable.swift
@@ -62,13 +62,17 @@ public extension CollectionReference {
     -> DocumentReference {
     return try await withCheckedThrowingContinuation { continuation in
       var document: DocumentReference?
-      document = self.addDocument(from: value, encoder: encoder) { error in
-        if let error {
-          continuation.resume(throwing: error)
-        } else {
-          // Our callbacks guarantee that we either return an error or a document.
-          continuation.resume(returning: document!)
+      do {
+        document = try self.addDocument(from: value, encoder: encoder) { error in
+          if let error {
+            continuation.resume(throwing: error)
+          } else {
+            // Our callbacks guarantee that we either return an error or a document.
+            continuation.resume(returning: document!)
+          }
         }
+      } catch {
+        continuation.resume(throwing: error)
       }
     }
   }