21.Swift学习之扩展

扩展

• 为现有的类、结构体、枚举类型、协议添加了新功能。扩展和 Objective-C 中的分类类似。
• 扩展可以：
  • 添加计算属性；
  • 定义方法；
  • 提供新的构造函数；
  • 使现有的类型遵循某协议
• Swift中使用extension关键字实现扩展

语法

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

扩展可以使已有的类型遵循一个或多个协议。在这种情况下，协议名的书写方式与类或结构体完全一样：

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

扩展计算属性

• 扩展可以向已有的类型添加计算实例属性和计算类型属性。

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 } } 25.0.km 25.0.m 25.0.cm 25.0.mm 3.0.ft 

扩展构造函数

• 扩展可向已有的类型添加新的初始化器

extension CGRect{ init(center: CGPoint, size: CGSize) { let originX = center.x - size.width * 0.5 let originY = center.y - size.height * 0.5 self.init(x: originX, y: originY, width: size.width, height: size.height) } } let centerRect = CGRect(center: CGPoint(x: 100.0, y: 100.0), size: CGSize(width: 100.0, height: 100.0)) 

扩展方法

• 扩展可以为已有的类型添加新的实例方法和类型方法。

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

扩展mutating方法

• 扩展结构体，如果要修改里面的值，必须在方法面前加上mutating

extension Int { mutating func square() { self = self * self } } //必须用变量，因为它的值要被修改 var someInt = 3 someInt.square() 

使现有的类型遵循某协议

class ViewController : UIViewController{ } extension ViewController : UITableViewDataSource{ func tableView(_ tableView: UITableView, numberOfRowsInSection section: Int) -> Int { return 10 } func tableView(_ tableView: UITableView, cellForRowAt indexPath: IndexPath) -> UITableViewCell { return UITableViewCell() } } 

面向协议编程

针对某个需要实现的功能，可以使用协议定义出接口，然后利用协议扩展提供默认的实现。需要这个功能，只需要声明遵守了这个协议即可，遵守某个协议的对象调用协议声明的方法时，如果类本身没有提供实现，协议扩展提供的默认实现会被调用。

• 案例一

protocol Eat { func eat() } class Person: Eat { func eat() { print("吃饭了") } } var p = Person() p.eat() 

• 改进

extension Eatable { func eat() { print("吃饭了") } } class Person: Eatable { } var p = Person() p.eat() 

• 再次修改

class Person: Eatable { func eat() { print("人要吃饭了") } } var p = Person() p.eat() 

• 案例二

protocol Coder { var haveFun:Bool {get set} var ownMoney:Bool {get set} } protocol Swifter { var codingLevel:Int {get set} } struct CoderA : Coder { var name:String var haveFun: Bool var ownMoney: Bool } struct CoderB : Coder, Swifter { var name:String var haveFun: Bool = true var ownMoney: Bool = true var codingLevel: Int = 3 } struct CoderC : Coder, Swifter{ var name:String var haveFun: Bool = true var ownMoney: Bool = true var codingLevel: Int = 5 } 

可以发现CoderB与CoderC有冗余

• 改进

protocol Coder { var haveFun:Bool {get set} var ownMoney:Bool {get set} } protocol Swifter { var codingLevel:Int {get set} } //where限定条件 extension Coder where Self:Swifter { var haveFun:Bool { return true} var ownMoney:Bool { return true} } struct CoderA : Coder { var name:String var haveFun: Bool var ownMoney: Bool } struct CoderB : Coder, Swifter { var name:String var haveFun: Bool var ownMoney: Bool var codingLevel: Int = 3 } struct CoderC : Coder, Swifter{ var name:String var haveFun: Bool var ownMoney: Bool var codingLevel: Int = 5 }