Swift - 擴展

擴展 就是為一個已有的類、結構體、枚舉類型或者協議類型添加新功能。這包括在沒有權限獲取原始源代碼的情況下擴展類型的能力(即 逆向建模 )。擴展和 Objective-C 中的分類類似。(與 Objective-C 不同的是,Swift 的擴展沒有名字。)

Swift 中的擴展可以:

  • 添加計算型屬性和計算型類型屬性
  • 定義實例方法和類型方法
  • 提供新的構造器
  • 定義下標
  • 定義和使用新的嵌套類型
  • 使一個已有類型符合某個協議

在 Swift 中,你甚至可以對協議進行擴展,提供協議要求的實現,或者添加額外的功能,從而可以讓符合協議的類型擁有這些功能。你可以從協議擴展獲取更多的細節。

注意
擴展可以為一個類型添加新的功能,但是不能重寫已有的功能。

擴展語法

使用關鍵字 extension 來聲明擴展:

extension SomeType {
    // 為 SomeType 添加的新功能寫到這里
}

可以通過擴展來擴展一個已有類型,使其采納一個或多個協議。在這種情況下,無論是類還是結構體,協議名字的書寫方式完全一樣:

extension SomeType: SomeProtocol, AnotherProctocol {
    // 協議實現寫到這里
}

通過這種方式添加協議一致性的詳細描述請參閱利用擴展添加協議一致性

注意
如果你通過擴展為一個已有類型添加新功能,那么新功能對該類型的所有已有實例都是可用的,即使它們是在這個擴展定義之前創建的。

計算型屬性

擴展可以為已有類型添加計算型實例屬性和計算型類型屬性。下面的例子為 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")
// 打印 “One inch is 0.0254 meters”
let threeFeet = 3.ft
print("Three feet is \(threeFeet) meters")
// 打印 “Three feet is 0.914399970739201 meters”

這些計算型屬性表達的含義是把一個 Double 值看作是某單位下的長度值。即使它們被實現為計算型屬性,但這些屬性的名字仍可緊接一個浮點型字面值,從而通過點語法來使用,并以此實現距離轉換。

在上述例子中,Double1.0 用來表示“1米”。這就是為什么計算型屬性 m 返回 self,即表達式 1.m 被認為是計算 Double1.0

其它單位則需要一些單位換算。一千米等于 1,000 米,所以計算型屬性 km 要把值乘以 1_000.00 來實現千米到米的單位換算。類似地,一米有 3.28024 英尺,所以計算型屬性 ft 要把對應的 Double 值除以 3.28024 來實現英尺到米的單位換算。

這些屬性是只讀的計算型屬性,為了更簡潔,省略了 get 關鍵字。它們的返回值是 Double,而且可以用于所有接受 Double 值的數學計算中:

let aMarathon = 42.km + 195.m
print("A marathon is \(aMarathon) meters long")
// 打印 “A marathon is 42195.0 meters long”

注意
擴展可以添加新的計算型屬性,但是不可以添加存儲型屬性,也不可以為已有屬性添加屬性觀察器。

構造器

擴展可以為已有類型添加新的構造器。這可以讓你擴展其它類型,將你自己的定制類型作為其構造器參數,或者提供該類型的原始實現中未提供的額外初始化選項。

擴展能為類添加新的便利構造器,但是它們不能為類添加新的指定構造器或析構器。指定構造器和析構器必須總是由原始的類實現來提供。

注意
如果你使用擴展為一個值類型添加構造器,同時該值類型的原始實現中未定義任何定制的構造器且所有存儲屬性提供了默認值,那么我們就可以在擴展中的構造器里調用默認構造器和逐一成員構造器。
正如在值類型的構造器代理中描述的,如果你把定制的構造器寫在值類型的原始實現中,上述規則將不再適用。

下面的例子定義了一個用于描述幾何矩形的結構體 Rect。這個例子同時定義了兩個輔助結構體 SizePoint,它們都把 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)
    }
}

這個新的構造器首先根據提供的 centersize 的值計算一個合適的原點。然后調用該結構體的逐一成員構造器 init(origin:size:),該構造器將新的原點和大小的值保存到了相應的屬性中:

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)

注意
如果你使用擴展提供了一個新的構造器,你依舊有責任確保構造過程能夠讓實例完全初始化。

方法

擴展可以為已有類型添加新的實例方法和類型方法。下面的例子為 Int 類型添加了一個名為 repetitions 的實例方法:

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

這個 repetitions(task:) 方法接受一個 () -> Void 類型的單參數,表示沒有參數且沒有返回值的函數。

