扩展介绍
扩展 就是为一个已有的类、结构体、枚举类型或者协议类型添加新功能。这包括在没有权限获取原始源代码的情况下扩展类型的能力(即 逆向建模 )。扩展和 Objective-C 中的分类类似。
OC与Swift对比
与 Objective-C 不同的是,Swift 的扩展没有名字。
扩展的功能
- 添加计算型属性和计算型类型属性
- 定义实例方法和类型方法
- 提供新的构造器
- 定义下标
- 定义和使用新的嵌套类型
- 是一个已有类型符合某个协议
- 对协议进行扩展,提供协议要求的实现,或者添加额外的功能。
扩展的注意点
- 扩展可以为一个类型添加新的功能,但是不能重写已有的功能
- 如果你通过扩展为一个已有类型添加新功能,那么新功能对该类型的所有已有实例都是可用的,即使它们是在这个扩展定义之前创建的。
- 扩展可以添加新的计算型属性,但是不可以添加存储型属性,也不可以为已有属性添加属性观察器。
- 扩展能为类添加新的便利构造器,但是它们不能为类添加新的指定构造器或析构器。指定构造器和析构器必须总是由原始的类实现来提供。
扩展的语法
使用关键字 extension 来声明扩展:
extension SomeType {
// 为 SomeType 添加的新功能写到这里
}
扩展计算型属性
扩展可以为已有类型添加计算型实例属性和计算型类型属性。
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
}
扩展实现协议
通过扩展使一个已有类型满足一个或多个协议
extension SomeType: SomeProtocol, AnotherProctocol {
// 协议实现写到这里
}
扩展构造器
- 扩展能为类添加新的便利构造器,但是它们不能为类添加新的指定构造器或析构器。指定构造器和析构器必须总是由原始的类实现来提供。
类类型扩展构造器规则比较简单,这里不再讲解,值类型扩展构造器比较复杂,下面详细说明。 - 如果结构体(值类型)或类(类类型)的所有属性都有默认值,同时没有自定义的构造器,那么swift会给这些结构体或类提供一个默认构造器,这个默认构造器将简单地创建一个所有属性值都设置为默认值的实例。
- 值类型如果定义了一个自定义的构造器,那么你将无法访问默认构造器,以及无法访问逐一成员构造器。
- 如果你使用扩展为一个值类型添加构造器,且该值类型的原始实现中未定义任何定制的构造器时,你可以在扩展中的构造器里调用逐一成员构造器。如果该值类型为所有存储型属性提供了默认值,你还可以在扩展中的构造器里调用默认构造器。
具体例子
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 结构体:
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)
}
}
扩展方法
扩展可以为已有类型添加新的实例方法和类型方法。
extension Int {
func repetitions(task: () -> Void) {
for _ in 0..<self {
task()
}
}
}
扩展可变实例方法
通过扩展添加的实例方法也可以修改该实例本身。结构体和枚举类型中修改 self 或其属性的方法必须将该实例方法标注为 mutating,正如来自原始实现的可变方法一样。
下面的例子为 Swift 的 Int 类型添加了一个名为 square 的可变方法,用于计算原始值的平方值:
extension Int {
mutating func square() {
self = self * self
}
}
var someInt = 3
someInt.square()
// someInt 的值现在是 9
扩展下标
扩展可以为已有类型添加新下标。
extension Int {
subscript(var digitIndex: Int) -> Int {
var decimalBase = 1
while digitIndex > 0 {
decimalBase *= 10
--digitIndex
}
return (self / decimalBase) % 10
}
}