Swift协议关联的类型和继承约束

Question

我想用UICollectionViewController构建一个iOS应用，该应用总是每行具有相同数量的单元格。 因为我不想让UICollectionViewController处理太多事情，所以我重构了代码并实现了有趣的事情，例如protocol associatedtype和泛型。 现在，我的应用程序由4个不同的.swift文件组成。

---

1. CustomFlowLayout.swift

CustomFlowLayout是一个简单的子类UICollectionViewFlowLayout ，让我们来设置其minimumInteritemSpacing ， minimumLineSpacing和sectionInset依赖注入得益于便利的初始化与性能。

import UIKit

class CustomFlowLayout: UICollectionViewFlowLayout {

    convenience init(minimumInteritemSpacing: CGFloat = 0,
                     minimumLineSpacing: CGFloat = 0,
                     sectionInset: UIEdgeInsets = .zero) {
        self.init()
        self.minimumInteritemSpacing = minimumInteritemSpacing
        self.minimumLineSpacing = minimumLineSpacing
        self.sectionInset = sectionInset
    }

    override init() {
        super.init()
    }

    required init?(coder aDecoder: NSCoder) {
        fatalError("init(coder:) has not been implemented")
    }

}

2. ColumnDataSource.swift

ColumnDataSource是NSObject的子类，符合UICollectionViewDataSource ， UICollectionViewDelegate和UICollectionViewDelegateFlowLayout 。 它实现collectionView(_:layout:sizeForItemAt:)以便每行显示正确的UICollectionViewCell数量。 还要注意， ColumnDataSource是一个通用类，要求我们在初始化时将其传递给类型参数。

import UIKit

class ColumnDataSource<FlowLayoutType: UICollectionViewFlowLayout>: NSObject, UICollectionViewDataSource, UICollectionViewDelegate, UICollectionViewDelegateFlowLayout {

    let cellsPerRow: Int

    init(cellsPerRow: Int) {
        self.cellsPerRow = cellsPerRow
        super.init()
    }

    // MARK: - UICollectionViewDataSource

    func collectionView(_ collectionView: UICollectionView, numberOfItemsInSection section: Int) -> Int {
        return 10
    }

    func collectionView(_ collectionView: UICollectionView, cellForItemAt indexPath: IndexPath) -> UICollectionViewCell {
        return collectionView.dequeueReusableCell(withReuseIdentifier: "Cell", for: indexPath)
    }

    // MARK: - UICollectionViewDelegateFlowLayout

    func collectionView(_ collectionView: UICollectionView, layout collectionViewLayout: UICollectionViewLayout, sizeForItemAt indexPath: IndexPath) -> CGSize {
        let flowLayout = collectionView.collectionViewLayout as! FlowLayoutType
        let marginsAndInsets = flowLayout.sectionInset.left + flowLayout.sectionInset.right + flowLayout.minimumInteritemSpacing * (CGFloat(cellsPerRow) - 1)
        let itemWidth = (collectionView.bounds.size.width - marginsAndInsets) / CGFloat(cellsPerRow)
        return CGSize(width: itemWidth, height: itemWidth)
    }

}

3. ColumnFlowLayoutable.swift

ColumnFlowLayoutable协议的目的是确保符合该协议的任何类都具有columnDataSource和customFlowLayout属性，其中columnDataSource类型的type参数与customFlowLayout类型相匹配。

import UIKit

protocol ColumnFlowLayoutable {

    associatedtype FlowLayoutType: UICollectionViewFlowLayout
    var columnDataSource: ColumnDataSource<FlowLayoutType> { get }
    var customFlowLayout: FlowLayoutType { get }

}

4. CollectionViewController.swift

CollectionViewController是UICollectionViewController的子类，它符合ColumnFlowLayoutable协议。 它还实现了viewWillTransition(to:with:)以便处理容器大小的更改。

import UIKit

class CollectionViewController: UICollectionViewController, ColumnFlowLayoutable {

    let columnDataSource = ColumnDataSource<CustomFlowLayout>(cellsPerRow: 2)
    let customFlowLayout = {
        CustomFlowLayout(minimumInteritemSpacing: $0, minimumLineSpacing: $0, sectionInset: UIEdgeInsets(top: $0, left: $0, bottom: $0, right: $0))
    }(10)

    override func viewDidLoad() {
        super.viewDidLoad()

        collectionView?.collectionViewLayout = customFlowLayout
        collectionView?.dataSource = columnDataSource
        collectionView?.delegate = columnDataSource
    }

    override func viewWillTransition(to size: CGSize, with coordinator: UIViewControllerTransitionCoordinator) {
        super.viewWillTransition(to: size, with: coordinator)
        collectionView?.collectionViewLayout.invalidateLayout()
    }

}

可以在以下Github存储库中找到完整的项目： CollectionViewColumnsProtocol 。

---

此代码可以正常工作。 我可以将其与CustomFlowLayout子类CustomFlowLayout ，并且仍然可以使用。 但是，我不能将其与ColumnDataSource子类一起使用 。

如果我尝试通过使用CollectionViewController的ColumnDataSource的子类（例如， class SubColumnDataSource: ColumnDataSource<CustomFlowLayout> ）来构建项目，则Xcode会引发以下构建时错误消息：

类型“ CollectionViewController”不符合协议“ ColumnFlowLayoutable”

为了允许CollectionViewController使用ColumnDataSource子类，我必须在ColumnFlowLayoutable协议中进行哪些更改？

Answer 1

创建具有与您的布局类型关联的类型的数据源协议：

protocol ColumnDataSourceProtocol: UICollectionViewDataSource, UICollectionViewDelegate, UICollectionViewDelegateFlowLayout {
    associatedtype Layout: UICollectionViewFlowLayout
}

使您的DataSource基类符合此协议。 如果编译器无法推断出关联类型，则可能需要添加一个类型typealias ：

class ColumnDataSource<FlowLayoutType: UICollectionViewFlowLayout>: NSObject, ColumnDataSourceProtocol {

    typealias Layout = FlowLayoutType

    // the rest stays the same
}

调整ColumnFlowLayoutable以关联数据源类型而不是布局类型。 将其约束到ColumnDataSourceProtocol ，可以访问其关联的Layout类型：

protocol ColumnFlowLayoutable {

    associatedtype DataSource: ColumnDataSourceProtocol

    var columnDataSource: DataSource { get }
    var customFlowLayout: DataSource.Layout { get }
}

现在，您可以将ColumnDataSource子类化：

class DataSource: ColumnDataSource<CustomFlowLayout> { /* ... */ }