Swift 命名空间形式扩展的实现

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