• 扩展的概念
• 扩展计算属性
• 扩展方法
• 扩展构造函数
• 扩展下标

扩展的概念

在Swift中可以使用一种扩展机制，在原有类型（类、结构体和枚举）的基础上添加新功能。扩展是一种“轻量级”的继承机制，即使原有类型被限制继承，我们仍然可以通过扩展机制“继承”原有类型的功能。

扩展机制能够添加的新功能

1、实例计算属性和静态计算属性。
2、实例方法和静态方法
3、构造函数
4、下标

注：扩展机制还有另外一个优势：它扩展的类型可以是类、结构体和枚举，而继承只能是类，不能是结构体和枚举。

声明扩展

extension     类型名 {
        //添加新功能
}

声明扩展的关键字是extension, "类型名"是Swift中已有的类型，包括类、结构体和枚举，但是我们仍然可以扩展整型、浮点型、布尔型、字符串等基本数据类型，这是因为这些类型本质上也是结构体类型。

扩展计算属性

我们可以在原始类型上扩展计算属性，包括实例计算属性和静态计算属性。这些添加计算属性的定义，与普通的计算属性的定义一样。

如下：对Int进行扩展（基本数据类型全部属于结构体类型）
对Int类型增加一个计算属性errorMessage，errorMessage是一个只读的计算属性，它没有set访问函数，只有get访问函数，而且把ge t访问函数省略了

extension Int {
    var errorMessage: String {
        var errorStr = ""
            switch self {
            case -7:
                errorStr = "没有数据。"
            case -6:
                errorStr = "日期没有输入。"
            case -5:
                errorStr = "内容没有输入。"
            case -4:
                errorStr = "ID没有输入。"
            case -3:
                errorStr = "数据访问失败。"
            case -2:
                errorStr = "您的账号最多能插入10条数据。"
            case -1:
                 errorStr = "用户不存在，请到51work6.com注册。"
            default:
                errorStr = ""
          }
            return errorStr
    }
}

let message = (-7).errorMessage
print("Error Code: -7, Error Message:\(message)")

注：扩展中不仅可以定义只读计算属性, 还可以定义读写计算属性、实例计算属性和静态计算属性。但不能定义存储属性。

扩展方法

我们可以在原始类型上扩展方法，包括实例方法和静态方法。这些添加方法的定义与普通方法的定义是一样的。

extension Double {
    static var interestRate: Double = 0.0668 //利率
    func interestBy1() -> Double {
        return self * Double.interestRate
    }
    mutating func interestBy2() {
        self = self * Double.interestRate
    }
    static func interestBy3(amount: Double) -> Double {
        return interestRate * amount
    }
}

let interest1 = (10_000.00).interestBy1()
print("利息1: \(interest1)")

var interest2 = 10_000.00
interest2.interestBy2()
print("利息2:\(interest2)")

var interest3 = Double.interestBy3(amount: 10_000.00)
print("利息3:\(interest3)")

注: mutating func interestBy2() 这个方法是变异方法，mutating 使用场景：一般在枚举和结构体中试图修改它的属性的时候使用。
这个属性一般都是存储属性，修改存储属性必须要用mutating。

在本例中：这个是结构体，想要试图修改属性，而这个属性就是当前实例，需要对当前实例进行修改。
无论是修改属性还是修改当前实例的内容，由于这个是结构体，这个结构体是值类型，要想对它们进行修改，就要把方法声明成变异的方法。

扩展构造函数

• 扩展类型的时候也可以添加新的构造函数。
• 值类型与引用类型扩展有所区别。
• 扩展类的时候能向类中添加新的便利构造函数，但不能添加新的指定构造函数或析构函数。指定构造函数和析构函数只能由原始类型提供。

值类型扩展构造函数

//值类型扩展构造函数
struct Rectangle {
    var width: Double
    var height: Double

    init(width: Double, height: Double) {
        self.width = width
        self.height = height
    }
}

extension Rectangle {
    init(length: Double) {
        self.init(width: length, height: length)
    }
}

var rect = Rectangle(width: 320.0, height: 480.0)
print("长方形：\(rect.width) x \(rect.height)")

var square = Rectangle(length: 500.0)
print("长方形：\(square.width) x \(square.height)")

引用类型扩展构造函数（扩展类的时候能向类中添加新的便利构造函数）

//引用类型扩展构造函数
class Person {
    var name: String
    var age: Int
    func description() -> String {
        return "\(name) 年龄是：\(age)"
    }

    init(name: String, age: Int) {
        self.name = name
        self.age = age
    }
}

extension Person {
    convenience init(name: String) {
        self.init(name: name, age: 8)
    }
}

let p1 = Person(name: "张三")
print("Person1: \(p1.description())")

let p2 = Person(name: "Tony", age: 28)
print("Person2: \(p2.description())")

扩展下标

我们可以把下标认为是特殊的属性，可以实现索引访问属性。我们可以在原始类型的基础上扩展下标功能。

功能：扩展String类型，增加了一个下标，增加下标之后，这个String类型就可以通过下标访问其中的单个字符

//扩展下标
extension String {
    subscript(index: Int) -> String {
        if index > self.count {
            return ""
        }
        var c: String = ""
        var i = 0
        for characters in self {
            if i == index {
                c = String(characters)
                break
            }
            i+=1
        }
        return c
    }
}

let s = "Hello World"
print(s[0]) //输出：H
print("ABC"[2]) //输出：C