UITableView 协议方法中更好的 switch 使用方案

UITableView 几乎是每一个原生 App 都会使用到的控件，所以你一定对 UITableViewDataSource、UITableViewDelegate 中的代理方法很熟悉。在一些代理方法中有时候我们需要对不同的 section、row 进行分开处理，此时我们都会使用 switch 来应对。所以接下来将会介绍一种更好的 switch 使用方案。

常规处理

大多数情况下，我们使用的 switch 方案都是类似下面的硬编码方式：

func tableView(_ tableView: UITableView, cellForRowAt indexPath: IndexPath) -> UITableViewCell {
    switch indexPath.section {
        case 0: // Dequeue a cell for section 0
        case 1: // Dequeue a cell for section 1
        // And so on...
        default: fatalError("Unhandled section \(indexPath.section)")
    }
}

上诉代码在功能上并没有什么权限，但是这种硬编码方式对后期的代码维护带来的隐性成本。其实一个好的编程习惯就是最大限度的避免硬编码，无论是 switch 还是设置 view 的 tag 的时候都应该采用更为“优雅”的方式。

更好的处理方式

对于硬编码的优雅的替换方案大多数情况下都是使用 enum 方式，在枚举类型中我们可以使用更具识别性的词语。这样不仅能一眼识别含义，而且在后期修改的过程中也不用满世界找硬编码的数字生怕有遗漏。

接下来直接上码：

protocol ReusableViewEnum {}

extension ReusableViewEnum where Self: RawRepresentable, Self.RawValue == Int {
    static var all: [Self] {
        var index = 0
        var allItems = [Self]()
        while let item = Self(rawValue: index) {
            allItems.append(item)
            index += 1
        }
        return allItems
    }

    static func build(with value: Int) -> Self {
        guard let row = Self(rawValue: value) else {
            fatalError("Unimplemented value: \(value)")
        }
        return row
    }
}

// ---------------------
// EXAMPLE USAGE CODE:

// You only have to define the enum
fileprivate enum ProfileSections: Int, ReusableViewEnum {
    case picture = 0
    case name
    case email
    case password
}

// And use it in your code:
func numberOfSections(in tableView: UITableView) -> Int {
    return ProfileSections.all.count
}

func tableView(_ tableView: UITableView, numberOfRowsInSection section: Int) -> Int {
    switch ProfileSection.build(with: section) {
        case .picture   : return 1
        case .name      : return 1  
        case .email     : return 1 
        case .password  : return 1 
    }
}   

func tableView(_ tableView: UITableView, cellForRowAt indexPath: IndexPath) -> UITableViewCell {
    switch ProfileSection.build(with: indexPath.section) {
        case .picture   : // Dequeue profile picture section
        case .name      : // Dequeue name section
        case .email     : // Dequeue email section
        case .password  : // Dequeue password section
    }
}

首先，我们什么了 ReusableViewEnum 协议并对其进行了拓展。在拓展中我们添加了一个计算属性 all， 用于表示遵循该协议并且 rawValue 为 Int的枚举类型的所有取值。另外增加了用 Int 来构建枚举类型的函数 build。

在使用实例中我们定义了枚举类型 ProfileSections，并在 UITableView 的代理方法中实现了上面 ReusableViewEnum 协议的应用。应用后的 switch 方案代码风格更为有效，后期维护效率也明显优于之前的硬编码方式。