如何在Firestore异步请求返回后显示UITableViewCell中的按钮等元素？

首先，咱们得先搞定最直观的闪烁问题：初始就把按钮设为隐藏——因为从Storyboard加载的Cell会默认遵循IB里的显示状态，所以必须在Cell初始化阶段就把按钮藏起来，彻底避免网络请求返回前的短暂显示。

接下来聊聊Firestore查询和闭包的使用：我不建议把查询逻辑放在MessageCell类里，毕竟TableView的Cell是会复用的，滚动时重复发起请求既浪费资源，还可能导致UI混乱。不过既然你想通过闭包实现，咱们一步步来优化：

步骤1：修改MessageCell，初始化隐藏按钮并优化闭包函数 #

先在Cell的初始化方法里隐藏按钮，同时更新查询函数，让它能接收用户标识（比如用户ID）定位到对应的Firestore文档，并且把查询到的布尔值通过闭包传回，这样才能决定是否显示按钮：

class MessageCell: UITableViewCell {
    @IBOutlet weak var nameCell: UILabel!
    @IBOutlet weak var button: UIButton!
    
    // 保存当前查询任务，Cell复用时可取消旧请求
    private var currentQueryTask: ListenerRegistration?
    
    override func awakeFromNib() {
        super.awakeFromNib()
        // 初始隐藏按钮，从根源避免闪烁
        button.isHidden = true
    }
    
    override func prepareForReuse() {
        super.prepareForReuse()
        // 复用前清理旧查询，防止旧请求返回错误更新UI
        currentQueryTask?.remove()
        currentQueryTask = nil
        // 重置按钮状态
        button.isHidden = true
    }
    
    // 优化函数：接收用户ID，闭包返回是否显示按钮的布尔值
    func checkButtonVisibility(forUserId userId: String, completion: @escaping (Bool) -> Void) {
        // 替换为你的Firestore查询逻辑，比如根据userId获取目标文档
        let targetDoc = Firestore.firestore().collection("users").document(userId)
        
        currentQueryTask = targetDoc.addSnapshotListener { snapshot, error in
            if let error = error {
                print("获取用户数据失败：\(error)")
                completion(false) // 出错默认隐藏按钮
                return
            }
            
            guard let snapshot = snapshot, snapshot.exists else {
                completion(false)
                return
            }
            
            // 替换为你实际的字段名，比如"showButton"
            let shouldShow = snapshot.data()?["showButton"] as? Bool ?? false
            completion(shouldShow)
        }
    }
}

步骤2：在ViewController的cellForRowAt里调用函数并更新UI #

在cellForRowAt中拿到Cell后，调用查询函数，并且在闭包里回到主线程更新按钮状态（Firestore回调默认在后台线程，不能直接操作UI）：

func tableView(_ tableView: UITableView, cellForRowAt indexPath: IndexPath) -> UITableViewCell {
    guard let cell = tableView.dequeueReusableCell(withIdentifier: "customCell", for: indexPath) as? MessageCell else {
        fatalError("取出的Cell不是MessageCell实例")
    }
    
    let user = changeDataSource(indexPath: indexPath as NSIndexPath)
    cell.nameCell.text = user.firstname
    
    // 调用Cell的查询函数，传入用户ID（假设User类有id属性）
    cell.checkButtonVisibility(forUserId: user.id) { [weak cell] shouldShow in
        DispatchQueue.main.async {
            cell?.button.isHidden = !shouldShow
        }
    }
    
    return cell
}

更优方案：预加载所有按钮状态 #

如果你的用户列表数据量不大，更高效的做法是在ViewController中批量获取所有用户的按钮显示状态，把它存在User模型里，这样在cellForRowAt里直接设置即可，完全避免每个Cell发起独立请求：

// 给User模型新增按钮状态字段
struct User {
    let firstname: String
    let id: String
    var shouldShowButton: Bool = false // 新增字段
}

// 在ViewController中批量加载用户及按钮状态
func loadUsers() {
    Firestore.firestore().collection("users").getDocuments { snapshot, error in
        if let error = error {
            print("加载用户列表失败：\(error)")
            return
        }
        
        guard let documents = snapshot?.documents else { return }
        self.users = documents.compactMap { doc in
            let firstname = doc.data()["firstname"] as? String ?? ""
            let id = doc.documentID
            let shouldShowButton = doc.data()["showButton"] as? Bool ?? false
            return User(firstname: firstname, id: id, shouldShowButton: shouldShowButton)
        }
        
        self.tableView.reloadData()
    }
}

// 然后在cellForRowAt里直接设置按钮状态
func tableView(_ tableView: UITableView, cellForRowAt indexPath: IndexPath) -> UITableViewCell {
    guard let cell = tableView.dequeueReusableCell(withIdentifier: "customCell", for: indexPath) as? MessageCell else {
        fatalError("取出的Cell不是MessageCell实例")
    }
    
    let user = users[indexPath.row]
    cell.nameCell.text = user.firstname
    cell.button.isHidden = !user.shouldShowButton
    
    return cell
}

这种方式既解决了闪烁问题，又避免了重复网络请求，性能更优。

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