Swift 的 extension 机制很强大,不仅可以针对自定义的类型,还能作用于系统库的类型,甚至基础类型比如 Int。当在对系统库做 extension 的时候,就会涉及到一个命名冲突的问题。Objective-C 时代的通行解决办法是在扩展方法名字的最前面加上 XXX_ 形式的前缀。这种形式不但解决了命名冲突的问题,而且增强了代码可读性。一旦阅读到这种风格的方法名,就知道是非系统的实现。Swift 社区最初的一段时间内,也是按照这种命名方式来做的。
Swifty
在前缀形式的扩展使用了一段时间之后,大家渐渐觉得前缀形式的 Objective-C 风格不再适合 Swift。在命名方式上,社区掀起了一股 Swifty 化的风潮。WWDC 2016 的 Session 403 Swift API Design Guidelines 中详细阐述了 Swifty 风格的命名规则,而 Swift 3 的大量 API 的改动,也是按照这种风格进行的演进。很多开源的 Swift 陆陆续续在接下来的版本中,抛弃了之前的前缀命名形式,改用了 namespace 形式的命名。比如 RxSwift 的 rx_ => rx.,SnapKit 的 snp_ => snp.
对我们来说,实际的开发过程中,也经常会对系统库中的已有类型做自定义的扩展,如果有一种通用的形式,来实现这种扩展,那就太好了。
原理
namespace 形式扩展的原理,就是对原类型进行一层封装。在 Swift 中,这个封装类型使用的是 Struct,然后,对这个 Struct 进行自定义的方法扩展。
代码实现
以我刚写的一个开源库 HandOfTheKing举例,这个库的名字起源于冰与火之歌中的’国王之手’,主要功能是提供一个基于 RxSwift 和 SnapKit 的 iOS App 开发环境,并包含一些 iOS 开发中的实用扩展,比如链式的 UI 布局代码实现。其中的 HandOfTheKing/Namespace.swift 文件里,包含了命名空间形式扩展的实现。源码如下:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 | public protocol NamespaceWrappable { associatedtype WrapperType var hk: WrapperType { get } static var hk: WrapperType.Type { get } } public extension NamespaceWrappable { var hk: NamespaceWrapper<self> { return NamespaceWrapper(value: self) } static var hk: NamespaceWrapper<self>.Type { return NamespaceWrapper.self } } public protocol TypeWrapperProtocol { associatedtype WrappedType var wrappedValue: WrappedType { get } init(value: WrappedType) } public struct NamespaceWrapper<t>: TypeWrapperProtocol { public let wrappedValue: T public init(value: T) { self.wrappedValue = value } }</t></self></self> |
使用方式
如果只想使用的话,把上面的源码拖到你的工程里,然后修改 hk 为任何你想要的前缀即可。扩展使用方式也很简单,举一个实际使用情况的例子,如果想要对 String 扩展一个名为 test 的方法,方法返回自身内容。除了写成方法,为了更便于使用,定义成计算属性也是可以的,代码如下:
1 2 3 4 5 6 | extension String : NamespaceWrappable { } extension TypeWrapperProtocol where WrappedType == String { var test: String { return wrappedValue } } |
然后就可以按照下面的形式来使用这个扩展
1 2 | let testStr = "foo" .hk.test print(testStr) |
在对 TypeWrapperProtocol 这个协议做 extension 时, where 后面的 WrappedType 约束可以使用 == 或者 :,两者是有区别的。如果扩展的是值类型,比如 String,Date 等,就必须使用 ==,如果扩展的是类,则两者都可以使用,区别是如果使用 == 来约束,则扩展方法只对本类生效,子类无法使用。如果想要在子类也使用扩展方法,则使用 : 来约束。
还有一些注意的地方
由于 namespace 相当于将原来的值做了封装,所以如果在写扩展方法时需要用到原来的值,就不能再使用 self,而应该使用 wrappedValue。
对类型扩展实现 NamespaceWrappable 协议,只需要写一次。如果对 UIView 已经写了 NamespaceWrappable 协议实现,则 UILabel 不需要再写。实际上写了之后,编译会报错。
如果在实现的 func 前加上 static 关键字,可以扩展出静态方法。
代码分析
解释一下实现的代码,由于使用了 protocol 和 generic 来实现,这里的代码不是很容易理解。
首先是定义了一个 NamespaceWrappable 协议,这个协议代表了支持 namespace 形式的扩展。并紧接着给这个协议 extension 了默认实现。这样实现了这个协议的类型就不需要自行实现协议所约定的内容了。
NamespaceWrappable 协议的默认实现返回了 NamespaceWrapper 这个带有泛型的 Struct。同时这个 Struct 实现了 TypeWrapperProtocol 协议。而 TypeWrapperProtocol 协议也带有泛型,而这两个泛型相互关联。这样就形成了最终的写法。
如果没有 TypeWrapperProtocol 这个协议,是可以直接对 NamespaceWrapper 这个泛型的 Struct 进行扩展的,对 Objective-C 的类来说这样的写法没有问题,但当尝试对 Swift 的值类型进行扩展时,会产生编译错误,比如下面这两种写法的代码:
1 2 3 4 | extension NamespaceWrapper where T == String { } extension NamespaceWrapper where T: String { } |
会产生不同的编译错误,有兴趣可以尝试。为了统一,则多封装了 TypeWrapperProtocol 这个协议。
http://www.cocoachina.com/ios/20171031/21000.html