案例代码下载

扩展

扩展为现有的类，结构，枚举或协议类型添加新功能。这包括扩展无法访问源代码的类型的能力（称为追溯建模）。扩展类似于Objective-C中的类别。（与Objective-C类别不同，Swift扩展没有名称。）

Swift中的扩展可以：

• 添加计算实例属性和计算类型属性
• 定义实例方法和类型方法
• 提供新的初始化程序
• 定义下标
• 定义并使用新的嵌套类型
• 使现有类型遵守协议

在Swift中，甚至可以扩展协议来提供其需求的实现，或者添加符合类型有用的附加功能。有关更多详细信息，请参阅协议扩展。

注意： 扩展可以为类型添加新功能，但它们不能覆盖现有功能。

扩展语法

使用extension关键字声明扩展：

extension SomeType {
    // 在这里添加新功能
}

扩展可以扩展现有类型以使其遵守一个或多个协议。要添加遵守的协议，按照为类或结构编写协议名称的方式编写协议名称：

extension SomeType: SomeProtocol, AnotherProtocol {
    // 实现协议
}

在使用扩展添加遵守的协议中描述了以这种方式添加遵守的协议。

扩展可用于扩展现有泛型类型，如扩展泛型类型中所述。还可以扩展泛型类型以有条件地添加功能，如带有Where子句的泛型扩展中所述。

注意： 如果定义扩展以向现有类型添加新功能，则新功能将在该类型的所有现有实例上可用，即使它们是在定义扩展之前创建的。

计算属性

扩展可以将计算实例属性和计算类型属性添加到现有类型。此示例将五个计算实例属性添加到Swift的内置Double类型，以便为使用距离单位提供基本支持：

extension Double {
    var km: Double { return self*1_000.0 }
    var m: Double { return self }
    var cm: Double { return self/100.0 }
    var mm: Double { return self/1_000.0}
    var ft: Double { return self / 3.28084 }
}
let oneInch = 25.4.mm
print("One inch is \(oneInch) meters")
let threeFeet = 3.ft
print("Three feet is \(threeFeet) meters")

这些计算的属性表示Double值应被视为某个长度单位。尽管它们是作为计算属性实现的，但可以使用点语法将这些属性的名称附加到浮点文字值，作为使用该文字值执行距离转换的方法。

在该示例中，Double值1.0被认为表示“一米”。这就是m计算属性返回self的原因- 表达式1.m被认为是Double值1.0的计数值。

其他单位需要一些转换表示为以米为单位的值。一公里与1,000米相同，因此km计算属性将值乘以1_000.00转换为以米表示的数字。类似地，一米等于3.28084英尺，因此ft计算属性将基础Double值除以3.28084，将其从英尺转换为米。

这些属性是只读的计算属性，因此get为了简洁起见，它们在没有关键字的情况下表示。它们的返回值是Double类型，并且可以在接受Double数学计算的任何地方使用：

let aMarathon = 42.km + 195.m
print("A marathon is \(aMarathon) meters long")

注意： 扩展可以添加新的计算属性，但不能添加存储的属性，也不能将属性观察者添加到现有属性。

初始化器

扩展可以向现有类型添加新的初始化程序。可以扩展其他类型以接受自己的自定义类型作为初始化参数，或者提供未包含在原实现中的其他初始化类型的选项。

扩展可以为类添加新的便利初始化器，但是它们不能向类中添加新的指定初始化器和deinitializers。指定的初始化程序和deinitializers必须始终由原始类实现提供。

如果使用扩展为所有存储属性都提供默认值但没有任何自定义初始化程序的值类型添加初始化程序，可以在值类型扩展程序的初始化程序中调用默认初始化程序和成员初始化程序。如果值类型已将初始化程序编写为原始实现的一部分，则不会出现这种情况，如代理中值类型的初始化程序所述。

如果使用扩展将结构初始化程序添加到在另一个模块中，则新的初始化程序在从定义模块调用初始化程序之前无法访问self。

下面的示例定义了一个Rect表示几何矩形的自定义结构。该示例还定义了两个称为Size和Point支撑结构，两者都为所有其属性提供的默认值0.0：

struct Size {
    var width = 0.0, height = 0.0
}
struct Point {
    var x = 0.0, y = 0.0
}
struct Rect {
    var origin = Point()
    var size = Size()
}

