swift. 扩展类添加属性_Swift 扩展中使用存储属性

Swift 扩展中使用存储属性

26 Dec 2017

Swift Extensions 的主要局限在于，没法添加存储属性。不过可以通过其他方案实现。

前言

Swift Extensions 可以为现有的类添加方法、结构体、枚举或协议，是比较常用的 Swift 特性之一。但是，Swift 没有直接实现在扩展中持有某些对象作为属性的功能。本文就是介绍怎样通过现在的 API 来实现我们想要的功能。

目标

举个例子，现在有扩展 UISearchBar+QMUI，需要存储属性 qmui_usedAsTableHeaderView，用来标记是否用作 TableHeaderView：

public var qmui_usedAsTableHeaderView: Bool?

编译时会报错误：

Extensions may not contain stored properties

很明显，Swift 不支持扩展中的存储属性。因此，无法使用 qmui_usedAsTableHeaderView 属性。

解决方案

联系到 OC 中的关联对象，很容易想到，是不是可以用 objc_getAssociatedObject 和 objc_setAssociatedObject 来存储与某个键相关联的对象。

这个方法需要两个参数：

object: Any!：关联的源对象。这里一般是把 self 作为参数。

key: UnsafeRawPointer!：指向关联对象的键的指针。

value: Any!：关联对象。

policy: objc_AssociationPolicy：用于保存对