Swift中如何DRY化collectionView(_:cellForItemAt:)重复代码？

绝对有办法把这块重复代码彻底干掉！你当前的写法肯定不是最优解，而你考虑的「让所有Cell遵循同一协议+数组管理」的思路完全找对了方向，我给你一步步拆解具体怎么落地，还有几个能让代码更优雅的小技巧。

第一步：定义统一协议，规范Cell行为 #

先创建一个协议，让所有Cell都遵循它，协议要包含两个核心内容：复用标识符（避免硬编码）、配置方法（把Cell的UI配置逻辑内聚到Cell本身）。

如果想要类型安全，可以用关联类型（associatedtype）来绑定Cell对应的模型，避免传错数据：

protocol CellConfigurable: UICollectionViewCell {
    // 静态复用标识符，每个Cell自己定义
    static var reuseIdentifier: String { get }
    // 关联对应的模型类型，保证配置时类型匹配
    associatedtype Model
    // 配置方法，接收对应模型
    func configure(with model: Model)
}

第二步：让每个Cell实现协议 #

接下来让你的每一种Cell都实现这个协议，把原本写在cellForItemAt里的配置逻辑移到Cell内部，比如：

// 文本Cell示例
class TextCell: UICollectionViewCell, CellConfigurable {
    static let reuseIdentifier = "TextCell"
    typealias Model = TextModel // 绑定对应的模型
    
    // 假设你有一个UILabel
    private let titleLabel = UILabel()
    
    override init(frame: CGRect) {
        super.init(frame: frame)
        // 这里做UI布局初始化
        contentView.addSubview(titleLabel)
        titleLabel.translatesAutoresizingMaskIntoConstraints = false
        NSLayoutConstraint.activate([
            titleLabel.centerXAnchor.constraint(equalTo: contentView.centerXAnchor),
            titleLabel.centerYAnchor.constraint(equalTo: contentView.centerYAnchor)
        ])
    }
    
    required init?(coder: NSCoder) {
        fatalError("init(coder:) has not been implemented")
    }
    
    func configure(with model: TextModel) {
        // 把配置逻辑放在这里，符合*单一职责原则*
        titleLabel.text = model.content
        titleLabel.textColor = model.textColor
    }
}

// 图片Cell示例
class ImageCell: UICollectionViewCell, CellConfigurable {
    static let reuseIdentifier = "ImageCell"
    typealias Model = ImageModel
    
    private let iconImageView = UIImageView()
    
    override init(frame: CGRect) {
        super.init(frame: frame)
        contentView.addSubview(iconImageView)
        iconImageView.translatesAutoresizingMaskIntoConstraints = false
        NSLayoutConstraint.activate([
            iconImageView.leadingAnchor.constraint(equalTo: contentView.leadingAnchor, constant: 8),
            iconImageView.trailingAnchor.constraint(equalTo: contentView.trailingAnchor, constant: -8),
            iconImageView.topAnchor.constraint(equalTo: contentView.topAnchor, constant: 8),
            iconImageView.bottomAnchor.constraint(equalTo: contentView.bottomAnchor, constant: -8)
        ])
    }
    
    required init?(coder: NSCoder) {
        fatalError("init(coder:) has not been implemented")
    }
    
    func configure(with model: ImageModel) {
        iconImageView.image = model.icon
        iconImageView.contentMode = .scaleAspectFit
    }
}

第三步：用统一的数据源模型包装Cell类型和数据 #

我们需要一个容器来把Cell类型和对应的模型绑定起来，这样数据源数组就能统一管理不同类型的Cell数据：

struct CellData {
    let cellType: CellConfigurable.Type
    let model: Any
}

// 为了保证类型安全，给CellData加一个泛型初始化方法，避免传错模型
extension CellData {
    init<Cell: CellConfigurable>(cellType: Cell.Type, model: Cell.Model) {
        self.cellType = cellType
        self.model = model
    }
}

现在你的数据源数组就可以这样定义和填充：

// 数据源数组
var collectionViewData: [CellData] = []

// 填充示例数据
collectionViewData.append(
    CellData(cellType: TextCell.self, model: TextModel(content: "Hello World", textColor: .black))
)
collectionViewData.append(
    CellData(cellType: ImageCell.self, model: ImageModel(icon: UIImage(named: "home_icon")!))
)

第四步：简化cellForItemAt方法 #

现在原本冗长的cellForItemAt会变得异常简洁，再也不用写一堆if/switch判断了：

func collectionView(_ collectionView: UICollectionView, cellForItemAt indexPath: IndexPath) -> UICollectionViewCell {
    let cellData = collectionViewData[indexPath.item]
    // 根据Cell类型复用Cell
    let cell = collectionView.dequeueReusableCell(
        withReuseIdentifier: cellData.cellType.reuseIdentifier,
        for: indexPath
    ) as! CellConfigurable
    
    // 调用Cell的配置方法
    cell.configure(with: cellData.model)
    
    return cell
}

额外优化：批量注册Cell #

不用一个个手动注册Cell，可以提前维护一个Cell类型数组，批量注册：

override func viewDidLoad() {
    super.viewDidLoad()
    
    // 所有需要用到的Cell类型
    let cellTypes: [CellConfigurable.Type] = [TextCell.self, ImageCell.self]
    
    // 批量注册
    cellTypes.forEach {
        collectionView.register($0, forCellWithReuseIdentifier: $0.reuseIdentifier)
    }
}

进阶：搭配UICollectionViewDiffableDataSource #

如果你用的是DiffableDataSource（iOS 13+），可以把这个思路和Diffable结合，让数据源管理更高效，而且同样不用修改Cell配置的核心逻辑，只需要调整数据源快照的创建方式即可。

总结 #

这种方式完美符合DRY原则：

• 后续新增Cell时，只需要让Cell遵循CellConfigurable协议，添加到数据源数组即可，完全不用修改cellForItemAt代码
• 把Cell的配置逻辑内聚到Cell本身，符合单一职责原则，代码更易维护
• 类型安全的设计避免了传错模型的风险

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