Swift使用Firebase Functions时如何避免回调地狱

你熟悉的JavaScript Promise对应Swift生态里的原生并发方案async/await，Firebase 8.0及以上版本的Swift SDK已经原生支持async/await语法，不需要引入第三方库，完全可以替代嵌套回调+DispatchGroup的混乱实现，逻辑和你写JS异步代码的习惯几乎一致，学习成本极低。

完全移除手动管理的DispatchGroup和嵌套回调，代码逻辑线性可读，错误可以统一捕获：

func fetchListData() async {
    var loadedObjects: [NewObject] = []
    do {
        // 第一层查询：拉取documentA集合
        let snapshotA = try await db.collection("documentA").getDocuments()
        
        // 并发处理每个文档的后续查询，自动等待所有任务完成，替代DispatchGroup的enter/leave逻辑
        try await withThrowingTaskGroup(of: NewObject.self) { taskGroup in
            for docA in snapshotA.documents {
                taskGroup.addTask {
                    let dataA = docA.data()["dataA"] as! String
                    // 第二层查询：拉取关联的documentB
                    let docB = try await db.collection("documentB").document(dataA).getDocument()
                    let dataB = docB.data()?["dataB"] as! String
                    // 第三层查询：拉取关联的documentC
                    let docC = try await db.collection("documentC").document(dataB).getDocument()
                    let dataC = docC.data()?["dataC"] as! String
                    return NewObject(dataA, dataB, dataC)
                }
            }
            // 收集所有查询组装完成的对象
            for try await object in taskGroup {
                loadedObjects.append(object)
            }
        }

        // 所有任务执行完成，回到主线程更新UI
        await MainActor.run {
            self.array = loadedObjects
            self.tableView.reloadData()
        }
    } catch {
        // 三层查询的所有错误都可以在这里统一处理，不需要在每层回调里单独判断err
        print("数据加载异常：\(error.localizedDescription)")
    }
}

// 调用时只需要在普通上下文里启动并发任务即可
// Task { await fetchListData() }

• Swift的async/await语法逻辑和JS完全一致：func xxx() async throws对应JS的async function，await关键字用法和JS无差异，throws对应JS Promise的reject状态，通过do/catch统一捕获错误
• Firebase所有带回调的异步API（数据库查询、云函数调用、存储上传等）都提供对应的async版本，不需要自己封装回调转Promise/async的逻辑，直接在原方法调用前加await即可拿到返回值
• 原来用DispatchGroup实现的「等待循环内所有异步任务完成」的需求，用withThrowingTaskGroup即可实现，不需要手动管理enter/leave，从根源上避免漏写leave导致的UI卡死、数组并发读写崩溃的问题
• UI刷新必须回到主线程，用MainActor.run包裹即可，不需要手动写DispatchQueue.main.async或者DispatchGroup的notify逻辑

• 先看Swift官方并发文档的基础章节，核心概念和JS Promise/async await重合度极高，有JS基础的话1-2小时就能掌握核心用法
• 对照Firebase Swift SDK的官方示例，把现有回调逻辑逐段替换成async写法，初期可以先抽离嵌套最深的查询逻辑成独立async方法，不需要一次性重构所有代码
• 改造过程中可以先保留原有回调逻辑做结果对比，确认async版本返回数据和原有逻辑一致后再删除旧代码，避免线上出问题

内容的提问来源于stack exchange，提问作者CichlidGold