1、前提

敏捷大拇指上的各位Swift开发者,大家应该常遇到在table view的delegate里处理若干个cell的问题,很多人的条件语句是这样的:

[Swift] 纯文本查看 复制代码
if indexPath.row == 0 || indexPath.row == 1 || indexPath.row == 3 {
	// do something
}


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

  • 当我有更多选项时长度已经超出字符限制
  • 没有办法实现可变数量的选项


简洁使用Swift实例(isIn),扩写Equatable协议,用活Range

简洁使用Swift实例(isIn),扩写Equatable协议,用活Range - 敏捷大拇指 - 简洁使用Swift实例(isIn),扩写Equatable协议,用活Range





2、解决方案

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

[Swift] 纯文本查看 复制代码
  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
    }





3、优化

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

[Swift] 纯文本查看 复制代码
// 定义
    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函数,其定义如下:

[Swift] 纯文本查看 复制代码
  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( : )函数。代码如下:

[Swift] 纯文本查看 复制代码
extension Equatable {
        func isIn(collection: T) -> Bool {
            return collection.contains(self)
        }
         
        func isIn(collection: Self...) -> Bool {
            return collection.contains(self)
        }
    }


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

[Swift] 纯文本查看 复制代码
   public struct Range: Equatable, CollectionType, CustomStringConvertible, CustomDebugStringConvertible {...}


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




4、后记

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