SwiftUI如何使用Firestore completionHandler返回的数据

现有代码问题说明 #

你编写的DBHandler类中getArtistProfileCardData方法的Firestore查询、文档解析逻辑本身可正常运行，不需要整体调整数据库管理方案，编译错误和数据无法绑定的问题完全是SwiftUI视图层的写法错误导致的：

• SwiftUI的View是结构体类型，不允许在属性声明的顶层位置直接写执行语句，所有异步请求、副作用逻辑必须通过视图生命周期修饰符（如.onAppear、.task）触发。
• 异步回调中拿到的数据只需要在主线程赋值给@State修饰的状态属性，即可自动触发UI重新渲染。

原有问题代码 #

DBHandler 实现（逻辑正常无需修改，可按需优化） #

func getArtistProfileCardData(artistName: String, completionHandler: @escaping (UserModel) -> ()){
    
    let usersRef = db.collection("users")
    usersRef.whereField("artistName", isEqualTo: artistName)
        .getDocuments { (querySnapshot, err) in
            if let err = err {
                        print("Error getting documents: \(err)")
                    } else {
                        for document in querySnapshot!.documents {
                            print("\(document.documentID) => \(document.data())")
                            do {
                                let userData = try document.data(as: UserModel.self)
                                completionHandler(userData)
                                // 此处打印retData.artistName可得到正确输出
                            } catch _ {
                                print("Error getting document from querySnapshot.")
                            }
                        }
                    }
        }
    }

报错的视图代码 #

struct ArtistProfileCard: View {
    @State var user: UserModel
    
    let dbhandler = DBHandler() 
    // 错误：结构体属性区不能直接写执行语句
    dbhandler.getArtistProfileCardData(artistName: "something") { userData in
        // 无法在此处直接将userData绑定到@State属性
    }
        
    var body: some View {
        VStack(alignment: .leading) {
            Spacer()
            Text(user.artistName ?? "??")
                .font(.title)
                .bold()
                .textCase(.uppercase)
            Text(user.occupation)
            Text("\(user.followers) Followers")
        }
        // 其余常规UI逻辑
    }
}

修正后的视图代码 #

struct ArtistProfileCard: View {
    // 给@State属性设置默认初始值，避免结构体初始化报错；如果需要加载态也可以改成可选类型UserModel?
    @State var user: UserModel = UserModel()
    let dbhandler = DBHandler() 
    // 支持从外部传入要查询的艺术家名称
    let artistName: String
        
    var body: some View {
        VStack(alignment: .leading) {
            Spacer()
            Text(user.artistName ?? "??")
                .font(.title)
                .bold()
                .textCase(.uppercase)
            Text(user.occupation ?? "暂无信息")
            Text("\(user.followers ?? 0) Followers")
        }
        .onAppear {
            // 视图首次加载时触发数据请求
            dbhandler.getArtistProfileCardData(artistName: artistName) { userData in
                // UI更新必须在主线程执行，避免未定义行为
                DispatchQueue.main.async {
                    self.user = userData
                }
            }
        }
        // 其余UI逻辑
    }
}

可选优化点 #

• 现有getArtistProfileCardData方法没有向外抛出错误，建议将回调签名改为completionHandler: @escaping (Result<UserModel, Error>) -> Void，把查询错误、解析错误传递给视图层，方便实现加载失败、空数据的兜底UI。
• 当前查询逻辑如果匹配到多条同名艺术家文档，会多次触发回调重复赋值，建议给查询加上.limit(to: 1)，只取第一条匹配结果。
• 如果需要实时监听用户数据变化，可以直接使用Firestore官方提供的SwiftUI扩展中的@FirestoreQuery属性包装器，无需手动编写回调逻辑，数据变更会自动驱动UI更新。

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