UICollectionView瀑布流布局拖动单元格时固定尺寸的解决方法

咱们先揪出问题的核心：你用的Pinterest风格瀑布流布局，核心是靠代理方法返回每个cell的高度来计算布局的。拖动单元格时布局会重新计算，如果此时没正确保留每个cell的原始高度，就会出现尺寸乱跳的情况。下面是具体的解决步骤：

1. 给布局添加高度缓存 #

在你的WidgetCollectionViewLayout里，新增一个字典来缓存每个indexPath对应的单元格高度，避免每次布局计算都重复请求代理（防止拖动过程中代理返回错误值）：

class WidgetCollectionViewLayout: UICollectionViewLayout {
    // 新增：缓存每个cell的高度，key是indexPath，value是对应的高度
    private var cellHeightCache: [IndexPath: CGFloat] = [:]
    
    // ... 你原来的其他属性
    
    // 初始化或数据源更新时，提前填充缓存
    func updateHeightCache() {
        guard let collectionView = collectionView, 
              let delegate = delegate as? WidgetCollectionViewLayoutDelegate else { return }
        
        let itemCount = collectionView.numberOfItems(inSection: 0)
        for item in 0..<itemCount {
            let indexPath = IndexPath(item: item, section: 0)
            if cellHeightCache[indexPath] == nil {
                let height = delegate.collectionView(collectionView, heightForPhotoAt: indexPath)
                cellHeightCache[indexPath] = height
            }
        }
    }
}

2. 布局计算优先用缓存高度 #

修改布局里计算单元格高度的逻辑，不再直接调用代理，而是先从缓存取值，缓存没有时再调用代理并缓存结果：

// 假设你原来计算高度的代码是这样的：
// let height = delegate.collectionView(collectionView, heightForPhotoAt: indexPath)

// 改成下面的逻辑：
private func heightForItem(at indexPath: IndexPath) -> CGFloat {
    guard let collectionView = collectionView,
          let delegate = delegate as? WidgetCollectionViewLayoutDelegate else { return 0 }
    
    if let cachedHeight = cellHeightCache[indexPath] {
        return cachedHeight
    }
    
    // 缓存没有时，调用代理并缓存
    let height = delegate.collectionView(collectionView, heightForPhotoAt: indexPath)
    cellHeightCache[indexPath] = height
    return height
}

之后在布局计算UICollectionViewLayoutAttributes的frame时，都用这个heightForItem方法来获取高度。

3. 拖动排序完成后更新缓存 #

当单元格拖动完成、位置改变后，需要同步更新缓存的indexPath映射，因为原来的单元格位置变了，缓存的key也要对应调整：

// 假设你实现了UICollectionViewDropDelegate的方法
func collectionView(_ collectionView: UICollectionView, performDropWith coordinator: UICollectionViewDropCoordinator) {
    guard let destinationIndexPath = coordinator.destinationIndexPath else { return }
    
    for dropItem in coordinator.items {
        guard let sourceIndexPath = dropItem.sourceIndexPath else { continue }
        
        // 1. 先更新你的数据源（比如数组移动元素）
        let movedModel = yourDataSourceArray[sourceIndexPath.item]
        yourDataSourceArray.remove(at: sourceIndexPath.item)
        yourDataSourceArray.insert(movedModel, at: destinationIndexPath.item)
        
        // 2. 更新高度缓存：把原位置的高度映射到新位置
        if let height = cellHeightCache[sourceIndexPath] {
            cellHeightCache.removeValue(forKey: sourceIndexPath)
            cellHeightCache[destinationIndexPath] = height
            
            // 如果是插入到中间位置，还要调整中间元素的缓存映射
            let startIndex = min(sourceIndexPath.item, destinationIndexPath.item)
            let endIndex = max(sourceIndexPath.item, destinationIndexPath.item)
            
            for i in startIndex..<endIndex {
                let oldPath = IndexPath(item: i + (sourceIndexPath.item < destinationIndexPath.item ? 1 : 0), section: 0)
                let newPath = IndexPath(item: i, section: 0)
                if let h = cellHeightCache[oldPath] {
                    cellHeightCache[newPath] = h
                    cellHeightCache.removeValue(forKey: oldPath)
                }
            }
        }
    }
    
    // 重新刷新布局，确保新位置的单元格尺寸正确
    collectionView.reloadData()
}

4. 确保拖动预览的尺寸正确 #

如果用系统的拖动API，拖动时的预览单元格可能出现尺寸偏差，你可以通过下面的方法固定预览尺寸：

func collectionView(_ collectionView: UICollectionView, dropPreviewParametersForItemAt indexPath: IndexPath) -> UIDropPreviewParameters? {
    let parameters = UIDropPreviewParameters()
    if let attributes = collectionView.layoutAttributesForItem(at: indexPath) {
        // 强制预览用原单元格的frame
        parameters.visiblePath = UIBezierPath(rect: attributes.frame)
    }
    return parameters
}

额外检查建议 #

• 尽量让heightForPhotoAt代理方法从数据源模型里取高度（比如模型里提前存储了图片的高度值），不要依赖单元格的子视图frame计算——拖动过程中单元格可能还没完成渲染，子视图frame会不准确。
• 如果是自定义的拖动手势，在拖动过程中更新单元格临时frame时，一定要用原始的尺寸值，不要动态重新计算。

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