美文网首页
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