iOS NSNotificationCenter与自定义通知的封

前言

作为iOS开发者，大家应该都使用过系统通知（NSNotificationCenter），无非就是三步，1. 注册通知，2.发送通知，3.销毁观察者，我在这里就不多解释了；。如果忘记销毁观察者，ios9之前是会崩溃的。因此我就有了自己实现全局一对多分发通知的想法，于是封装了PSSNotificationCenter

系统通知如何使用

通知的使用为3步：

1. 注册通知

[[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(receiveSysNoti) name:@"target" object:nil];

- (void)receiveSysNoti:(NSNotification *)noti {
    NSLog(@"%@", [[NSThread currentThread] isMainThread] ? @"收到系统通知：主线程" : @"收到系统通知：子线程");
}

2. 在需要的时候发送通知

[[NSNotificationCenter defaultCenter] postNotificationName:@"target" object:nil];

3. 移出观察者
   如果iOS9之前，不移除观察者是会崩溃的，而且崩溃不会有断点，如果你继承了友盟，甚至不会给你打印崩溃信息，很难找，所以务必要记得。

    // 移除目标所有通知
    [[NSNotificationCenter defaultCenter] removeObserver:self];
    // 移除目标对应通知
    [[NSNotificationCenter defaultCenter] removeObserver:self name:@"target" object:nil];

系统通知还可以使用block的方式：

但是使用block方式会有内存问题，即便用上面两个方法移除，block本身依旧被NotificationCenter持有，当发送通知时依然会运行；请看代码

// 注册通知
[[NSNotificationCenter defaultCenter] addObserverForName:@"block" object:nil queue:nil usingBlock:^(NSNotification * _Nonnull note) {
    NSLog(@"block方式受到系统通知");
}];

// 点击注销通知
- (IBAction)click3:(id)sender {
    // [[NSNotificationCenter defaultCenter] removeObserver:self];
    [[NSNotificationCenter defaultCenter] removeObserver:self name:@"block" object:nil];
}
// 点击发送通知
- (IBAction)clickSysNoti:(id)sender {
    [[NSNotificationCenter defaultCenter] postNotificationName:@"block" object:nil];
}

上述代码，首先点击发送通知，控制台会打印block方式受到系统通，然后点击注销通知，再点击发送通知，依然会打印；如果退出当前控制器，并且再进入一次控制器，点击发送通知，控制器会打印2次，证明block没有被销毁（知道怎么正确注销的同学欢迎留言告知）
那么为什么会出现这种情况呢？

首先，在使用block方式注册通知的时候，我们只是传了block，并没有传Observer监听者，NSNotificationCenter直接持有block，而我们用的两个注销方式都是针对target-action方式的；我并没有找到block方式的通知的注销方法，所以如果你使用通知，我建议使用target-action方式，或者用我封装的PSSNotificationCenter

系统通知的原理，以及出现的常见问题

使用target-action方式的原理大概是这个样子的：

在ios9之前，有些类是不支持weak指针的，所以使用的是unsafe_unretained类型的指针；顾名思义，unsafe_unretained的意思就是不安全的，不会对对象持有的指针，weak所指向的地址被注销时，指针自动指向nil，向nil指针发送消息是不会崩溃的；而unsafe_unretained指针指向的内存被销毁时，是不会指向nil的，从而导致野指针，所以ios9之前是需要注销的；
当你调用注销方法时，NSNotificationCenter会把指针指向nil，避免出现野指针

内存大概是这个样子的

NSNotificationCenter 内部根据每个通知名称name都引用着一个数组（这个数组应该是弱引用数组NSHashTable），数组里存着每个注册者的target-action，当发通知时（[[NSNotificationCenter defaultCenter] postNotificationName:@"target" object:nil];），会根据NotificationName找到对应的数组，然后便利数组里的target-action，达到批量分发的功能。

使用block方式大概是这个样子的：

注册通知时，不需要你传Observer，NSNotificationCenter根据NotificationName持有一个数组，然后把block放到这个数组中，当你调用postNotification时，根据NotificationName找到对应数组并遍历调用；（但是会出现block不会被释放的问题）

子线程发送通知时：

在发送时都是遍历target-action或block调用方法，因此，在子线程发送通知，方法调用也是发生在子线程，也就是说，接收也是默认在子线程的，所以如果你需要在接收通知时刷新UI，建议跳转到主线程哦。

target-action方式需要声明单独的方法接收，如果支持ios9之前的版本还得注销，真是好麻烦。block方式呢，还会出现内存问题，API提供的较少，因此我决定，封装一款自己的通知 - PSSNotificationCenter

PSSNotificationCenter 自己封装的全局通知

git地址: https://github.com/Pangshishan/PSSNotification
觉得好用记得点赞哦

封装这个框架，首先，不能存在内存问题；其次，用法必须简单容易理解。因此我选择block方式进行封装；

• 使用方法：不需要注销，也不存在内存问题，子线程发通知也会自动跳转到主线程接收通知，是线程安全的。

// 注册通知
    // 不带NotificationName，字用默认的name kDefaultNotificationName。observer必须传，而且要是NSObject
    [[PSSNotificationCenter defaultCenter] addEvent:^(id info) {
        NSLog(@"%@", info);
    } observer:self.obj_1];
    
    [[PSSNotificationCenter defaultCenter] addEvent:^(id info) {
        NSLog(@"%@", info);
    } observer:self];
    
    [[PSSNotificationCenter defaultCenter] addEvent:^(id info) {
        
    } eventName:@"PSS" observer:self];

// 发送通知
    [[PSSNotificationCenter defaultCenter] postDefaultNotification:@"11111"];
    [[PSSNotificationCenter defaultCenter] postNotificationByName:@"PSS" info:@"附带信息"];

实现原理及源代码

就像系统的NSNotificationCenter一样，我们也需要一个单利PSSNotificationCenter，然后说一说我们内存方案：

内存
上图中的EventSet是通过运行时动态给VC1（也可以说是监听者，NSObject类型）添加的属性

PSSNotificationCenter下有一个Dictionary，结构大致如下：

@property (nonatomic, strong) NSMutableDictionary <NSString *, NSMapTable<NSString *, PSSEventSet *> *> *eventDict;

@{
      @"通知的名字_1": @{ // 这个字典是NSMapTable，可以对持有的Value弱引用
              @"观察者内存地址生成的字符串_1": PSSEventSet对象,
              @"观察者内存地址生成的字符串_2": PSSEventSet对象,
              },
      @"通知的名字_1": @{ // 这个字典是NSMapTable
              @"观察者内存地址生成的字符串_1": PSSEventSet对象,
              @"观察者内存地址生成的字符串_2": PSSEventSet对象,
              },
};

NSMapTable： 类似于字典，可以对value进行弱引用；

PSSEventSet： 动态添加给观察者的属性

@interface PSSEventSet : NSObject
@property (nonatomic, strong) NSMutableArray<PSSBlockObject *> *blockObjectArray;
@end

PSSBlockObject： 用于持有block

typedef void(^PSSNotiEvent)(id info);

@interface PSSBlockObject : NSObject

@property (nonatomic, copy) PSSNotiEvent eventHandler;

@end

** .h中暴露的方法 **

#define kDefaultNotificationName @"PSSDefaultNotification"
@class PSSEventSet;
@interface PSSNotificationCenter : NSObject

+ (instancetype)defaultCenter;

- (void)addEvent:(PSSNotiEvent)event observer:(NSObject *)observer;
- (void)addEvent:(PSSNotiEvent)event eventName:(NSString *)eventName observer:(NSObject *)observer;

/// info: 传值
- (void)postNotificationByName:(NSString *)name info:(id)info;
- (void)postDefaultNotification:(id)info;
/// 移出对应通知事件
- (void)removeNotificationName:(NSString *)name;
/// 移出所有通知下的 observer对应的事件（不给此observer发送事件了）
- (void)removeObserver:(NSObject *)observer;
- (void)removeObserverByEventSet:(PSSEventSet *)eventSet;
/// 移出对应通知下，对应observer的事件
- (void)removeNotificationName:(NSString *)name observer:(NSObject *)observer;
/// 移出所有事件
- (void)removeAllNoti;

@end

项目git地址，欢迎使用

觉得有用给个star呗