Swift结合Firebase实现聊天首页显示最后发送消息

我看你已经在尝试获取聊天记录并展示最后一条消息了，目前的代码已经走到了遍历每个聊天对象的步骤，接下来只需要针对每个对话节点查询最新的一条消息就行。我帮你完善代码并解释关键点：

完整实现代码 #

首先我们可以定义一个模型来存储聊天项的信息，这样后续处理和展示会更清晰：

// 定义聊天项模型，包含必要信息
struct ChatItem {
    let userId: String
    let lastMessage: String
    let timestamp: TimeInterval
    // 可按需扩展：用户名、头像URL、未读消息数等字段
}

func getAllMsg() {
    self.users = []
    // 先校验当前用户是否登录，避免强制解包崩溃
    guard let fromId = Auth.auth().currentUser?.uid else {
        print("用户未登录，无法获取聊天记录")
        return
    }
    
    let chatRef = Database.database().reference().child("privateMessages").child(fromId)
    
    chatRef.observeSingleEvent(of: .value) { [weak self] snapshot in
        guard let self = self else { return }
        
        // 遍历当前用户的所有聊天对象
        for userSnap in snapshot.children.allObjects as! [DataSnapshot] {
            let chatUserID = userSnap.key
            
            // 查询该聊天对象下的**最后一条消息**，用queryLimited(toLast:1)只拉取最新一条，提升性能
            let lastMessageRef = Database.database().reference()
                .child("privateMessages")
                .child(fromId)
                .child(chatUserID)
                .queryLimited(toLast: 1)
            
            lastMessageRef.observeSingleEvent(of: .value) { messageSnapshot in
                // 解析最后一条消息的数据
                guard let messageSnap = messageSnapshot.children.allObjects.first as? DataSnapshot,
                      let messageData = messageSnap.value as? [String: Any],
                      let lastMessage = messageData["message"] as? String,
                      let timestamp = messageData["timestamp"] as? TimeInterval else {
                    // 处理消息为空或格式异常的情况
                    return
                }
                
                // 创建聊天项对象
                let chatItem = ChatItem(userId: chatUserID, lastMessage: lastMessage, timestamp: timestamp)
                
                // 必须在主线程更新UI，Firebase回调默认在后台线程
                DispatchQueue.main.async {
                    self.users.append(chatItem)
                    // 按消息时间戳降序排序，让最新的聊天显示在最顶部
                    self.users.sort { $0.timestamp > $1.timestamp }
                    // 刷新列表视图（假设你用UITableView展示）
                    self.tableView.reloadData()
                }
            }
        }
    }
}

关键注意事项 #

• 线程安全：Firebase的数据库回调是在后台线程执行的，所以更新UI（比如刷新表格）一定要用DispatchQueue.main.async切换到主线程，否则会导致崩溃
• 性能优化：使用queryLimited(toLast: 1)只获取每个对话的最后一条消息，而不是拉取全部聊天记录，大幅减少数据传输量
• 数据排序：获取到所有聊天项后，按timestamp降序排序，确保最新的聊天会话显示在首页最上方
• 错误处理：添加了用户未登录、消息数据异常的校验，避免强制解包导致的崩溃
• 模型扩展：如果需要展示聊天对象的用户名、头像，可以在获取到chatUserID后，额外查询Firebase的users节点获取对应信息，补充到ChatItem中

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