SwiftUI聊天应用发送消息后ChatLogView自动返回主视图问题求助

问题根源 #

调用persistRecentMessage()更新Firebase的recent_messages集合后，MainMessagesViewModel的快照监听器会触发recentMessages数组更新。由于RecentMessage未正确实现稳定的标识逻辑，MessagesView中的ForEach会重新渲染所有列表项，导致当前活跃的NavigationLink（指向ChatLogView）被销毁，最终自动退回主视图。

具体修复步骤 #

1. 为RecentMessage实现稳定的Identifiable协议 #

使用Firebase文档的documentId作为唯一且稳定的id，确保SwiftUI能准确识别元素变化：

struct RecentMessage: Identifiable {
    let id: String
    let documentId: String
    let text: String
    let email: String
    let profilePictureUrl: String
    let timestamp: Timestamp
    
    init(documentId: String, data: [String: Any]) {
        self.id = documentId // 用Firebase文档ID作为Identifiable的核心标识
        self.documentId = documentId
        self.text = data["text"] as? String ?? ""
        self.email = data["email"] as? String ?? ""
        self.profilePictureUrl = data["profilePicURL"] as? String ?? ""
        self.timestamp = data["timestamp"] as? Timestamp ?? Timestamp()
    }
}

2. 优化recentMessages的更新逻辑 #

避免不必要的数组删除/插入操作，直接更新已有元素，减少视图重建范围：

private func fetchRecentMessages() {
    guard let uid = FirebaseManager.shared.auth.currentUser?.uid else {return}
    
    FirebaseManager.shared.firestore
        .collection("recent_messages")
        .document(uid)
        .collection("messages")
        .addSnapshotListener { querySnapshot, err in
                
            if let err = err {
                print(err)
                return
            }
            
            DispatchQueue.main.async { // 确保在主线程更新@Published变量
                querySnapshot?.documentChanges.forEach({ change in
                    
                    let docId = change.document.documentID
                    let data  = change.document.data()
                    let updatedMessage = RecentMessage(documentId: docId, data: data)
                    
                    if let index = self.recentMessages.firstIndex(where: { $0.id == docId }) {
                        // 已有消息直接更新，而非删除再插入
                        self.recentMessages[index] = updatedMessage
                    } else {
                        self.recentMessages.insert(updatedMessage, at: 0)
                    }
                })
            }
        }
}

3. 修复MessagesView的NavigationLink目标 #

替换占位的Text("")为实际的ChatLogView，并确保传递正确的对话方数据：

struct MessagesView: View {
    
    @EnvironmentObject var vm : MainMessagesViewModel
    
    var body: some View{
        ScrollView {
            ForEach (vm.recentMessages) { recentMessage in
                VStack {
                    NavigationLink {
                        // 根据最近消息数据初始化ChatLogView
                        ChatLogView(interlocutor: User(
                            uid: recentMessage.toId,
                            email: recentMessage.email,
                            profilePictureUrl: recentMessage.profilePictureUrl
                        ))
                    } label: {
                        HStack {
                            WebImage(url: URL(string:recentMessage.profilePictureUrl))
                                .resizable()
                                .scaledToFill()
                                .frame(width: 64, height: 64)
                                .clipShape(Circle())
                                .padding(.horizontal)
                            
                            VStack (alignment: .leading) {
                                Text(recentMessage.email)
                                    .font(.headline)
                                Text(recentMessage.text)
                                    .font(.subheadline)
                                    .foregroundColor(.gray)
                            }
                            
                            Spacer()
                            
                            // 格式化显示消息时间
                            Text(recentMessage.timestamp.dateValue(), style: .time)
                                .font(.caption)
                                .foregroundColor(.gray)
                        }
                        .padding(.horizontal)
                        Divider()
                    }
                }
            }
        }
        .navigationBarTitle(Text("ჩატები"), displayMode: .inline)
    }
}

4. 确保User结构体也实现Identifiable #

struct User: Identifiable {
    let id: String
    let uid: String
    let email: String
    let profilePictureUrl: String
    
    init(uid: String, email: String, profilePictureUrl: String) {
        self.id = uid
        self.uid = uid
        self.email = email
        self.profilePictureUrl = profilePictureUrl
    }
    
    init(data: [String: Any]) {
        self.id = data["uid"] as? String ?? ""
        self.uid = data["uid"] as? String ?? ""
        self.email = data["email"] as? String ?? ""
        self.profilePictureUrl = data["profilePictureUrl"] as? String ?? ""
    }
}

修复原理 #

1. 稳定标识：用Firebase文档ID作为元素唯一标识，SwiftUI能精准识别数组元素的更新/插入操作，避免全量重建列表。
2. 最小化视图更新：直接修改数组中已有元素，而非删除再插入，减少视图层级的重建范围，避免活跃的NavigationLink被意外销毁。
3. 主线程更新：确保@Published变量的修改在主线程执行，符合SwiftUI的线程安全要求。

修改后，发送消息时Recent Messages视图会正常更新，同时ChatLogView不会自动退出。

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