iOS开发之Swift篇（13）—— 扩展

版本

Xcode 11.3.1
Swift 5.1.3

扩展

扩展可以给一个现有的类、结构体、枚举、还有协议添加新的功能，但是不能重写已有的功能。
Swift 中的扩展可以：

• 添加计算型属性（不能添加存储型属性）
• 定义实例方法和类方法
• 提供新的构造器
• 定义下标
• 定义和使用新的嵌套类型
• 使一个已有类型符合某个协议

语法

extension SomeType {
  // 在这里给 SomeType 添加新的功能
}

计算型属性

扩展可以给现有类型添加计算型实例属性和计算型类属性。
1. 计算型实例属性

extension Int {
    
    var add: Int {
        return self + 10
    }
    
    var sub: Int {
        return self - 10
    }
}

let addition = 15.add
let substraction = 15.sub
print(addition, substraction)
// 25 5

2. 计算型类属性

struct AA {
    static var num = 10
}

extension AA {
    static var aad: Int {
        return self.num + 10
    }
}

let aa = 5.add
print(aa)
// 15

之前说过， 扩展不能添加存储型属性。比如在上面的例子中，给 AA 的类属性 num 添加如下扩展，会导致编译报错：

    static var num: Int {
        return self.num + 10
    }

方法

扩展可以给现有类型添加新的实例方法和类方法。
1. 实例方法

class Counter {
    
    var count = 0
}

extension Counter {
    func increment() -> Int {
        return count + 1
    }
}

let counter = Counter()
let currentCount = counter.increment()
print(currentCount)
// 1

2. 类方法

extension Int {
    static func printFunc() {
        print("我被打印了")
    }
}

Int.printFunc()
// 我被打印了

构造器

扩展可以向已有类型添加新的构造器。
扩展可以向类中添加新的便利构造器，但是它们不能向类中添加新的指定构造器或析构函数 deinit() 。

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()
}

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

let centerRect = Rect(center: Point(x: 5.0, y: 5.0), size: Size(width: 4.0, height: 4.0))
print(centerRect.origin)
// Point(x: 3.0, y: 3.0)

因为 Rect 结构体给所有的属性 (origin和size) 都提供了默认值，所以它自动获得了一个默认构造器和一个成员构造器。这些构造器可以用来创建新的 Rect 实例。
这个新的构造器首先根据提供的 center 和 size 计算一个适当的原点。然后这个构造器调用结构体自带的成员构造器 init(origin:size:)，它会将新的 origin 和 size 值储存在适当的属性中。

下标

扩展可以给现有的类型添加新的下标。
下面的例子中，对 Swift 的 Int 类型添加了一个整数类型的下标。下标 [n] 从数字右侧开始，返回小数点后的第 n 位：

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

print(123[0])
// 3
print(123[1])
// 2
print(123[2])
// 1

嵌套类型

扩展可以向已有的类、结构体和枚举添加新的嵌套类型：

extension Int {
    
   enum calculate
   {
      case add
      case sub
      case mult
   }

   var print: calculate {
      switch self
      {
         case 0:
            return .add
         case 1:
            return .sub
         default:
            return .mult
       }
   }
}

func result(numb: [Int]) {
   for i in numb {
      switch i.print {
         case .add:
            print("10")
          case .sub:
            print("20")
         case .mult:
            print("30")

      }
   }
}

result(numb: [1, 2, 0])
/**
 20
 30
 10
 */