使用NSDiffableDataSourceSnapshot时多分区数据填充异常问题

核心问题：值类型快照的捕获与修改错误 #

你当前代码的核心问题在于**NSDiffableDataSourceSnapshot是值类型**，当你在异步回调中修改外部定义的snapshot变量时，每个闭包捕获的都是该快照的独立拷贝——三个回调里的修改操作完全互不影响，最后在group.notify中应用的还是最初只添加了分区的原始快照。这也是你尝试不同方法后，要么只加载一个分区，要么图片错乱的根本原因。

另外，若集合视图数据源的cellProvider方法没有正确区分分区与对应的数据数组，也会导致图片加载到错误分区。

修复方案 #

1. 统一在所有请求完成后构建快照 #

先将三个服务返回的数据保存到实例变量，等所有请求完成后，再一次性构建完整的快照并应用：

// 假设你已定义对应的实例变量存储图片数据
var firstPhotos: [Photo] = []
var secondPhotos: [Photo] = []
var thirdPhotos: [Photo] = []

// 初始化请求组
let group = DispatchGroup()

// 请求第一分区数据
group.enter()
firstService.fetchFirst { [weak self] result in
    defer { group.leave() }
    guard let self = self else { return }
    switch result {
    case .success(let photos):
        self.firstPhotos = photos
    case .failure(let error):
        print("第一分区请求失败：\(error)")
    }
}

// 请求第二分区数据
group.enter()
secondService.fetchSecond { [weak self] result in
    defer { group.leave() }
    guard let self = self else { return }
    switch result {
    case .success(let photos):
        self.secondPhotos = photos
    case .failure(let error):
        print("第二分区请求失败：\(error)")
    }
}

// 请求第三分区数据
group.enter()
thirdService.fetchThird { [weak self] result in
    defer { group.leave() }
    guard let self = self else { return }
    switch result {
    case .success(let photos):
        self.thirdPhotos = photos
    case .failure(let error):
        print("第三分区请求失败：\(error)")
    }
}

// 所有请求完成后，统一构建并应用快照
group.notify(queue: .main) { [weak self] in
    guard let self = self else { return }
    
    var snapshot = NSDiffableDataSourceSnapshot<SectionLayoutKind, Int>()
    snapshot.appendSections([.one, .two, .three])
    
    // 为每个分区添加对应数量的item标识符（这里用索引作为标识符，需确保cellProvider对应正确）
    snapshot.appendItems(Array(0..<self.firstPhotos.count), toSection: .one)
    snapshot.appendItems(Array(0..<self.secondPhotos.count), toSection: .two)
    snapshot.appendItems(Array(0..<self.thirdPhotos.count), toSection: .three)
    
    self.dataSource.apply(snapshot, animatingDifferences: false)
}

2. 确保Cell Provider正确匹配分区与数据 #

检查你的数据源cellProvider方法，确保根据当前分区正确读取对应的数据数组：

// 初始化数据源时的cellProvider示例
let dataSource = UICollectionViewDiffableDataSource<SectionLayoutKind, Int>(collectionView: collectionView) { [weak self] collectionView, indexPath, itemId in
    guard let self = self else { return nil }
    
    let cell = collectionView.dequeueReusableCell(withReuseIdentifier: "PhotoCell", for: indexPath) as! PhotoCell
    let section = SectionLayoutKind(rawValue: indexPath.section)!
    
    // 根据分区读取对应数据
    switch section {
    case .one:
        let photo = self.firstPhotos[itemId]
        cell.configure(with: photo)
    case .two:
        let photo = self.secondPhotos[itemId]
        cell.configure(with: photo)
    case .three:
        let photo = self.thirdPhotos[itemId]
        cell.configure(with: photo)
    }
    
    return cell
}

额外建议 #

• 若你的图片数据模型有唯一标识符（如id字段），建议将快照的ItemIdentifierType改为模型的唯一标识符类型（而非Int），这样能避免因索引变化导致的数据源错乱问题。
• 可以为每个分区的请求添加独立的错误处理逻辑，比如请求失败时显示占位图或提示信息。

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