Swift Add Variable For A Exist Class

发表于 | 分类于 |

在Swift中如何为一个存在的类添加变量，新建一个Person.swift文件，代码如下：

1
2
3
4
5
6
7
8
import Foundation
class Person: NSObject {
    var name: NSString?
    func eatFood() -> Void {
        print("People can eat")
    }
}

现在新建PersonExtension.swift，代码如下：

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
import Foundation
private var key: Void?
extension Person {
    var gender: String? {
        get {
            return objc_getAssociatedObject(self, &key) as? String
        }
        set {
            return objc_setAssociatedObject(self, &key, newValue, .OBJC_ASSOCIATION_RETAIN_NONATOMIC)
        }
    }
}

现在我们则为Person新添加了一个变量,可以使用一下代码访问新变量

1
2
3
4
let per = Person()
per.name = "roy"
per.gender = "boy"
per.eatFood()