如何计算Firestore所有文档quantity字段总和并显示在标签中？

你的需求完全可以实现，不过现有代码存在几个可以优化的地方：

• 每次遍历文档时都更新标签文本，频繁操作UI没有意义
• 未初始化totalVotes，如果变量之前有旧值会导致求和结果错误
• Firestore的回调在后台线程执行，直接更新UI可能引发线程安全问题

下面是优化后的两种实现方案：

方案一：优化循环累加写法 #

func getDatas2() {
    guard let uid = UID else {
        print("UID为空，无法获取数据")
        return
    }
    
    // 重置总和，避免累计旧值
    var totalVotes = 0
    
    self.db.collection("Users").document(uid).collection("AiBuyPost").getDocuments { snapshot, error in
        if let error = error {
            print("获取数据失败：\(error.localizedDescription)")
            return
        }
        
        guard let documents = snapshot?.documents, !documents.isEmpty else {
            print("没有找到相关文档")
            // 无数据时设置标签为0
            DispatchQueue.main.async {
                self.new.text = "0"
            }
            return
        }
        
        // 遍历所有文档累加quantity值
        for document in documents {
            if let quantity = document.data()["quantity"] as? Int {
                totalVotes += quantity
            }
        }
        
        // 必须在主线程更新UI
        DispatchQueue.main.async {
            self.new.text = String(totalVotes)
        }
    }
}

方案二：使用reduce简化求和 #

利用Swift的reduce方法可以更简洁地完成总和计算，省去手动循环的代码：

func getDatas2() {
    guard let uid = UID else {
        print("UID为空，无法获取数据")
        return
    }
    
    self.db.collection("Users").document(uid).collection("AiBuyPost").getDocuments { snapshot, error in
        if let error = error {
            print("获取数据失败：\(error.localizedDescription)")
            return
        }
        
        guard let documents = snapshot?.documents else {
            DispatchQueue.main.async {
                self.new.text = "0"
            }
            return
        }
        
        // 用reduce直接计算所有quantity的总和
        let totalVotes = documents.reduce(0) { sum, document in
            sum + (document.data()["quantity"] as? Int ?? 0)
        }
        
        DispatchQueue.main.async {
            self.new.text = String(totalVotes)
        }
    }
}

关键优化点说明 #

• 用guard let处理UID为空的情况，避免强制解包UID!引发崩溃
• 每次请求前重置总和变量，确保结果是当前所有文档的真实总和
• 所有UI更新操作放在DispatchQueue.main.async中，遵守iOS的UI线程规则
• 针对无数据的场景做了处理，确保标签显示合理值

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