swift使用#function为Extension添加关联属性

在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: UnsafeRawBufferPointer) -> UnsafeRawPointer in
                return uint8Ptr.load(as: UnsafeRawPointer.self)
            })
            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)
        }
    }

swift使用#function为Extension添加关联属性