extension
扩展能为现有的 类，结构体，枚举，协议 添加功能。这包括为您无权访问的源代码扩展类型的能力。类似于OC中的categories，但是又不太一样，swift中的扩展没有名字
extension 在swift中做的事
- 可以添加计算实例属性和计算类型属性
- 定义实例方法和类型方法
- 提供新的初始化方式
- 定义下标
- 定义和使用新的嵌套类型
- 使现有类型符合协议
特点
- extension 内的代码不能被子类重写
- 可以用来扩展协议，为那些 require 的方法 提供实现，或者添加符合类型可以利用的 额外功能
- 为现有的类型 添加 新的 属性，方法 和 下标
语法
- 声明扩展 要用关键字
extension
extension SomeType {
// new functionality to add to SomeType goes here
}
- 在为已知类型扩展时也可以实现一个或多个协议，方式 与 为class，struct 实现协议一样
协议名字要放在类型名字后面，以冒号分割
extension SomeType: SomeProtocol, AnotherProtocol {
// implementation of protocol requirements goes here
}
例如：
protocol TextRepresentable {
var textualDescription: String { get }
}
//Dice 类 遵守协议 TextRepresentable，并实现协议中的方法
//这样任意一个 Dice 对象，都被认为是 TextRepresentable 对象
extension Dice: TextRepresentable {
var textualDescription: String {
return "A \(sides)-sided dice"
}
}
- 扩展一个协议
swift 协议中的方法没有可选项，也就是所有的协议方法都应该被实现，但是可以利用扩展一个协议的方式，默认实现可选方法，这样在 实现协议的类型中 就没必要去实现可选方法
// MARK: - 协议
protocol ExampleProtocol {
var simpleDescription: String { get }
//swift 协议中没有可选方法，但是可以在扩展中 对可选方法进行默认实现
//这样 实现该协议的类型 就不用实现 可选方法
var simpleDesc: String { get set }
mutating func adjust()
}
//扩展协议，实现可选方法
extension ExampleProtocol {
var simpleDesc: String {
get {
return ""
}
set {
simpleDesc = newValue
}
}
}
// 遵守协议，实现 requirements 方法
class SimpleClass: ExampleProtocol {
var simpleDescription: String = ""
func adjust() {
}
}
- 扩展通用类型
当你扩展一个通用类型的时候，你不会提供一个参数列表作为扩展的一部分，相反的，原始类型中定义的参数列表在扩展的主体内是可用的
struct Stack<Element> {
var items: [Element] = []
mutating func push(_ item: Element) {
items.append(item)
}
mutating func pop() -> Element {
return items.removeLast()
}
}
//扩展Stack，添加计算属性，获取栈顶数据
//值得注意的是，扩展中并没有定义任何类型参数列表
// 而Stack 中已经存在的类型参数 Element，在扩展中可以被使用
extension Stack {
var topItem: Element? {
return items.isEmpty ? nil : items[items.count - 1]
}
}
- 带有
where子句的扩展
你可以使用 带有 通用where子句 作为扩展的一部分，以下例子中 扩展 上文中的structStack
如果 Element 没有遵守 Equatable 协议，那么会导致编译时错误
//以下扩展的条件是 Element 遵守 Equatable 协议
extension Stack where Element: Equatable {
func isTop(_ item: Element) -> Bool {
guard let topItem = items.last else {
return false
}
return topItem == item
}
}
//使用
var stackOfStrings = Stack<String>()
stackOfStrings.push("zhouwu")
stackOfStrings.isTop("wangzhang")
//调用 isTop，Element 并没有 遵守 Equatable 协议，会编译时错误
struct NotEquatable { }
var notEquatableStack = Stack<NotEquatable>()
let notEquatableValue = NotEquatable()
notEquatableStack.push(notEquatableValue)
notEquatableStack.isTop(notEquatableValue)
- 扩展可以添加计算实例属性 和计算类型属性 到现有的类型中(注意是计算属性 不是 存储属性)
以下例子中 就是添加了 5个 计算实例属性 到 Double 类型中
添加的属性都是只读的计算属性，为了简洁 省略了关键字get
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 }
}
- 扩展可以为类型添加新的初始化程序，但是只能为类
class扩展便利构造器，不能扩展新的指定构造器，指定构造器必须在原始类中实现，每个类都必须有指定构造器，但并不是唯一，可以有多个，便利构造器也可以有多个
构造器原则：
<1> 每个指定构造器 必须调用 直系父类的指定构造器
<2> 每个便利构造器必须调用自身的一个构造器（可以是指定的也可以是便利的）
<3> 便利构造器最终都会调用指定构造器
可以为值类型添加 初始化 程序
//每个存储属性都给了一个初始值，所以系统会生成一个默认的构造器`init()`
//还有一个成员初始化构造器 Size(width: <#T##Double#>, height: <#T##Double#>)
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()
}
//使用（两个自动生成的构造器，只有给了初始值 才会有第一个构造器）
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))
为类扩展便利初始化构造器
class Person: NSObject {
var name: String = ""
var sex: String = ""
//创建指定构造器
init(tName: String, tSex: String) {
name = tName
sex = tSex
super.init()
}
}
extension Person {
convenience init(tname: String) {
//必须调用这个类自身的另一个构造器
self.init(tName:tname, tSex:"nv")
}
}
- 扩展可以添加新的 实例方法 和 类型方法
//为 int 类型 添加实例方法
// repetitions 方法接收一个参数，类型是 参数为空，返回值为空的 闭包
extension Int {
func repetitions(task: () -> Void) {
for _ in 0..<self {
task()
}
}
}
// 使用
3.repetitions {
print("Hello!")
}
如果扩展中的方法 需要修改 值类型 本身（struct，enum），或者修改自己的属性，那么需要在方法前添加关键字mutating,引用类型不需要，因为引用类型的属性本身对方法完全透明，可以修改
//Int 为值类型，扩展方法中需要修改本身，所以要加关键字 mutating
extension Int {
mutating func square() {
self = self * self
}
}
//使用
var someInt = 3
someInt.square()
- 添加下标
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]
- 扩展可以为 classes, structures, and enumerations 添加嵌套类型
例如：为int 嵌套 enum 类型 Kind
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
}
}
}
//使用
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])
- 扩展协议使用通用
where子句
protocol Container {
associatedtype Item
mutating func append(_ item: Item)
var count: Int { get }
subscript(i: Int) -> Item { get }
}
extension Container where Item: Equatable {
func startWith(_ item: Item) -> Bool {
return count >= 1 && self[0] == item
}
}
extension Array: Container {}
//使用
_ = [9, 9, 9].startWith(43)
- 以上
where子句 都是指定实现某个协议，也可以指定 为特定的类型
extension Container where Item == Double {
}
网友评论