Swift 扩展

前言

• 扩展就是给一个现存类、结构体、枚举或者协议添加新的属性或者方法，无需修改目标的源代码，就可以把想要的代码加到目标上面。

  • 扩展可以用来扩展现有类型的计算属性、构造器、方法和下标。
  • 不能添加一个已经存在的方法或者属性。
  • 添加的属性不能是存储属性，只能是计算属性。

• 在定义扩展时，在类型名前使用关键字 extension。

  extension 某个现有类型 {
  
      // 增加新的功能
  }

1、扩展计算属性

• 扩展计算属性

  class Transport {
  
      var scope: String
  
      init(scope: String) {
          self.scope = scope
      }
  }

  extension Transport {
  
      // 扩展计算属性
      var extProperty: String {
  
          get {
              return scope
          }
      }
  }

  var myTrans = Transport(scope: "ludi")
  print(myTrans.extProperty)                          // ludi

2、扩展构造器

• 扩展构造器

  class Transport {
  
      var price = 30
  
      var scope: String
  
      init(scope: String) {
          self.scope = scope
      }
  }

  extension Transport {
  
      // 扩展构造器
      convenience init(price: Int, scope: String) {
  
          self.init(scope: scope)
          self.price = price
      }
  }

  var myTra1 = Transport(price: 50, scope: "ludi")    // 使用扩展构造器，价格为 50
  var myTra2 = Transport(scope: "haiyang")            // 使用原构造器，介个属性的值仍为 30

3、扩展方法

• 使用扩展增加方法的做法非常灵活，除了可在自定义的类中扩展方法外，还可以扩展基本数据类型的方法，比如扩展整数类型。

  // 扩展基本数据类型
  extension Int {
  
      func calculate() -> Int {
          return self * 2
      }
  }

  var i = 3
  i.calculate()       // 6    

4、扩展下标

• 扩展下标

  extension Int {
  
      // 扩展下标
      subscript(num: Int) -> Int {
          return self * num
      }
  }

  var i = 3
  i[2]                    // 6

5、扩展协议

• 详细讲解见 “Swift 协议” 章节。