RxSwift -如何在数据源更改时保留UICollectionView的contentOffSet和选择

let dataSource = BehaviorRelay<[YourDataType]>(value: initialData)

dataSource.subscribe(onNext: { [weak self] newData in
    // 保存当前的contentOffset和选择
    let currentContentOffset = self?.collectionView.contentOffset
    let currentSelection = self?.collectionView.indexPathsForSelectedItems
    
    // 更新数据源
    self?.dataSource.accept(newData)
    
    // 恢复之前保存的contentOffset和选择
    self?.collectionView.contentOffset = currentContentOffset ?? .zero
    self?.collectionView.selectItem(at: currentSelection, animated: false, scrollPosition: .none)
}).disposed(by: disposeBag)