FirebaseAuth状态监听中如何结合自定义集合获取完整用户数据

1. 优化自定义User的初始化逻辑 #

首先建议把基于whereField的查询改成直接通过文档ID获取——如果你的users_v5集合是用Firebase用户的uid作为文档ID存储的（这是Firestore存储用户数据的最佳实践），这样查询更高效，也避免了查询结果为空的数组越界问题：

init(from firebaseUser: FirebaseAuth.User) async throws {
    let userDocRef = Firestore.firestore().collection("users_v5").document(firebaseUser.uid)
    // 假设你用了Firestore的Codable扩展（比如FirebaseFirestoreSwift）
    let userData = try await userDocRef.getDocument(as: User.self)
    
    self.id = userData.id
    self.name = userData.name
    self.email = userData.email
    self.imageURL = userData.imageURL
    self.birthday = userData.birthday
    self.phoneNumber = userData.phoneNumber
    self.usersPosts = userData.usersPosts
    self.eventsAttending = userData.eventsAttending
}

如果你的文档ID不是uid，那保留原来的whereField查询，但要加空值判断，避免数组越界：

init(from firebaseUser: FirebaseAuth.User) async throws {
    let usersReference = Firestore.firestore().collection("users_v5")
    let querySnapshot = try await usersReference.whereField("id", isEqualTo: firebaseUser.uid).getDocuments()
    guard let userData = try querySnapshot.documents.first?.data(as: User.self) else {
        throw NSError(domain: "UserError", code: 1, userInfo: [NSLocalizedDescriptionKey: "用户数据未找到"])
    }
    
    // 赋值逻辑同上
    self.id = userData.id
    self.name = userData.name
    // ...其他字段
}

2. 修改AuthService的认证状态监听 #

Firebase的addStateDidChangeListener是同步回调，不能直接在里面调用异步方法，你需要用Task来包裹异步操作，同时处理可能的错误，更新当前用户状态：

假设你的AuthService是这样的结构：

import FirebaseAuth
import FirebaseFirestoreSwift

class AuthService: ObservableObject {
    @Published var currentUser: User?
    private var authStateListener: AuthStateDidChangeListenerHandle?
    
    init() {
        setupAuthListener()
    }
    
    private func setupAuthListener() {
        authStateListener = Auth.auth().addStateDidChangeListener { [weak self] auth, firebaseUser in
            guard let self = self else { return }
            
            // 清空当前用户（如果登出）
            guard let firebaseUser = firebaseUser else {
                self.currentUser = nil
                return
            }
            
            // 启动异步任务获取完整用户数据
            Task {
                do {
                    let fullUser = try await User(from: firebaseUser)
                    // 回到主线程更新UI相关的Published属性
                    DispatchQueue.main.async {
                        self.currentUser = fullUser
                    }
                } catch {
                    print("获取用户数据失败: \(error.localizedDescription)")
                    // 可根据需求处理错误，比如设置默认用户或提示用户
                    DispatchQueue.main.async {
                        self.currentUser = nil
                    }
                }
            }
        }
    }
    
    // 别忘了在合适的时机移除监听（比如deinit）
    deinit {
        if let listener = authStateListener {
            Auth.auth().removeStateDidChangeListener(listener)
        }
    }
}

关键注意点 #

• 用[weak self]避免循环引用，在回调里捕获self确保安全
• 异步操作完成后必须切回主线程更新@Published属性，因为UI更新必须在主线程执行
• 处理错误情况：比如用户文档不存在、网络错误等，避免崩溃或静默失败
• 如果你用的是UIKit而非SwiftUI，只需把@Published换成合适的回调闭包或通知，更新用户状态的逻辑保持一致

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