Swift中Firestore DocumentID的正确返回方式及优化函数实现问询

嘿，我太懂你被嵌套闭包搞得头大的感觉了！Firestore的异步操作如果只用传统闭包写法，很容易变成“金字塔地狱”，频繁发通知也确实不是理想方案。现在Swift有更简洁的异步处理方式，我给你两种常用的优雅实现：

1. 首选：使用async/await（iOS 15+ / macOS 12+） #

这是目前Swift处理异步最直观、最清爽的方案，完全摆脱嵌套闭包的困扰。你可以直接写一个异步函数，用await等待Firestore查询结果，然后返回DocumentID。

获取单个匹配文档的DocumentID #

import FirebaseFirestore

// 异步函数：根据字段值返回第一个匹配文档的DocumentID
func getDocumentID(for fieldValue: String) async throws -> String? {
    let db = Firestore.firestore()
    // 替换成你的集合名和目标字段名
    let query = db.collection("yourCollection")
        .whereField("targetField", isEqualTo: fieldValue)
        .limit(to: 1) // 只取第一个匹配的文档
    
    // await自动等待查询完成，无需嵌套闭包
    let querySnapshot = try await query.getDocuments()
    // 返回第一个文档的ID，无匹配则返回nil
    return querySnapshot.documents.first?.documentID
}

获取多个匹配文档的DocumentID数组 #

如果需要返回所有符合条件的DocumentID，稍微修改即可：

func getAllDocumentIDs(for fieldValue: String) async throws -> [String] {
    let db = Firestore.firestore()
    let query = db.collection("yourCollection")
        .whereField("targetField", isEqualTo: fieldValue)
    
    let querySnapshot = try await query.getDocuments()
    // 把每个文档的ID映射成数组返回
    return querySnapshot.documents.map { $0.documentID }
}

调用方式 #

在需要的地方用Task包裹调用，轻松处理结果和错误：

// 比如在ViewController中调用
Task {
    do {
        if let docID = try await getDocumentID(for: "yourTargetValue") {
            print("找到的DocumentID: \(docID)")
            // 若需更新UI，可切换到主线程
            await MainActor.run {
                self.resultLabel.text = docID
            }
        } else {
            print("没有找到匹配的文档")
        }
    } catch {
        print("查询出错: \(error.localizedDescription)")
    }
}

2. 兼容低版本：使用Combine框架 #

如果你的项目需要支持iOS 15以下版本，Combine框架是另一个不错的选择，通过Publisher传递结果，同样避免嵌套闭包：

实现函数 #

import Combine
import FirebaseFirestore

func getDocumentID(for fieldValue: String) -> AnyPublisher<String?, Error> {
    let db = Firestore.firestore()
    let query = db.collection("yourCollection")
        .whereField("targetField", isEqualTo: fieldValue)
        .limit(to: 1)
    
    return Future { promise in
        query.getDocuments { snapshot, error in
            if let error = error {
                promise(.failure(error))
                return
            }
            let docID = snapshot?.documents.first?.documentID
            promise(.success(docID))
        }
    }
    .eraseToAnyPublisher()
}

调用方式 #

通过订阅Publisher获取结果，记得管理好订阅生命周期：

var cancellables = Set<AnyCancellable>()

getDocumentID(for: "yourTargetValue")
    .receive(on: DispatchQueue.main) // 切换到主线程更新UI
    .sink(
        receiveCompletion: { completion in
            if case .failure(let error) = completion {
                print("查询失败: \(error.localizedDescription)")
            }
        },
        receiveValue: { docID in
            if let docID = docID {
                print("获取到DocumentID: \(docID)")
                self.resultLabel.text = docID
            } else {
                print("无匹配文档")
            }
        }
    )
    .store(in: &cancellables)

为什么这两种方法更好？ #

• 彻底告别嵌套闭包的“金字塔”结构，代码可读性大幅提升
• 无需依赖通知中心，逻辑更集中，维护起来更省心
• async/await的写法几乎和同步代码一样直观，学习成本极低

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