Swift 擴展(Extension)總結

概要

擴展是給已經存在的類（class），結構體（structure），枚舉類型（enumeration）和協議（protocol）增加新的功能。類似Objective-C中的Category，不同的是，Extension沒有名字。擴展可以做以下事情：

• 增加計算實例屬性和計算類型屬性
• 定義實例方法和類型方法
• 提供新的初始化器
• 定義下標
• 定義和使用新的內置類型
• 讓一個存在的類型服從一個協議

註：擴展可以增加新的功能，但是不能覆蓋已有的功能

擴展的語法

使用extension關鍵字，如下所示。

extension SomeType{
   //new functionality to add to SomeType goes here
}

extension SomeType: SomeProtocol, AnotherProtocol{
   //implementation of protocol requirements goes here
}

註意：如果你給已經存在的類型擴展了新的功能，不管你在什麼時候擴展，儘管你是在實例定義之後再擴展的，那麼這個類型所有的實例都可以使用這個新功能，

計算屬性

擴展可以給已經存在的類型增加計算實例屬性和計算類型屬性，如以下擴展了Double類型：

extension Double{

   var km : Double { return self * 1000.0 }
   var m: Double { return self }
   var cm: Double { return self / 100.0 }
   var mm: Double { return self / 1000.0 }
   var ft: Double { return self / 3.28084 }
}

let oneInch = 25.4.mm;
print("One inch is \(oneInch)meters")

構造方法（Initializers）

可以擴展一個類型實現自定義構造方法，但是只能增加便利構造方法（Convenience Initializers），不能增加指定構造方法(Designated Initializers)。

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{

   funcrepetitions(task: () -> Void) {
     for _ in 0..<self {
          task()
       }
    }
}

4.repetitions { //尾隨閉包
   print("hello")
}

可變實例方法（Mutating Instance Methods）

擴展增加的實例方法可以修改實例本身。結構體和枚舉類型中的方法如果想要修改實例本身或者屬性的話需要用mutating來修飾方法，所以擴展這樣的方法也需要加mutating。

extension Int{

   mutating func square() {
     self = self * self
     print(self)
    }

}

var someInt = 3
someInt.square()

下標

擴展可以給存在的類型增加新的下標，如下的例子：

extension Int{

   subscript(digitIndex: Int) -> Int {
     var decimalBase = 1
     for _ in 0..<digitIndex {
       decimalBase *= 10
     }
     return(self/ decimalBase) % 10
    }

}

746381295[0] // returns 5
746381295[1] // returns 9
746381295[2] // returns 2
746381295[8] // returns 7

內置類型

擴展可以給存在的類、結構體和枚舉類型增加新的內置類型。

extension Int{
   enumKind {
     casenegative, zero, positive
    }

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

}

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])

// Prints "+ + - 0 - 0 +

作者：泗哥
鏈接：https://www.jianshu.com/p/edf4cad8dfd5
來源：簡書
著作權歸作者所有。商業轉載請聯繫作者獲得授權，非商業轉載請註明出處。