擴展 就是為一個已有的類、結(jié)構(gòu)體碘耳、枚舉類型或者協(xié)議類型添加新功能显设。這包括在沒有權(quán)限獲取原始源代碼的情況下擴展類型的能力。擴展和 Objective-C 中的分類類似辛辨。(與 Objective-C 不同的是捕捂,Swift 的擴展沒有名字。)
- Swift 中的擴展可以:
1斗搞、添加計算型屬性和計算型`類`型屬性
2绞蹦、定義`實例`方法和`類`型方法
3、提供新的便歷構(gòu)造器
4榜旦、定義下標
5幽七、定義和使用新的嵌套類型
6、使一個已有類型符合某個協(xié)議
注意
擴展可以為一個類型添加新的功能溅呢,但是不能重寫已有的功能澡屡。
一、擴展語法
- 使用關(guān)鍵字
extension
來聲明擴展:
extension SomeType {
// 為 SomeType 添加的新功能寫到這里
}
- 可以通過擴展來擴展一個已有類型咐旧,使其采納一個或多個協(xié)議驶鹉。在這種情況下,無論是類還是結(jié)構(gòu)體铣墨,協(xié)議名字的書寫方式完全一樣:
extension SomeType: SomeProtocol, AnotherProctocol {
// 協(xié)議實現(xiàn)寫到這里
}
注意
如果你通過擴展為一個已有類型添加新功能室埋,那么新功能對該類型的所有已有實例都是可用的,即使它們是在這個擴展定義之前創(chuàng)建的伊约。
二姚淆、計算型屬性
- 擴展可以為已有類型添加計算型實例屬性和計算型類型屬性。
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
關(guān)鍵字。它們的返回值是Double
超埋,而且可以用于所有接受Double
值的數(shù)學計算中:
let aMarathon = 42.km + 195.m
print("A marathon is \(aMarathon) meters long")
// 打印 “A marathon is 42195.0 meters long”
注意
擴展可以添加新的計算型屬性搏讶,但是不可以添加存儲型屬性,也不可以為已有屬性添加屬性觀察器霍殴。
三媒惕、構(gòu)造器
- 擴展可以為已有類型添加新的構(gòu)造器。這可以讓你擴展其它類型来庭,將你自己的定制類型作為其構(gòu)造器參數(shù)妒蔚,或者提供該類型的原始實現(xiàn)中未提供的額外初始化選項。
- 擴展能為類添加新的便利構(gòu)造器,但是它們不能為類添加新的指定構(gòu)造器或析構(gòu)器面睛。指定構(gòu)造器和析構(gòu)器必須總是由原始的類實現(xiàn)來提供絮蒿。
注意:
如果你使用擴展為一個值類型添加構(gòu)造器尊搬,
同時該值類型的原始實現(xiàn)中未定義任何定制的構(gòu)造器且所有存儲屬性提供了默認值叁鉴,
那么我們就可以在擴展中的構(gòu)造器里調(diào)用默認構(gòu)造器和逐一成員構(gòu)造器。
如果你把定制的構(gòu)造器寫在值類型的原始實現(xiàn)中佛寿,上述規(guī)則將不再適用幌墓。
- 下面的例子定義了一個用于描述幾何矩形的結(jié)構(gòu)體
Rect
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()
}
- 因為結(jié)構(gòu)體
Rect
未提供定制的構(gòu)造器,因此它會獲得一個逐一成員構(gòu)造器
冀泻。又因為它為所有存儲型屬性提供了默認值常侣,它又會獲得一個默認構(gòu)造器
。
let defaultRect = Rect()
let memberwiseRect = Rect(origin: Point(x: 2.0, y: 2.0),
size: Size(width: 5.0, height: 5.0))
- 你可以提供一個額外的接受指定中心點和大小的構(gòu)造器來擴展
Rect
結(jié)構(gòu)體:
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)
}
}
- 這個新的構(gòu)造器首先根據(jù)提供的
center
和size
的值計算一個合適的原點弹渔。然后調(diào)用該結(jié)構(gòu)體的逐一成員構(gòu)造器init(origin:size:)
胳施,該構(gòu)造器將新的原點和大小的值保存到了相應的屬性中:
let centerRect = Rect(center: Point(x: 4.0, y: 4.0),
size: Size(width: 3.0, height: 3.0))
// centerRect 的原點是 (2.5, 2.5),大小是 (3.0, 3.0)
注意
如果你使用擴展提供了一個新的構(gòu)造器肢专,你依舊有責任確保構(gòu)造過程能夠讓實例完全初始化舞肆。
四、方法
- 擴展可以為已有類型添加新的實例方法和類型方法博杖。下面的例子為
Int
類型添加了一個名為repetitions
的實例方法:
extension Int {
func repetitions(task: () -> Void) {
for _ in 0..<self {
task()
}
}
}
- 定義該擴展之后椿胯,你就可以對任意整數(shù)調(diào)用
repetitions(task:)
方法,將閉包中的任務執(zhí)行整數(shù)對應的次數(shù):
3.repetitions {
print("Hello!")
}
// Hello!
// Hello!
// Hello!
可變實例方法
- 通過擴展添加的實例方法也可以修改該實例本身剃根。結(jié)構(gòu)體和枚舉類型中修改
self
或其屬性的方法必須將該實例方法標注為mutating
哩盲,正如來自原始實現(xiàn)的可變方法一樣。
extension Int {
mutating func square() {
self = self * self
}
}
var someInt = 3
someInt.square()
// someInt 的值現(xiàn)在是 9
五狈醉、下標
- 擴展可以為已有類型添加新下標廉油。這個例子為
Swift
內(nèi)建類型Int
添加了一個整型下標。該下標[n]
返回十進制數(shù)字從右向左數(shù)的第n
個數(shù)字:
extension Int {
subscript(digitIndex: Int) -> Int {
var decimalBase = 1
for _ in 0..<digitIndex {
decimalBase *= 10
}
return (self / decimalBase) % 10
}
}
746381295[0]
// 返回 5
746381295[1]
// 返回 9
746381295[2]
// 返回 2
746381295[8]
// 返回 7
六苗傅、嵌套類型
- 擴展可以為已有的類娱两、結(jié)構(gòu)體和枚舉添加新的嵌套類型:
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
}
}
}
- 現(xiàn)在,這個嵌套枚舉可以和任意
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])
// 打印 “+ + - 0 - 0 + ”
注意
由于已知number.kind
是Int.Kind
類型金吗,因此在switch
語句中十兢,Int.Kind
中的所有成員值都可以使用簡寫形式,例如使用.negative
而不是Int.Kind.negative
摇庙。