美文网首页
swift使用#function为Extension添加关联属性

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

作者: Yu_Chou | 来源:发表于2018-09-20 10:08 被阅读120次

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是行不通的。查看类型发现#functonString类型,而_cmdObjective-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)
        }
    }

相关文章

网友评论

      本文标题:swift使用#function为Extension添加关联属性

      本文链接:https://www.haomeiwen.com/subject/scegnftx.html