- 扩展可以给一个现有的类,结构体,枚举,还有协议添加新的功能。
- 拥有不需要访问被扩展类型源代码就能完成扩展的能力(即逆向建模)。(扩展和 Objective-C 的分类很相似,与 Objective-C 分类不同的是,Swift 扩展是没有名字的。)
Swift 中的扩展可以:
- 添加计算型实例属性和计算型类属性
- 定义实例方法和类方法
- 提供新的构造器
- 定义下标
- 定义和使用新的嵌套类型
- 使已经存在的类型遵循(conform)一个协议
在 Swift 中,你甚至可以扩展协议以提供其需要的实现,或者添加额外功能给遵循的类型所使用。
注意:扩展可以给一个类型添加新的功能,但是不能重写已经存在的功能。
1. 扩展的语法
使用 extension 关键字声明扩展:
extension SomeType {
// 在这里给 SomeType 添加新的功能
}
扩展可以扩充一个现有的类型,给它添加一个或多个协议。协议名称的写法和类或者结构体一样:
extension SomeType: SomeProtocol, AnotherProtocol {
// 协议所需要的实现写在这里
}
注意:对一个现有的类型,如果你定义了一个扩展来添加新的功能,那么这个类型的所有实例都可以使用这个新功能,包括那些在扩展定义之前就存在的实例。
2. 计算型属性
- 扩展可以给现有类型添加计算型实例属性和计算型类属性。
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") // 打印“One inch is 0.0254 meters”
let threeFeet = 3.ft
print("Three feet is \(threeFeet) meters") // 打印“Three feet is 0.914399970739201 meters”
这些属性都是只读的计算型属性,所以为了简便,它们的表达式里面都不包含 get 关键字。
- 它们使用 Double 作为返回值类型,并可用于所有接受 Double 类型的数学计算中:
let aMarathon = 42.km + 195.m
print("A marathon is \(aMarathon) meters long")
// 打印“A marathon is 42195.0 meters long”
注意
1. 扩展可以添加新的计算属性,但是它们不能添加存储属性
2. 不能向现有的属性添加属性观察者。
3. 构造器
- 扩展可以给现有的类型添加新的构造器。
- 它使你可以把自定义类型作为参数来供其他类型的构造器使用
- 或者在类型的原始实现上添加额外的构造选项。
- 扩展可以给一个类添加新的便利构造器。
- 但是它们不能给类添加新的指定构造器或者析构器。
- 指定构造器和析构器必须始终由类的原始实现提供。
- 如果你使用扩展给一个值类型添加构造器只是用于给所有的存储属性提供默认值,并且没有定义任何自定义构造器,那么你可以在该值类型扩展的构造器中使用默认构造器和成员构造器。如果你把构造器写到了值类型的原始实现中,那么就不属于在扩展中添加构造器。
- 如果你使用扩展给另一个模块中定义的结构体添加构造器,那么新的构造器直到定义模块中使用一个构造器之前,不能访问 self。
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()
}
let defaultRect = Rect()
let memberwiseRect = Rect(origin: Point(x: 2.0, y: 2.0),
size: Size(width: 5.0, height: 5.0))
- 可以通过扩展 Rect 结构体来提供一个允许指定 point 和 size 的构造器:
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)
}
}
let centerRect = Rect(center: Point(x: 4.0, y: 4.0),
size: Size(width: 3.0, height: 3.0))
// centerRect 的 origin 是 (2.5, 2.5) 并且它的 size 是 (3.0, 3.0)
注意:如果你通过扩展提供一个新的构造器,你有责任确保每个通过该构造器创建的实例都是初始化完整的。
4. 方法
- 扩展可以给现有类型添加新的实例方法和类方法。
extension Int {
func repetitions(task: () -> Void) { // repetitions(task:) 方法仅接收一个 () -> Void 类型的参数,它表示一个没有参数没有返回值的方法。
for _ in 0..
- 定义了这个扩展之后,你可以对任意整形数值调用 repetitions(task:) 方法,来执行对应次数的任务:
3.repetitions {
print("Hello!")
}
// Hello!
// Hello!
// Hello!
4.1 可变实例方法
- 通过扩展添加的实例方法同样也可以修改(或 mutating(改变))实例本身。
- 结构体和枚举的方法,若是可以修改 self 或者它自己的属性,则必须将这个实例方法标记为 mutating,就像是改变了方法的原始实现。
extension Int {
mutating func square() {
self = self * self
}
}
var someInt = 3
someInt.square() // someInt 现在是 9
5. 下标
- 扩展可以给现有的类型添加新的下标。
- 对 Swift 的 Int 类型添加了一个整数类型的下标。下标 [n] 从数字右侧开始,返回小数点后的第 n 位:
- 123456789[0] 返回 9
- 123456789[1] 返回 8
- ……以此类推:
extension Int {
subscript(digitIndex: Int) -> Int {
var decimalBase = 1
for _ in 0..
- 如果操作的 Int 值没有足够的位数满足所请求的下标,那么下标的现实将返回 0,将好像在数字的左边补上了 0:
746381295[9]
// 返回 0,就好像你进行了这个请求:
0746381295[9]
6. 嵌套类型
- 扩展可以给现有的类,结构体,还有枚举添加新的嵌套类型:
extension Int {
enum Kind { // Kind,表示特定整数所代表的数字类型。具体来说,它表示数字是负的、零的还是正的。
case negative, zero, positive
}
var kind: Kind { // kind,它返回被操作整数所对应的 Kind 枚举 case 分支。
switch self {
case 0:
return .zero
case let x where x > 0:
return .positive
default:
return .negative
}
}
}
- 任意 Int 的值都可以使用这个嵌套类型:
/*
printIntegerKinds(_:),
使用一个 Int 类型的数组作为输入
然后依次迭代这些值
对于数组中的每一个整数,方法会检查它的 kind 计算型属性
然后打印适当的描述
*/
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])
// 打印“+ + - 0 - 0 + ”
注意:
number.kind 已经被认为是 Int.Kind 类型。
所以,在 switch 语句中所有的 Int.Kind case 分支可以被缩写,就像使用 .negative 替代 Int.Kind.negative.。