- 扩展的概念
- 扩展计算属性
- 扩展方法
- 扩展构造函数
- 扩展下标
扩展的概念
在Swift中可以使用一种扩展机制,在原有类型(类、结构体和枚举)的基础上添加新功能。扩展是一种“轻量级”的继承机制,即使原有类型被限制继承,我们仍然可以通过扩展机制“继承”原有类型的功能。
扩展机制能够添加的新功能
1、实例计算属性和静态计算属性。
2、实例方法和静态方法
3、构造函数
4、下标
注:扩展机制还有另外一个优势:它扩展的类型可以是类、结构体和枚举,而继承只能是类,不能是结构体和枚举。
声明扩展
extension 类型名 {
//添加新功能
}
声明扩展的关键字是extension, "类型名"是Swift中已有的类型,包括类、结构体和枚举,但是我们仍然可以扩展整型、浮点型、布尔型、字符串等基本数据类型,这是因为这些类型本质上也是结构体类型。
扩展计算属性
我们可以在原始类型上扩展计算属性,包括实例计算属性和静态计算属性。这些添加计算属性的定义,与普通的计算属性的定义一样。
如下:对Int进行扩展(基本数据类型全部属于结构体类型)
对Int类型增加一个计算属性errorMessage,errorMessage是一个只读的计算属性,它没有set访问函数,只有get访问函数,而且把ge t访问函数省略了
extension Int {
var errorMessage: String {
var errorStr = ""
switch self {
case -7:
errorStr = "没有数据。"
case -6:
errorStr = "日期没有输入。"
case -5:
errorStr = "内容没有输入。"
case -4:
errorStr = "ID没有输入。"
case -3:
errorStr = "数据访问失败。"
case -2:
errorStr = "您的账号最多能插入10条数据。"
case -1:
errorStr = "用户不存在,请到51work6.com注册。"
default:
errorStr = ""
}
return errorStr
}
}
let message = (-7).errorMessage
print("Error Code: -7, Error Message:\(message)")
注:扩展中不仅可以定义只读计算属性, 还可以定义读写计算属性、实例计算属性和静态计算属性。但不能定义存储属性。
扩展方法
我们可以在原始类型上扩展方法,包括实例方法和静态方法。这些添加方法的定义与普通方法的定义是一样的。
extension Double {
static var interestRate: Double = 0.0668 //利率
func interestBy1() -> Double {
return self * Double.interestRate
}
mutating func interestBy2() {
self = self * Double.interestRate
}
static func interestBy3(amount: Double) -> Double {
return interestRate * amount
}
}
let interest1 = (10_000.00).interestBy1()
print("利息1: \(interest1)")
var interest2 = 10_000.00
interest2.interestBy2()
print("利息2:\(interest2)")
var interest3 = Double.interestBy3(amount: 10_000.00)
print("利息3:\(interest3)")
注: mutating func interestBy2() 这个方法是变异方法,mutating 使用场景:一般在枚举和结构体中试图修改它的属性的时候使用。
这个属性一般都是存储属性,修改存储属性必须要用mutating。
在本例中:这个是结构体,想要试图修改属性,而这个属性就是当前实例,需要对当前实例进行修改。
无论是修改属性还是修改当前实例的内容,由于这个是结构体,这个结构体是值类型,要想对它们进行修改,就要把方法声明成变异的方法。
扩展构造函数
- 扩展类型的时候也可以添加新的构造函数。
- 值类型与引用类型扩展有所区别。
- 扩展类的时候能向类中添加新的便利构造函数,但不能添加新的指定构造函数或析构函数。指定构造函数和析构函数只能由原始类型提供。
值类型扩展构造函数
//值类型扩展构造函数
struct Rectangle {
var width: Double
var height: Double
init(width: Double, height: Double) {
self.width = width
self.height = height
}
}
extension Rectangle {
init(length: Double) {
self.init(width: length, height: length)
}
}
var rect = Rectangle(width: 320.0, height: 480.0)
print("长方形:\(rect.width) x \(rect.height)")
var square = Rectangle(length: 500.0)
print("长方形:\(square.width) x \(square.height)")
引用类型扩展构造函数(扩展类的时候能向类中添加新的便利构造函数
)
//引用类型扩展构造函数
class Person {
var name: String
var age: Int
func description() -> String {
return "\(name) 年龄是:\(age)"
}
init(name: String, age: Int) {
self.name = name
self.age = age
}
}
extension Person {
convenience init(name: String) {
self.init(name: name, age: 8)
}
}
let p1 = Person(name: "张三")
print("Person1: \(p1.description())")
let p2 = Person(name: "Tony", age: 28)
print("Person2: \(p2.description())")
扩展下标
我们可以把下标认为是特殊的属性,可以实现索引访问属性。我们可以在原始类型的基础上扩展下标功能。
功能:扩展String类型,增加了一个下标,增加下标之后,这个String类型就可以通过下标访问其中的单个字符
//扩展下标
extension String {
subscript(index: Int) -> String {
if index > self.count {
return ""
}
var c: String = ""
var i = 0
for characters in self {
if i == index {
c = String(characters)
break
}
i+=1
}
return c
}
}
let s = "Hello World"
print(s[0]) //输出:H
print("ABC"[2]) //输出:C