作为一个iOS开发者，最常用的任务就是通过自定义cell的子类，来实现UITableView或者UICollectionView的自定义。并且 UITableView 和 UICollectionView 在注册自定义cell子类这一块都有非常类似的API:
public func registerClass(cellClass: AnyClass?, forCellWithReuseIdentifier identifier: String)
public func registerNib(nib: UINib?, forCellWithReuseIdentifier identifier: String)
对于注册cell的自定义最常用的解决办法就是，声明一个reuseIdentifier的常量，像下面这样:
private let reuseIdentifier = "BookCell"
class BookListViewController: UIViewController, UICollectionViewDataSource {
@IBOutlet private weak var collectionView: UICollectionView!
override func viewDidLoad() {
super.viewDidLoad()
let nib = UINib(nibName: "BookCell", bundle: nil)
self.collectionView.registerNib(nib, forCellWithReuseIdentifier: reuseIdentifier)
}
func collectionView(collectionView: UICollectionView, cellForItemAtIndexPath indexPath: NSIndexPath) -> UICollectionViewCell {
let cell = collectionView.dequeueReusableCellWithReuseIdentifier(reuseIdentifier, forIndexPath: indexPath)
if let bookCell = cell as? BookCell {
// TODO: configure cell
}
return cell
}
}
接下来让我们尝试着使用泛型来让它简单化和安全化。
首先，如果在我们的代码当中能不需要到处声明reuse identifier常量，那就再好不过了。而事实上，我们可以直接使用自定义cell的类名来当做 默认的reuseIdentifier。
我们可以通过创建一个Reuseable Views的协议并且创建默认的声明方法给 UIView 的子类们。
protocol ReusableView: class {
static var defaultReuseIdentifier: String { get }
}
extension ReusableView where Self: UIView {
static var defaultReuseIdentifier: String {
return NSStringFromClass(self)
}
}
extension UICollectionViewCell: ReusableView {
}
通过让 UICollectionViewCell 遵循 ReusableView 协议，我们可以得到每个cell子类的一个唯一的重用标识。
let identifier = BookCell.defaultReuseIdentifier
// identifier = "MyModule.BookCell"
接下来，我们通过同样的方法，将注册Nib步骤中的一些脏代码给去除掉。
我们创建一个 Nib Loadable Views 的协议并通过协议拓展添加一个默认方法实现。
protocol NibLoadableView: class {
static var nibName: String { get }
}
extension NibLoadableView where Self: UIView {
static var nibName: String {
return NSStringFromClass(self).componentsSeparatedByString(".").last!
}
}
extension BookCell: NibLoadableView {
}
通过让我们的 BookCell 类遵循 NibLoadableView 协议，现在我们就有了一个更安全和方便的方法去获得到Nib的名称。
let nibName = BookCell.nibName
// nibName = "BookCell"
有这两个协议，我们可以通过使用 Swift的泛型 并且通过拓展 UICollectionView 来简化cell的注册和使用。
extension UICollectionView {
func register<T: UICollectionViewCell where T: ReusableView>(_: T.Type) {
registerClass(T.self, forCellWithReuseIdentifier: T.defaultReuseIdentifier)
}
func register<T: UICollectionViewCell where T: ReusableView, T: NibLoadableView>(_: T.Type) {
let bundle = NSBundle(forClass: T.self)
let nib = UINib(nibName: T.nibName, bundle: bundle)
registerNib(nib, forCellWithReuseIdentifier: T.defaultReuseIdentifier)
}
func dequeueReusableCell<T: UICollectionViewCell where T: ReusableView>(forIndexPath indexPath: NSIndexPath) -> T {
guard let cell = dequeueReusableCellWithReuseIdentifier(T.defaultReuseIdentifier, forIndexPath: indexPath) as? T else {
fatalError("Could not dequeue cell with identifier: \(T.defaultReuseIdentifier)")
}
return cell
}
}
注意这里，我们创建了两个版本的注册方法，一个是用来注册 ReusableView 子类用的，一个是用来注册 ReusableView 和 NibLoadableView的子类。这很好的将view controller的特定的注册方法分离出来。
另一个比较棒的细节就是 dequeueReusableCell 方法不再需要给他任何重用标识字符串而且可以直接使用cell的子类作为返回值。
现在cell的注册和使用代码看起来棒极了 :) 。
class BookListViewController: UIViewController, UICollectionViewDataSource {
@IBOutlet private weak var collectionView: UICollectionView!
override func viewDidLoad() {
super.viewDidLoad()
self.collectionView.register(BookCell.self)
}
func collectionView(collectionView: UICollectionView, cellForItemAtIndexPath indexPath: NSIndexPath) -> UICollectionViewCell {
let cell: BookCell = collectionView.dequeueReusableCell(forIndexPath: indexPath)
// TODO: configure cell
return cell
}
...
}
总结
如果你是从 Objective-C 转到 Swift 的话，研究Swift强大的新特性比如 协议拓展、 泛型， 从而找到更优雅的实现方式和替代方法是非常值得的。
