集合视图重载后单元格UIMenu关联异常问题求助

问题现象 #

给CollectionView单元格内的UIButton添加UIMenu后，若在UIMenu展示状态下调用collectionView.reloadData()，会出现两个异常：

• UIMenu的位置会跳到其他单元格上
• 点击UIMenu的操作时，执行的是新单元格的数据逻辑，而非触发UIMenu的原单元格数据

比如点击第二个单元格（对应value=1）弹出UIMenu，等待10秒触发reload后点击分享，预期打印Share 1，实际却打印Share 3。

复现步骤 #

• 运行应用
• 点击任意单元格弹出UIMenu
• 等待10秒触发collectionView.reloadData()
• 点击UIMenu中的「分享」操作
• 观察输出结果与预期不符

问题根源 #

CollectionView采用单元格复用机制，reloadData()时系统会回收已展示的单元格，重新分配给新的数据源位置。而UIMenu是绑定在单元格的UIButton上的，单元格被复用时，旧的UIMenu不会自动销毁，会跟着单元格一起被分配到新位置。同时，UIAction的闭包如果捕获的是单元格实例（self），当单元格被复用后，self.value已经被更新为新数据源的值，导致点击操作时输出错误数据。

问题关键代码片段：

UIAction(title: "Share") { [weak self] _ in
    guard let self else { return }
    print("Share \(self.value)")
}

解决方案 #

方案1：捕获数据源值而非单元格实例 #

直接在配置单元格时，让UIAction闭包捕获当前的value常量，而非单元格的self。这样即使单元格被复用，闭包也会保留最初的数值，从根源避免数据错位：

func configure(with value: Int) {
    self.value = value
    let shareAction = UIAction(title: "Share") { _ in
        print("Share \(value)")
    }
    let menu = UIMenu(children: [shareAction])
    button.menu = menu
    button.showsMenuAsPrimaryAction = true
}

方案2：复用单元格前重置UIMenu #

在自定义单元格的prepareForReuse()方法中，移除按钮的UIMenu，确保复用的单元格不会携带旧的UIMenu：

override func prepareForReuse() {
    super.prepareForReuse()
    button.menu = nil
    button.showsMenuAsPrimaryAction = false
}

之后在configure(with value: Int)方法中重新创建并绑定对应数据的UIMenu即可。

方案3：reload前关闭所有UIMenu #

在调用reloadData()前，遍历所有可见单元格，关闭它们的UIMenu，避免UIMenu附着在复用的单元格上：

func safeReloadCollectionView() {
    collectionView.visibleCells.forEach { cell in
        guard let customCell = cell as? CustomCell else { return }
        customCell.button.menu = nil
    }
    collectionView.reloadData()
}

后续所有需要reload的地方都调用这个方法即可。

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