由于在Swift中通过扩展(Extension)只能添加计算类型,不能添加储存类型,所以在需要添加储存类型的时候,要使用一种合适的方法来存。
使用的最多的方法
在网上搜索如何添加储存类型,出来的方法都是通过Runtime来存储。也就是通过这两个方法:
public func objc_setAssociatedObject(_ object: Any, _ key: UnsafeRawPointer, _ value: Any?, _ policy: objc_AssociationPolicy)
public func objc_getAssociatedObject(_ object: Any, _ key: UnsafeRawPointer) -> Any?
由于网上都是这个方法的介绍,并且本文的重点并不是这种方法,,所以就丢下一个东城夕阳的例子,不详细介绍了。
extension UIView {
// MARK:- RuntimeKey 动态绑属性
struct RuntimeKey {
static let kProgressHud = UnsafeRawPointer.init(bitPattern: "kProgressHud".hashValue)
}
var HUD: MBProgressHUD? {
set {
objc_setAssociatedObject(self, UIView.RuntimeKey.kProgressHud!, newValue, .OBJC_ASSOCIATION_RETAIN_NONATOMIC)
}
get {
return objc_getAssociatedObject(self, UIView.RuntimeKey.kProgressHud!) as? MBProgressHUD
}
}
}
遇到的坑
由于工程的需要,我需要给 URLRequest 添加一个储存类型,于是就按照网上的方法进行了操作,但是发现无论怎么反复检查,objc_getAssociatedObject 出来的永远是 nil ,百度了半天都没有找到解决方法,后来在这里找到了答案,原话如下:
You cannot add associated objects to a Swift Array (or any Swift value type). objc_setAssociatedObject() and objc_getAssociatedObject() are from the Objective-C runtime, they expect an instance of NSObject as first argument.
Your code compiles and runs only because any Swift value is automatically bridged to an object if necessary.
When you call objc_setAssociatedObject(self, ...) then self is bridged to a (temporary) instance of NSArray, and the association is made on that object.
Later, when objc_getAssociatedObject(self, ...) is called, another (temporary) instance of NSArray is created, and that has no associated object.
That's why you get nil as the result.
发现原因是要拓展的类型不是一个 NSObject 后,我就 Jump to Definition 查看了一下 URLRequest 的定义,发现它果然是一个 结构体(struct) 不是一个 NSObject,于是问题就来了,如何给一个 Swift value type 或者 struct 中添加一个储存属性呢
解决坑
之前的那个问题,我也就Google了一会就出来了,但是这个问题却找了老半天,最后在这里找到了方法,作者是通过使用 纯Swift 的方法来进行存储,首先是 代码v.1.0
extension URLRequest {
struct Holder {
static var _myComputedProperty:Bool = false
}
var myComputedProperty:Bool {
get {
return Holder._myComputedProperty
}
set {
Holder._myComputedProperty = newValue
}
}
}
但是这时有几个问题:
- Holder结构的访问权限问题。实际上,所有人都可以直接通过 Holder 来访问变量,无需进一步操作即可修改值
- 拓展类的多实例问题。当创建了好多个 URLRequest 后,由于 static 关键词的原因,他们将共享这个值
为了解决上述的问题,于是就利用了 哈希表 来保存属于不同实例的该属性,推出了 代码v.2.0
extension URLRequest {
private struct Holder {
static var _myComputedProperty = [String:Bool]()
}
var myComputedProperty:Bool {
get {
return Holder._myComputedProperty[self.debugDescription] ?? false
}
set {
Holder._myComputedProperty[self.debugDescription] = newValue
}
}
}
self.debugDescription是一种非常"原始"的方法,我们应该选用更加智能、简洁的方法来区分不同的实例,现在这么写看起来就像是在debug
为了使得代码更加的美观,于是就用 静态变量 来模拟 哈希表,因此没有理由依赖于 struct 方法,最终版本 v.3.0 如下
extension URLRequest {
private static var _myComputedProperty = [String:Bool]()
var myComputedProperty:Bool {
get {
let tmpAddress = String(format: "%p", unsafeBitCast(self, to: Int.self))
return URLRequest._myComputedProperty[tmpAddress] ?? false
}
set {
let tmpAddress = String(format: "%p", unsafeBitCast(self, to: Int.self))
URLRequest._myComputedProperty[tmpAddress] = newValue
}
}
}
这类办法有个缺点就是 值类型进行copy的时候 不能保证其Int不变 所以哈希值有可能会改变。所以最好的方法还是用OC的类 或者 老老实实创建一个类吧。
Happy swifty coding!