Swift中Runtime的UnsafeRawPointer的问题

2017-03-25

Swift和OC的一些不同

在些文章长按复制的Label的时候，使用到了一个方法

1
2
3

objc_getAssociatedObject()
objc_setAssociatedObject()

使用Runtime对一个Object进行赋值取值，由于对Runtime用的还不如Java的Reflection多，匆匆忙忙的用了，被坑了一把

//Object C
OBJC_EXPORT id objc_getAssociatedObject(id object, const void *key)
OBJC_EXPORT void objc_setAssociatedObject(id object, const void *key, id value, objc_AssociationPolicy policy)
//Swift
public func objc_getAssociatedObject(_ object: Any!, _ key: UnsafeRawPointer!) -> Any!
public func objc_setAssociatedObject(_ object: Any!, _ key: UnsafeRawPointer!, _ value: Any!, _ policy: objc_AssociationPolicy)

key 参数的不同

通过函数定义可以看出来

在OC中的key是一个 const void * 类型,代表的是C语言的一个地址，可以是一个函数，也可以是一个变量
在Swift中使用的是一个UnsafeRawPointer的对象且为非空

如果使用字符串key进行操作

//OC中不会报错
objc_getAssociatedObject(obj, @"someKey");
objc_setAssociatedObject(obj, @"someKey", someValue, OBJC_ASSOCIATION_ASSIGN);

//Error Swift中会报错
objc_getAssociatedObject(obj, "someKey")
objc_setAssociatedObject(self, "someKey", someValue, objc_AssociationPolicy.OBJC_ASSOCIATION_ASSIGN)

产生报错的原因就是在Swift中 对常量字符串的处理，并不一定是同一个地址 OC中是相当于把字符串的“someKey”的地址传入了方法，

OC中两次都是同一个地址，所以没问题，但是在Swift中两次字符串不一定是同一个地址，为了确保无误

//声明一个全局变量
private var copyKey:String = "COPY"

//使用这个全局变量的地址作为Key，保证不变性
objc_setAssociatedObject(self, &copyKey, true, objc_AssociationPolicy.OBJC_ASSOCIATION_ASSIGN)
objc_getAssociatedObject(self,&copyKey) as! Bool

具体的原因可以去详读有关UnsafeRawPointer的官方文档

到底应该如何设置key

通常来讲，我们的key需要是一个在C语言可以被访问到，并且不会变的地址, 那么通常需要相当的就是 static 全局变量

1
2
3

static char someObjectKey                   //使用 &someObjectKey 作为key值
#define SOME_KEY @"SOME_KEY"                // OC中的常量字符串都是存在同一个地址
static NSString *const someKey = @"someKey" // 静态常量

既然是地址就可以，有一种更优雅的方法，就是直接使用当前函数的 “@selector” 的地址作为Key

1
2
3

- (void)someFunction {
    objc_setAssociatedObject(self, @selector(someFunction), nil, OBJC_ASSOCIATION_ASSIGN);
}