Swift实现类扩展添加属性

Swift不允许在extension(类扩展)中直接添加属性。但是我们在实际开发中可能会遇到这种情况。
解决方案:使用objc_set/getAssociatedObject(关联属性)来实现。

场景一:给自定义button添加一个类扩展,关联name属性

class RightButton: UIButton {
    override init(frame: CGRect) {
        super.init(frame: frame)
        self.name = "123"
        self.setTitle(name, for: .normal)
    }
    required init?(coder: NSCoder) {
        fatalError("init(coder:) has not been implemented")
    }

}
private var key: Void?
extension RightButton {
    var name: String?{
        get {
            return objc_getAssociatedObject(self, &key) as? String
        }
        set {
            objc_setAssociatedObject(self, &key, newValue, .OBJC_ASSOCIATION_RETAIN)
        }
    }
    
}

场景二:自定义协议btnProtocol,给协议的类扩展关联一个nick的只读属性,让自定义的button遵守该协议,获取该属性的值

protocol btnProtocol {
    var age: Int {get}
}

extension btnProtocol {
    var nick: String?{
        get {
            return "aaa"
        }
    }
}
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容