Swift学习笔试19——扩展（Extension）

Swift中的扩展就是OC中的分类。但是这里的扩展可以对类、结构体、枚举类进行添加。使用扩展，我们可以在不接触源代码的前提下，对已有的代码进行功能上的添加。

Swift中的扩展提供的一些功能：

1、添加计算属性或类计算属性，但是不能添加存储属性。

2、添加实例方法或类别方法。但是不能覆盖已经存在的方法。

3、定义新的构造器。

4、定义和使用新的嵌套类。

5、令已经存在的类型遵守协议。

假设我们已经有了一个Animal类，但是不能得到它的源代码。

class Animal {
    var life: Int = 0
    var height: Double = 0
}

我们利用扩展，对Animal添加一个计算属性和方法、构造器还有嵌套类。 当然也可以添加下标，但这里不演示了。

extension Animal {
    //扩展计算属性
    var lifeYear : Double { return Double(self.life) / 365 }
    //扩展构造器
    convenience init(life: Int, height: Double) {
        self.init()
        self.life = life
        self.height = height
    }
    //扩展方法
    func makeSound() -> String {
        return "I can live \(lifeYear) years"
    }
    // 扩展嵌套类
    enum Size {
        case Big, Small, SuperSmall
    }
    // 扩展一个使用嵌套类的计算属性
    var size: Size {
        switch height {
        case let h where h > 10:
            return .Big
        case let h where h > 0.1:
            return .Small
        default:
            return .SuperSmall
        }
    }
}

var a = Animal(life: 365 * 7, height: 0.001)
print(a.size)   // SuperSmall
print(a.lifeYear)   // 7.0
print(a.makeSound())   // I can live 7.0 years

上面代码需要注意的是扩展构造器的时候，在构造器的时候讲过，在扩展中添加构造器是不会影响默认构造器的产生。不能添加原来已有的构造器。

另外在类扩展里面，是不给添加指定构造器或析构器，只能添加便利构造器。在便利构造器中，要先调用原来的构造器来初始化类后，才能对类进行访问。

扩展还有一个作用就是让类遵守协议。这部分留到协议里面讲。

下面将以前将过的麻烦的String类型做一个扩展，使得方便截串和获取长度。

extension String {
    //获取字符串长度
    var length: Int { return self.characters.count }
    
    //获取部分字符串，如果不在范围内，返回nil。如果end大于字符串长度，那么截取到最后
    subscript (start: Int, end: Int) -> String? {
        if start > self.length || start < 0 || start > end {
            return nil
        }
        let begin = self.startIndex.advancedBy(start)
        var terminal: Index
        if end >= length {
            terminal = self.startIndex.advancedBy(length)
        } else {
            terminal = self.startIndex.advancedBy(end + 1)
        }
        let range = Range(start: begin, end: terminal)
        return self.substringWithRange(range)
    }
    //获取某个字符，如果不在范围内，返回nil
    subscript (index: Int) -> Character? {
        if index > self.length - 1 || index < 0 {
            return nil
        }
        return self[self.startIndex.advancedBy(index)]
    }
}

然后就可以这样截串

var s = "哈0123456789"
print(s.length) // 11
print(s[0,2])    // Optional("哈01")
print(s[9])     // Optional("8")