Firestore技术问题：获取文档完整子对象而非FIRDocumentReference

嘿，这个需求我太熟悉了！Firestore默认不会自动帮你解析嵌套的FIRDocumentReference（Swift里对应DocumentReference）字段，得我们手动处理或者封装逻辑来关联获取子文档数据。下面给你几种实用的方案：

方案一：手动解析单个引用字段 #

如果你的文档里只有少量引用字段（比如示例里的item），直接针对每个引用调用getDocument获取数据，再替换原数据里的引用就行：

// 假设你已经获取到主文档的snapshot
if let data = snapshot.data(),
   let itemRef = data["item"] as? DocumentReference {
    // 获取子文档数据
    itemRef.getDocument { (itemSnapshot, error) in
        if let error = error {
            print("获取子文档失败：\(error.localizedDescription)")
            return
        }
        if let itemData = itemSnapshot?.data() {
            // 组合主文档数据和子文档数据
            var fullData = data
            fullData["item"] = itemData
            // 现在fullData就是你期望的结果啦
            print(fullData)
            // 输出示例：{ amount = 1; weight = 70; item = { id = 12; name = "QWE" }; }
        }
    }
}

方案二：批量获取多个引用字段 #

如果文档里有多个DocumentReference字段，一个个单独请求效率太低，可以用批量获取API一次性拉取所有关联文档：

if let data = snapshot.data() {
    // 先收集所有的DocumentReference字段
    let allRefs = data.values.compactMap { $0 as? DocumentReference }
    
    guard !allRefs.isEmpty else {
        // 没有引用字段，直接返回原数据
        print(data)
        return
    }
    
    // 批量获取所有子文档
    Firestore.firestore().getDocuments(from: allRefs) { (snapshots, error) in
        if let error = error {
            print("批量获取子文档失败：\(error.localizedDescription)")
            return
        }
        
        // 构建「引用 -> 子文档数据」的映射表
        let refDataMap = snapshots?.documents.reduce(into: [DocumentReference: [String: Any]]()) { result, docSnapshot in
            result[docSnapshot.reference] = docSnapshot.data()
        } ?? [:]
        
        // 替换原数据里的所有引用为对应子文档数据
        var fullData = data
        for (key, value) in fullData {
            if let ref = value as? DocumentReference, let resolvedData = refDataMap[ref] {
                fullData[key] = resolvedData
            }
        }
        
        // 处理完成的完整数据
        print(fullData)
    }
}

方案三：封装可复用的工具函数 #

如果经常需要处理这种场景，可以给DocumentSnapshot扩展一个工具方法，自动帮你解析所有引用字段：

import FirebaseFirestore

extension DocumentSnapshot {
    /// 解析文档中所有的DocumentReference字段，返回包含完整子对象的数据
    func dataWithResolvedReferences(completion: @escaping ([String: Any]?, Error?) -> Void) {
        guard let originalData = self.data() else {
            completion(nil, nil)
            return
        }
        
        // 收集所有引用字段
        let refs = originalData.values.compactMap { $0 as? DocumentReference }
        guard !refs.isEmpty else {
            completion(originalData, nil)
            return
        }
        
        // 批量获取子文档
        Firestore.firestore().getDocuments(from: refs) { (snapshots, error) in
            if let error = error {
                completion(nil, error)
                return
            }
            
            // 构建引用映射
            let refMap = snapshots?.documents.reduce(into: [DocumentReference: [String: Any]]()) { map, doc in
                map[doc.reference] = doc.data()
            } ?? [:]
            
            // 替换引用为子数据
            var resolvedData = originalData
            for (key, value) in resolvedData {
                if let ref = value as? DocumentReference, let subData = refMap[ref] {
                    resolvedData[key] = subData
                }
            }
            
            completion(resolvedData, nil)
        }
    }
}

使用的时候就超级简洁：

// 获取主文档后直接调用扩展方法
yourDocumentSnapshot.dataWithResolvedReferences { (fullData, error) in
    if let fullData = fullData {
        print(fullData) // 直接拿到包含完整子对象的数据
    } else if let error = error {
        print("解析失败：\(error)")
    }
}

额外说明 #

如果你的子文档里还嵌套了DocumentReference，上面的方法只会解析第一层。如果需要递归解析所有层级的引用，可以修改工具函数，加入递归逻辑遍历子数据里的引用字段，不过大部分业务场景下解析第一层就足够啦。

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