Swift 扩展(extension)详解
2018-08-13 本文已影响1197人
WSJay
在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..<self {
task()
}
}
}
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..<digitIndex {
decimalBase *= 10
}
return (self / decimalBase) % 10
}
}
746381295[0]
// returns 5
746381295[1]
// returns 9
746381295[2]
// returns 2
746381295[8]
// returns 7
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 + "
