swift 中tableView 多个cell判断的简洁写法

大家应该常遇到在table view的delegate里处理若干个cell的问题.很多人的条件语句是这样的:

if indexPath.row == 0 || indexPath.row == 1 || indexPath.row == 3 {

// do something

}

这个写法非常常规,也清晰明了.但是致命的缺点是

当我有更多选项时长度已经超出字符限制

没有办法实现可变数量的选项

解决方案

简单处理下,可以将所有可能放到数组或者范围(如果连续)中.然后调用其contains方法用于校验.代码如下:

if (2...8).contains(indexPath.row) {

// do something

}

if [2,4,6,9].contains(indexPath.row) {

// do something

}

if 2...8 ~= indexPath.row {                // 匹配运算符

// do something

}

优化

这样虽然解决了前提所述的两个问题,但是写法不够美观,语义不够清晰.如果说可以使用isIn(:)函数处理,就显得美观多了.实现如下:

// 定义

extension Int {

func isIn(range: Range) -> Bool {

return range.contains(self)

}

func isIn(ints: T) -> Bool {

return ints.contains(self)

}

func isIn(ints: Int...) -> Bool {  //这里使用可变参数函数来避免必须将参数放入数组

return ints.contains(self)

}

}

// 使用

if indexPath.row.isIn(4,5,6) {

// do something

}

这样实现了美观简洁地使用,但是还不够通用.比如UInt,Float等可以支持Range或数组的类型却无法使用.如果要更加通用,必须研究下contains函数,其定义如下:

extension SequenceType where Generator.Element : Equatable {

/// Returns `true` iff `element` is in `self`.

@warn_unused_result

public func contains(element: Self.Generator.Element) -> Bool

}

那么显然我们需要扩写Equatable协议,让其实现isIn(:)函数.代码如下:


extension Equatable {

func isIn(collection: T) -> Bool {

return collection.contains(self)

}

func isIn(collection: Self...) -> Bool {

return collection.contains(self)

}

}

这里需要说明下Range不需额外定义的理由.首先Range的定义如下:

1

public struct Range: Equatable, CollectionType, CustomStringConvertible, CustomDebugStringConvertible {...}

它遵循于CollectionType(继承自SequenceType),它的Element遵循于ForwardIndexType(继承自Equatable),所以它是符合extension Equatable中第一个函数的.

后记

至此,我们用最简洁的方式写出了contains的反函数.这一切要感谢Swift强大的特性支持我们在不同维度约束条件,既安全又强大.

推荐阅读更多精彩内容