泛型是Swift里的一项强大功能，但也比较抽象难懂，我也一直不太能搞懂泛型的使用，今天写了一段有点意思的泛型代码，这里分享出来。
先给出完整的代码，然后我再说说如何写的。
class UniformDataSource<ModelType, CellType: UICollectionViewCell>: NSObject, UICollectionViewDataSource {
init(modelPointer: UnsafeBufferPointer<ModelType>, identifier: String, customization: @escaping (CellType, ModelType) -> Void) {
self.modelPointer = modelPointer
self.identifier = identifier
self.customization = customization
super.init()
}
var modelPointer: UnsafeBufferPointer<ModelType>
let identifier: String
let customization: (CellType, ModelType) -> Void
func collectionView(_ collectionView: UICollectionView, numberOfItemsInSection section: Int) -> Int {
return modelPointer.count
}
func collectionView(_ collectionView: UICollectionView, cellForItemAt indexPath: IndexPath) -> UICollectionViewCell {
let cell = collectionView.dequeueReusableCell(withReuseIdentifier: identifier, for: indexPath) as! CellType
customization(cell, modelPointer[indexPath.item])
return cell
}
}
说明一下，这只作学习用途，如果在项目中想用这样的代码，请慎重。
这段代码是受Lighter View Controllers这篇文章的第一部分"Separate Out Data Source and Other Protocols"所写的。该部分的大致内容是想在Objective-C中，分离出UITableView的dataSource以简化View Controller的体量。但在Swift中，Model层的数据更多地使用struct来封装（至少我的项目里，model几乎都是struct），不像OC里用对象。这种value type和reference type的区别使得用Swift实现该篇文章里的做法变得困难。我们暂时把这个点先放一放，过后再来讨论。
下面开始一步步分析，我们要用泛型分离UICollectionView的dataSource！
第一步，创建一个类，名叫UniformDataSource。
class UniformDataSource: NSObject, UICollectionViewDataSource {
// 这个回调是optional的，因此当section为1时，我们可以注释掉。它默认返回 1。
/*
func numberOfSections(in collectionView: UICollectionView) -> Int {
return 1
}
*/
func collectionView(_ collectionView: UICollectionView, numberOfItemsInSection section: Int) -> Int {
// 返回model数量
return 0
}
func collectionView(_ collectionView: UICollectionView, cellForItemAt indexPath: IndexPath) -> UICollectionViewCell {
let cell = collectionView.dequeueReusableCell(withReuseIdentifier: "cell", for: indexPath)
// 这里写一些对cell的定制操作...
return cell
}
}
我们希望这个UniformDataSource可以在多个UICollectionViewController中使用，即使model数据可能各不相同。于是我们就需要引入泛型了。为我们的数据给一个占位符ModelType。然后在对cell的定制操作中，不同的cell类会有不同的属性和方法，为了能够好好地跟不同的cell玩耍，我们也给cell一个占位符CellType，但这个CellType不能随随便便，它需要是UICollectionViewCell，那我们就给它一个类型约束，限制它一定是UICollectionViewCell。
如此一来，UniformDataSource的声明就变成了这样：
class UniformDataSource<ModelType, CellType: UICollectionViewCell>: NSObject, UICollectionViewDataSource {
...
}
（好吧看到尖括号确实还是有点怕怕，但它们其实无非就是 一个模型类型 和 一个Cell类型）
第二步，我们看定制操作那个回调。对于定制操作，我们可以用一个闭包来表达，它由上层的UICollectionViewController传入即可。
let customization: (CellType, ModelType) -> Void
func collectionView(_ collectionView: UICollectionView, cellForItemAt indexPath: IndexPath) -> UICollectionViewCell {
let cell = collectionView.dequeueReusableCell(withReuseIdentifier: "cell", for: indexPath) as! CellType
customization(cell, xxx) // xxx 是对应的model
return cell
}
OK，可能你已经看到那个xxx了，现在我们来填完这个models。这里，因为我们Swift喜欢玩struct，而struct是value type的，因此如果我们像OC那样把models写在view controller中，然后传给我们这个UniformDataSource对象的话，就不是传引用，而是传值了哦。如果models已经比较大，有上百个，那一次传值岂不是开销很大？
在接下来看到的代码中，我为了不直接传值，不得不使用指针来实现对在上层UICollectionViewController中模型的引用。如果你对在Swift中玩指针不太熟悉，可以先自己去捣鼓一下。
先对Array写一个extension，返回一个指针，像这样：
extension Array {
var bufferPointer: UnsafeBufferPointer<Element> {
return withUnsafeBufferPointer { (pointer: UnsafeBufferPointer<Element>) -> UnsafeBufferPointer<Element> in
return pointer
}
}
}
withUnsafeBufferPointer需要传入一个闭包，闭包有且仅有一个参数UnsafeBufferPointer<Element>，返回任意类型。为了将这个指针传回出闭包外，我们就将返回的类型也设为跟指针一样的类型，即UnsafeBufferPointer<Element>。机智的你也许已经发现，上面的extension可以简化成以下的写法：
extension Array {
var bufferPointer: UnsafeBufferPointer<Element> {
return withUnsafeBufferPointer { $0 }
}
}
神清气爽！Swift的魅力！🎉
回到UniformDataSource类中，我们加一个属性，用于保存从上层UICollectionViewController中传入的模型数据：
var modelPointer: UnsafeBufferPointer<ModelType>
这样cellForItemAtIndexPath那个回调也可以完成了：
func collectionView(_ collectionView: UICollectionView, cellForItemAt indexPath: IndexPath) -> UICollectionViewCell {
let cell = collectionView.dequeueReusableCell(withReuseIdentifier: identifier, for: indexPath) as! CellType
customization(cell, modelPointer[indexPath.item])
return cell
}
顺便也可以完成numberOfItemsInSection了：
func collectionView(_ collectionView: UICollectionView, numberOfItemsInSection section: Int) -> Int {
return modelPointer.count
}
最后一步，加个init方法，也把reuseIdentifier加进来：
init(modelPointer: UnsafeBufferPointer<ModelType>, identifier: String, customization: @escaping (CellType, ModelType) -> Void) {
self.modelPointer = modelPointer
self.identifier = identifier
self.customization = customization
super.init()
}
完成。完整的代码就在最上面。
然而，实际使用时还是比较别扭，为什么呢？看下面的测试代码。
struct TestData { }
class TestCollectionViewCell: UICollectionViewCell {
func customize(with data: TestData) { }
}
class TestCollectionViewController: UICollectionViewController {
var dataSource: UniformDataSource<TestData, TestCollectionViewCell>?
var datas: [TestData]! {
didSet {
dataSource?.modelPointer = datas.bufferPointer
}
}
override func viewDidLoad() {
super.viewDidLoad()
dataSource = UniformDataSource(
modelPointer: datas.bufferPointer,
identifier: "cell",
customization: { (cell: TestCollectionViewCell, data: TestData) in
cell.customize(with: data)
})
}
}
就是，每次datas变动后，我们都需要为dataSource重新设置指针。。。
就这样。Enjoy!!
