在swift
项目中,一般是用Extension
的方式来组织代码,将实现的各种协议等分别写在不同的扩展中。
class ZYViewController: UIViewController {}
extension ZYViewController: UITableViewDelegate{}
extension ZYViewController: UINavigationControllerDelegate{}
但是Extension
只能直接添加计算属性,不能直接添加存储属性,这一点和OC的分类很像。不过,同样我们可以使用关联属性来解决这个问题。
Extension
中添加关联属性的一般写法
在Extension
中定义一个私有的AssociatedKeys
结构体,然后为其添加静态属性testAssociated
,使用&AssociatedKeys.testAssociated
对关联属性进行存取。
extension UIViewController {
private struct AssociatedKeys {
static var testAssociated: String = "testAssociatedKey"//也可以是其他类型
}
var testAssociated: String {
get{
guard let t = objc_getAssociatedObject(self, &AssociatedKeys.testAssociated) as? String else {
return ""
}
return t
}
set{
objc_setAssociatedObject(self, &AssociatedKeys.testAssociated, newValue, .OBJC_ASSOCIATION_RETAIN_NONATOMIC)
}
}
}
OC添加关联属性
一开始在OC中添加关联属性时,也是单独定义一个Key。后来发现了一种取巧的办法,就是直接使用_cmd
。
- (NSString *)testAssociated
{
return objc_getAssociatedObject(self, _cmd);
}
- (void)setTestAssociated:(NSString *)testAssociated
{
objc_setAssociatedObject(self, @selector(testAssociated), testAssociated, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
}
根据OC的写法使用#function添加关联属性(有bug)
function是swift2.2中增加的,用于获取所在方法的name。其作用和OC的_cmd是一样的。于是修改关联属性写法。
var testAssociated:String {
get{
guard let t = objc_getAssociatedObject(self, #function) as? String else {
return ""//默认值
}
return t
}set{
objc_setAssociatedObject(self, #function, newValue, .OBJC_ASSOCIATION_RETAIN_NONATOMIC)
}
}
在使用时发现get只是偶尔能获取到正确的值,经常都是默认值。
可见直接使用#function
是行不通的。查看类型发现#functon
是String
类型,而_cmd
在Objective-C
的方法中表示当前方法的selector
。直接使用不行,就没有办法了?
使用#function添加关联属性的正确姿势
既然#function
不能直接使用,能不能将其转化为一个稳定的值?查看objc_getAssociatedObject
的key,发现其是一个UnsafeRawPointer
类型。于是将#function
转换为Data
再转为UnsafeRawPointer
。
let key = #function.data(using: .utf8)?.withUnsafeBytes({ (uint8Ptr: UnsafePointer<UInt8>) -> UnsafeRawPointer in
return UnsafeRawPointer(uint8Ptr)
})
但是最终发现这个key其实也是一个随机值。于是修改最终代码如下。使用#function
作为key
来缓存转换后的值,每次转换前先从keysDic
中查找。
struct AssociationKey {
static var keysDic:[String: UnsafeRawPointer] = [:]
static func from(_ str:String) -> UnsafeRawPointer{
var key = keysDic[str]
if key == nil {
key = str.data(using: .utf8)?.withUnsafeBytes({ (uint8Ptr: UnsafePointer<UInt8>) -> UnsafeRawPointer in
return UnsafeRawPointer(uint8Ptr)
})
keysDic[str] = key
}
return key!
}
}
最后我们可以修改关联属性写法如下
var testAssociated:String {
get{
guard let t = objc_getAssociatedObject(self, AssociationKey.from(#function)) as? String else {
return ""
}
return t
}set{
objc_setAssociatedObject(self, AssociationKey.from(#function), newValue, .OBJC_ASSOCIATION_RETAIN_NONATOMIC)
}
}
网友评论