由于Rect结构为其所有属性提供默认值，因此它会自动接收默认初始化程序和成员初始化程序，如默认初始化程序中所述。这些初始化程序可用于创建新Rect实例：

let defaultRect = Rect()
let memberwiseRect = Rect(origin: Point(x: 2.0, y: 2.0), size: Size(width: 5.0, height: 5.0))

可以扩展Rect结构以提供附加的初始化程序，该初始化程序采用特定的中心点和大小：

extension Rect {
    init(center: Point, size: Size) {
        let originX = center.x - (size.width / 2)
        let originY = center.y - (size.height / 2)
        self.init(origin: Point(x: originX, y: originY), size: size)
    }
}

这个新的初始化程序首先根据提供的center点和size值计算适当的原点。初始化程序然后调用结构的自动成员初始化程序init(origin:size:)，它将新的原点和大小值存储在适当的属性中：

let centerRect = Rect(center: Point(x: 4.0, y: 4.0), size: Size(width: 3.0, height: 3.0))

注意： 如果提供带有扩展的新初始化程序，则仍然有责任确保在初始化程序完成后每个实例都已完全初始化。

方法

扩展可以向现有类型添加新的实例方法和类型方法。以下示例为Int类型添加了一个叫做repetitions的新实例方法：

extension Int {
    func repetitions(tarsk: () -> Void) {
        for _ in 0..<self {
            tarsk()
        }
    }
}

repetitions(task:)方法采用() -> Void类型的单个参数，表示没有参数且不返回值的闭包。

定义此扩展后，可以在任何整数上调用repetitions(task:)方法以执行多次任务：

3.repetitions {
    print("Hello!")
}

Mutating实例方法

扩展添加的实例方法也可以修改（或改变）实例本身。修改结构和枚举自身或其属性的方法必须将实例方法标记为mutating，就像从原始实现中的Mutating方法一样。

下面的例子为Int类型添加了一个名为squareSwift的新Mutating方法，它将原始值平方：

extension Int {
    mutating func squareSwift() {
        self = self*self
    }
}
var someInt = 3
someInt.squareSwift()
print(someInt)

下标

扩展可以向现有类型添加新下标。此示例向Swift的内置Int类型添加整数下标。此下标[n]返回从右侧开始n位后的十进制数：

• 123456789[0] 返回9
• 123456789[1] 返回8
• …等等：

extension Int {
    subscript(digitIndex: Int) -> Int {
        var decimalBase = 1
        for _ in 0..<digitIndex {
            decimalBase *= 10
        }
        return (self/decimalBase)%10
    }
}
print(746381295[0])
print(746381295[1])
print(746381295[2])
print(746381295[8])

如果Int值没有足够的数字用于请求的索引，则下标实现将返回0，就好像该数字已用零填充到左侧：

print(746381295[9])
print(0746381295[9])

嵌套类型

扩展可以向现有的类，结构和枚举添加新的嵌套类型：

extension Int {
    enum Kind {
        case negative, zero, positive
    }
    var kind: Kind {
        switch self {
        case 0:
            return .zero
        case let x where x > 0:
            return .positive
        default:
            return .negative
        }
    }
}

此示例为Int类型添加了一个新的嵌套枚举。此枚举叫做Kind表示特定整数表示的数字种类。具体而言，它表示该数字是负数，零还是正数。

此示例还为Int调用添加了一个新的计算实例属性kind，该属性返回该Kind整数的相应枚举值。

嵌套枚举现在可以与任何Int值一起使用：

func printIntegerKinds(_ numbers: [Int]) {
    for number in numbers {
        switch number.kind {
        case .negative:
            print("- ", terminator: "")
        case .zero:
            print("0 ", terminator: "")
        case .positive:
            print("+ ", terminator: "")
        }
    }
    print("")
}
printIntegerKinds([3, 19, -27, 0, -6, 0, 7])

此函数printIntegerKinds(_:)采用输入Int值的数组并依次迭代这些值。对于数组中的每个整数，函数会考虑该整数的计算属性kind，并打印相应的描述。

注意： number.kind已知是Int.Kind类型。因此，所有Int.Kind案例值都可以在switch语句中以简写形式.negative写入，而不是Int.Kind.negative。