Swift 扩展(extension)详解

在Swift中扩展可以:

  • 添加计算实例属性和计算类型属性;
  • 定义实例方法和类型方法;
  • 提供新构造器;
  • 定义下标;
  • 定义和使用新的嵌套类型;
  • 使现有类型符合协议;

注意:扩展可以向类型添加新功能,但不能覆盖现有功能。

1.扩展的语法

定义扩展使用关键字extension

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

注意:如果定义扩展向现有类型添加新功能,那么该新功能将在该类型的所有现有实例上可用,即使它们是在定义扩展之前创建的。

2.计算属性

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

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")
// Prints "One inch is 0.0254 meters"
let threeFeet = 3.ft
print("Three feet is \(threeFeet) meters")
// Prints "Three feet is 0.914399970739201 meters"

这些属性是只读的计算属性,因此为了简洁起见,不使用get关键字来表示它们。它们的返回值是Double类型的,可以在接受Double类型的数学计算中使用:

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

注意: 扩展可以添加新的计算属性,但它们不能添加存储属性,或向现有属性添加属性观察者。

3.构造器

  • 扩展可以向类添加新的便利构造器,但是它们不能向类添加新的指定构造器或析构器。指定构造器或析构器必须始终由原始类实现提供。
  • 如果使用扩展将构造器添加到值类型中,该值类型为其所有存储属性提供默认值,并且没有任何自定义构造器,则可以在扩展的构造器中调用该值类型的默认构造器和全能构造器。
  • 如果使用扩展将构造器添加到声明在另一个模块中的结构体中,则新的构造器只有从定义的模块中调用了构造器才能访问self
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)
    }
}

let centerRect = Rect(center: Point(x: 4.0, y: 4.0),
                      size: Size(width: 3.0, height: 3.0))
// centerRect's origin is (2.5, 2.5) and its size is (3.0, 3.0)

注意:如果使用扩展来提供一个新初始化器,要确保构造器完成之后,每个实例都被完全初始化。

4.方法

(1). 向已存在类型添加新的实例方法和类型方法;

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

(2).添加可变的实例方法
通过扩展增加的实例方法可以修改实例本身。如果结构体和枚举的方法要修改self或其属性,则需要使用mutating关键字标明。

extension Int {
    mutating func square() {
        self = self * self
    }
}
var someInt = 3
someInt.square()
// someInt is now 9

5.下标

  • 扩展可以向已存在的类型添加下标
extension Int {
    subscript(digitIndex: Int) -> Int {
        var decimalBase = 1
        for _ in 0..

6.嵌套类型

  • 扩展可以向已存在的类,结构体和枚举添加新的嵌套类型
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类型值的嵌套枚举类型:

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 + "

7.其他专题模块

Swift 4.2 基础专题详解

你可能感兴趣的:(Swift 扩展(extension)详解)