前提
大家应该常遇到在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<Int>) -> Bool {
return range.contains(self)
}
func isIn<T: SequenceType where T.Generator.Element == Int>(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<T: SequenceType where T.Generator.Element == Self>(collection: T) -> Bool {
return collection.contains(self)
}
func isIn(collection: Self...) -> Bool {
return collection.contains(self)
}
}
这里需要说明下Range不需额外定义的理由.首先Range的定义如下:
public struct Range<Element : ForwardIndexType> : Equatable, CollectionType, CustomStringConvertible, CustomDebugStringConvertible {...}
它遵循于CollectionType(继承自SequenceType),它的Element遵循于ForwardIndexType(继承自Equatable),所以它是符合extension Equatable中第一个函数的.
后记
至此,我们用最简洁的方式写出了contains的反函数.这一切要感谢Swift强大的特性支持我们在不同维度约束条件,既安全又强大.
