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类型参数匹配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的子类一起使用，它仍然有效。 但是，我无法将其用于ColumnDataSource 的子类。

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

  

输入＆＃39; CollectionViewController＆＃39;不符合协议＆＃39; ColumnFlowLayoutable＆＃39;

我需要在ColumnFlowLayoutable协议中更改哪些内容才能允许CollectionViewController使用ColumnDataSource的子类？

Answer 1

使用布局类型的关联类型创建DataSource协议：

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> { /* ... */ }