如何在Swift中使用Codable添加Firestore文档并等待操作完成

你的思路方向是对的，但原代码里直接操作Task的value属性是行不通的（它是只读的），而且用SnapshotListener来监听文档存在其实有点绕——FirebaseFirestoreSwift的addDocument(from:)本身就提供了带completion handler的重载，我们可以用Swift的withCheckedThrowingContinuation把它包装成标准的async/await方法，完美满足你等待操作完成后返回DocumentReference的需求。

正确实现代码 #

import Foundation
import FirebaseFirestore
import FirebaseFirestoreSwift

class UserService {
    let firestore = Firestore.firestore()
    
    func createUserAwaiting(user: User) async throws -> DocumentReference {
        return try await withCheckedThrowingContinuation { continuation in
            do {
                // 调用带completion的addDocument重载
                let docRef = try firestore.collection("users").addDocument(from: user) { error in
                    if let error = error {
                        // 操作失败，抛出错误
                        continuation.resume(throwing: error)
                    } else {
                        // 操作成功，返回生成的文档引用
                        continuation.resume(returning: docRef)
                    }
                }
            } catch {
                // 捕获编码失败的错误（比如User对象不符合Encodable要求）
                continuation.resume(throwing: error)
            }
        }
    }
}

代码说明 #

1. 用Continuation转换回调API：withCheckedThrowingContinuation是Swift官方推荐的将基于回调的异步API转换为async/await风格的方式，它帮我们处理了异步流程的挂起和恢复。
2. 处理两种错误场景：
   • 第一种是编码错误：调用addDocument(from:)时，如果User对象无法被正确编码（比如缺少Encodable实现的属性），会直接抛出错误，我们捕获后通过continuation传递出去。
   • 第二种是Firestore操作错误：比如网络问题、权限不足等，这些错误会在completion handler中返回，我们同样通过continuation抛出。
3. 确保操作完成后返回：Firestore的completion handler会在文档成功提交到服务器后触发（离线状态下会在重新联网同步成功后触发），此时返回的DocumentReference已经指向一个有效的Firestore文档。

原代码的问题分析 #

• Task的value属性是只读的，无法直接赋值，所以你写的ta.value = ref根本无法执行。
• 用SnapshotListener监听文档存在是冗余的：addDocument的completion handler已经能准确告知我们操作是否完成，不需要额外监听文档快照。
• 这种方式容易引发内存泄漏风险，如果listener没有被正确移除，或者Task的生命周期管理不当。

内容的提问来源于stack exchange，提问作者Rodrigo González