简洁使用Swift实例(isIn)

前提

大家应该常遇到在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强大的特性支持我们在不同维度约束条件,既安全又强大.