Swift利用协议优化NSNotificationCenter
NSNotificationCenter存在的问题
通知没有统一的命名格式
对于通知的命名没有强制的要求，一个项目里可能有多种不同的命名规则。比如：
|
1
2
3
4
5
6
|
class Barista { let notification = "coffeeMadeNotification"}class Trainee { let coffeeMadeNotificationName = "Coffee Made"} |
通知名称可能冲突
因为对于通知名称没有限制，可能在不同的对象定义了同样的通知名，这样会导致难以意料的bug。
|
1
2
3
4
5
6
|
class Barista { let notification = "coffeeMadeNotification"}class Trainee { let coffeeMadeNotificationName = "coffeeMadeNotification"} |
通知的名称是字符串
字符串真的太危险了，很容易出现打错的情况。
|
1
2
|
NSNotificationCenter.defaultCenter() .postNotificationName("coffeeMadNotfication") |
利用protocol的解决方案
我们通过设计一个protocol来解决上面提到的问题。
|
1
|
protocol Notifier { } |
通过枚举统一通知名称
给这个协议增加一个关联类型：
|
1
2
3
|
protocol Notifier { associatedType Notification: RawRepresentable} |
所有要发送通知的对象或者结构体都要实现Notifier这个协议，然后提供一个实现了RawRepresentable的类型。其实就是一个字符串枚举。
|
1
2
3
4
5
6
|
class Barista : Notifier { enum Notification : String { case makingCoffee case coffeeMade }} |
这样就可以有一个统一的方式获取通知名称：
|
1
2
3
|
let coffeeMade = Barista.Notification.coffeeMade.rawValueNSNotificationCenter.defaultCenter() .postNotificationName(coffeeMade) |
避免通知名称冲突
我们可以为通知添加一个唯一的命名空间（namespace）来避免冲突。这里想到的解决方案是使用实现这个协议的object名称，因为每个object的名称在一个项目里是唯一的。
|
1
2
3
4
5
6
|
let baristaNotification = "\(Barista).\(Barista.Notification.coffeeMade.rawValue)"let traineeNotification = "\(Trainee).\(Trainee.Notification.coffeeMade.rawValue)"// baristaNotification: Barista.coffeeMade// traineeNotification: Trainee.coffeeMade |
但是每个通知都要手动添加就太蛋疼了。我们给这个协议加一个拓展方法来生成唯一的通知名称。因为这个方法只需要内部知道，所以标记为private。
|
1
2
3
4
5
|
public extension Notifier where Notification.RawValue == String { private static func nameFor(notification: Notification) -> String { return "\(self).\(notification.rawValue)" }} |
Notifier的扩展方法
添加观察者
最后一个通知的参数类型就是前面定义的那个枚举类型，这样就不用输入通知名称的字符串。
|
1
2
3
4
5
|
static func addObserver(observer: AnyObject, selector: Selector, notification: Notification) { let name = nameFor(notification) NSNotificationCenter.defaultCenter() .addObserver(observer, selector: selector, name: name, object: nil)} |
这样在使用的时候，在实现协议的object上直接方便的添加观察者：
|
1
|
Barista.addObserver(customer, selector: #selector(Customer.drink(_:)), notification: .coffeeMade) |
发送通知
调用的时候应该是这样的：
|
1
|
Barista.postNotification(.coffeeMade) |
这里利用了swfit的默认参数，object和userinfo设置一个默认的空值。实现如下：
|
1
2
3
4
5
|
static func postNotification(notification: Notification, object: AnyObject? = nil, userInfo: [String : AnyObject]? = nil) { let name = nameFor(notification) NSNotificationCenter.defaultCenter() .postNotificationName(name, object: object, userInfo: userInfo)} |
移除观察
这个实现就不贴了。和前面两个方法类似。调用的时候是这样的：
|
1
|
Barista.removeObserver(customer, notification: .coffeeMade) |
总结
通 过灵活利用swift的语言特性：协议关联类型，协议可以添加默认的方法实现以及方法的默认参数，利用自定义的Notifier协议封装了 NSNotificationCenter的调用方式，解决了传统NSNotificationCenter调用的可能产生的三个潜在风险。
源码：https://github.com/andyyhope/Blog_NSNotificationCenterProtocol
