NSNotificationCenter 使用姿势

在 Swift 中，为了避免 Notification 名称直接使用字符串不安全，引入了 NSNotification.Name 类型，rawValue 为 String。虽然这样安全性有了一定的保证，但用起来就不是那么顺手了。

必须先把字符串用 NSNotification.Name 类型封装起来，再传给 NSNotificationCenter：

class A {
    static let notificationName = Notification.Name("aNotificationName")
    NotificationCenter.default.post(name: notificationName, object: nil)
}

不过，这么写安全性并没有得到保障，因为 NSNotificationCenter 是全局的，如果有其它类也使用 aNotificationName 字符串创建通知名称，就会造成通知错乱。

稍微优化一下，给 NSNotification.Name 写一个扩展，用静态常量来表示通知名称：

extension Notification.Name {
    static let aNotification = NSNotification.Name("aNotification")
    static let bNotification = NSNotification.Name("bbNotification")
}

这样用起来就方便些:

NotificationCenter.default.addObserver(, selector: , name: .aNotification, object: nil)
NotificationCenter.default.post(name: .aNotification, object: nil)

不过这么处理还会有些问题，声明跟系统通知方式相同，使用编译器提示会跟系统的通知混在一起不好找。还有就是，因为观察者模式降低了对象之间的偶尔，也就导致写下面这样的代码，你都不知道想监听的是谁，做了什么。

NotificationCenter.default.addObserver(self, selector:#selector(notificationReceived) , name: .aNotification, object: nil)

这些信息就需要程序员自己在方法名或者通知名中流露，如果是别人命名的可能就会看不懂。即使顺着通知名跳过去，还得再去找是哪个类发出来的。

使用 Swift protocol 可以解决这些的困扰。

先上代码：

protocol Notifier {
    associatedtype Notification: RawRepresentable
}

extension Notifier where Notification.RawValue == String {
    fileprivate static func name(for notification: Notification) -> String {
        return "\(self).\(notification.rawValue)"
    }

    func postNotification(_ notification: Notification, object: Any? = nil, userInfo: [String : Any]? = nil) {
        Self.postNotification(notification, object: object, userInfo: userInfo)
    }

    static func postNotification(_ notification: Notification, object: Any? = nil, userInfo: [String : Any]? = nil) {
        let aName = name(for: notification)
        NotificationCenter.default
            .post(name: Foundation.Notification.Name(rawValue: aName), object: object, userInfo: userInfo)
    }

    static func addObserver(_ observer: Any, selector: Selector, notification: Notification, object: Any? = nil) {
        let aName = name(for: notification)
        NotificationCenter.default
            .addObserver(observer, selector: selector, name: NSNotification.Name(rawValue: aName), object: object)
    }

    static func removeObserver(_ observer: Any, notification: Notification, object: Any? = nil) {
        let aName = name(for: notification)
        NotificationCenter.default
            .removeObserver(observer, name: NSNotification.Name(rawValue: aName), object: object)
    }
}

实现起来并不难，只是用协议扩展对 NotificationCenter 的接口做了一层封装，需要发通知的类关联一个 RawRepresentable 类型就实现了这个协议。

因为 enum 默认遵循 RawRepresentable 协议 ，这里配合枚举来定义通知的类型简直完美。

struct Cat {
    private func feelHungry() {
        self.postNotification(.hungry)
    }
}

extension Cat: Notifier {
    enum Notification: String {
        case hungry
        case sleepy
    }
}

Cat 向外发通知：

self.postNotification(.hungry)

其它类注册并处理通知：

Cat.addObserver(self, selector: #selector(feedCat), notification: .hungry)

    @objc
    func feedCat() { }

这样从 addObserver 就可以清晰的看出，观察的对象以及对应的行为，而且通知名还很简洁。

到这里 NSNotificationCenter 的使用姿势已经介绍完毕。

再补充探讨两点 NSNotificationCenter 相关的内容：

1、
在 iOS 9 之前，observer 如果没有调用 removeObserver，有时候会出现野指针的奔溃，是因为 NSNotificationCenter 不会持有 observer，对 observer 的引用是 unowned 的。如果 observer 已经销毁，却没有从通知中心移除，通知中心还是会给它发消息，如果这块地址被覆盖的话就出现无法响应消息的奔溃。

在 iOS 9 之后，可以不用调用 removeObserver了，NSNotificationCenter 以 weak 的方式引用 observer，释放后指针会自动置为 nil。

2、
Notification 中含有一个 object 参数，一般使用的时候都会传个 nil。开始接触的时候以为只是用于广播传值的，其实并仅仅是这样。

public init(name: Notification.Name, object: Any? = default, userInfo: [AnyHashable : Any]? = default)

我们知道 name 是用来标记通知类型的，object 可以理解为通知的二级标记，当 name 相同时，通知的分发就交给 object 来判断了。

如果addObserver 时 object为 nil，就会忽略 object 用 name 来判断观察者是谁。如果 name 为 nil，会根据 object 来判断观察者是谁。

那如果这么写会怎么样？

NotificationCenter.default.addObserver(self, selector: #selector(received), name: nil, object: nil)

name 和 object都为 nil， NotificationCenter 会忽略通知类型的匹配，只要有通知就向给这个 observer，received就会收到所有类型的通知，当然也包括系统的。

Swift中Notification.Name这么难用怎么办
Let’s Build NSNotificationCenter
Swift: NotificationCenter Protocol

如果喜欢本文，可以收藏或者关注，第一时间获得更新推送。