定義該擴展之后,你就可以對任意整數調用 repetitions(task:) 方法,將閉包中的任務執行整數對應的次數:

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

可以使用尾隨閉包讓調用更加簡潔:

3.repetitions {
    print("Goodbye!")
}
// Goodbye!
// Goodbye!
// Goodbye!

可變實例方法

通過擴展添加的實例方法也可以修改該實例本身。結構體和枚舉類型中修改 self 或其屬性的方法必須將該實例方法標注為 mutating,正如來自原始實現的可變方法一樣。

下面的例子為 Swift 的 Int 類型添加了一個名為 square 的可變方法,用于計算原始值的平方值:

extension Int {
    mutating func square() {
        self = self * self
    }
}
var someInt = 3
someInt.square()
// someInt 的值現在是 9

下標

擴展可以為已有類型添加新下標。這個例子為 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
    }
}
746381295[0]
// 返回 5
746381295[1]
// 返回 9
746381295[2]
// 返回 2
746381295[8]
// 返回 7

如果該 Int 值沒有足夠的位數,即下標越界,那么上述下標實現會返回 0,猶如在數字左邊自動補 0

746381295[9]
// 返回 0,即等同于:
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])
// 打印 “+ + - 0 - 0 + ”

函數 printIntegerKinds(_:) 接受一個 Int 數組,然后對該數組進行迭代。在每次迭代過程中,對當前整數的計算型屬性 kind 的值進行評估,并打印出適當的描述。

注意
由于已知 number.kindInt.Kind 類型,因此在 switch 語句中,Int.Kind 中的所有成員值都可以使用簡寫形式,例如使用 . Negative 而不是 Int.Kind.Negative

最后編輯于
?著作權歸作者所有,轉載或內容合作請聯系作者
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發布,文章內容僅代表作者本人觀點,簡書系信息發布平臺,僅提供信息存儲服務。
  • 序言:七十年代末,一起剝皮案震驚了整個濱河市,隨后出現的幾起案子,更是在濱河造成了極大的恐慌,老刑警劉巖,帶你破解...
    沈念sama閱讀 228,443評論 6 532
  • 序言:濱河連續發生了三起死亡事件,死亡現場離奇詭異,居然都是意外死亡,警方通過查閱死者的電腦和手機,發現死者居然都...
    沈念sama閱讀 98,530評論 3 416
  • 文/潘曉璐 我一進店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人,你說我怎么就攤上這事。” “怎么了?”我有些...
    開封第一講書人閱讀 176,407評論 0 375
  • 文/不壞的土叔 我叫張陵,是天一觀的道長。 經常有香客問我,道長,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 62,981評論 1 312
  • 正文 為了忘掉前任,我火速辦了婚禮,結果婚禮上,老公的妹妹穿的比我還像新娘。我一直安慰自己,他們只是感情好,可當我...
    茶點故事閱讀 71,759評論 6 410
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著,像睡著了一般。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發上,一...
    開封第一講書人閱讀 55,204評論 1 324
  • 那天,我揣著相機與錄音,去河邊找鬼。 笑死,一個胖子當著我的面吹牛,可吹牛的內容都是我干的。 我是一名探鬼主播,決...
    沈念sama閱讀 43,263評論 3 441
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼!你這毒婦竟也來了?” 一聲冷哼從身側響起,我...
    開封第一講書人閱讀 42,415評論 0 288
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎,沒想到半個月后,有當地人在樹林里發現了一具尸體,經...
    沈念sama閱讀 48,955評論 1 336
  • 正文 獨居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內容為張勛視角 年9月15日...
    茶點故事閱讀 40,782評論 3 354
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發現自己被綠了。 大學時的朋友給我發了我未婚夫和他白月光在一起吃飯的照片。...
    茶點故事閱讀 42,983評論 1 369
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖,靈堂內的尸體忽然破棺而出,到底是詐尸還是另有隱情,我是刑警寧澤,帶...
    沈念sama閱讀 38,528評論 5 359
  • 正文 年R本政府宣布,位于F島的核電站,受9級特大地震影響,放射性物質發生泄漏。R本人自食惡果不足惜,卻給世界環境...
    茶點故事閱讀 44,222評論 3 347
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧,春花似錦、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 34,650評論 0 26
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至,卻和暖如春,著一層夾襖步出監牢的瞬間,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 35,892評論 1 286
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人。 一個月前我還...
    沈念sama閱讀 51,675評論 3 392
  • 正文 我出身青樓,卻偏偏與公主長得像,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當晚...
    茶點故事閱讀 47,967評論 2 374

推薦閱讀更多精彩內容