关于NSNotiFication和NSNotificationCenter的原理及实现

使用场景

消息通知机制是一种以广播的方式将消息传递给所有观察者的机制，比如说现在有A，B，C，D四个控制器，当A的某个参数发生改变时，需要通知到C，D控制器，这时就可以用通知的方式来实现。

用法

如上的例子的写法如下：

/**
 A控制器的某个属性值发生改变调用该方法
 */
 - (void)somePropertyValueChanged
{
    //发送一个通知，名字是somePropertyValueChanged
    [[NSNotificationCenter defaultCenter] postNotificationName:@"somePropertyValueChanged" object:nil];
}

发送通知的方法总共有两种，分别是：

 - (void)postNotificationName:(NSNotificationName)aName object:(id)anObject;

 - (void)postNotificationName:(NSNotificationName)aName object:(id)anObject userInfo:(NSDictionary *)aUserInfo;

说明一下上面的参数：

• aName代表的是这个通知的名字，观察者在注册的时候需要注册这个名字。
• aUserInfo就是通知传递的参数，是一个字典类型。
• anObject是一种相当于二次确认的作用来的，总共有三种情况：
  
  1. 发送通知的object和注册观察者的object都为空，那么就相当于不进行二次确认，只要名字相同就可以接收到。
  2. 发送通知的object为空，注册观察者的object不为空，那么这个观察者就接受不到该条通知。
  3. 发送通知的object不为空，注册观察者的object为空，依然接收不到该通知。

---

C控制器注册观察者的方法如下:

- (void)viewDidLoad {
    [super viewDidLoad];
    //C控制器注册观察者，objec设为字符串test，调用方法- (void)test;
    [[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(test) name:@"somePropertyValueChanged" object:@"test"];
}

- (void)test
{
    NSLog(@"接收到了");
}

---

D控制器注册观察者的方法如下:

- (void)viewDidLoad {
    [super viewDidLoad];
    //D控制器注册观察者，objec设为nil，调用方法- (void)test;
    [[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(test) name:@"somePropertyValueChanged" object:nil];
}

- (void)test
{
    NSLog(@"接收到了");
}

这时当A发送通知时，只有D能收到该通知。这里要注意的是，观察者的注册需要在发送通知之前。

---

注册观察者有以下几种方式:

 - (void)addObserver:(id)observer selector:(SEL)aSelector name:(NSNotificationName)aName object:(id)anObject;

 - (id<NSObject>)addObserverForName:(NSNotificationName)name object:(id)obj queue:(NSOperationQueue *)queue usingBlock:(void (^)(NSNotification *note))block;

 - (void)addObserver:(NSObject *)observer forKeyPath:(NSString *)keyPath options:(NSKeyValueObservingOptions)options context:(void *)context;

• 第一种方式是比较常用的添加Oberver的方式，接到通知时执行aSelector。
• 第二种方式是基于Block来添加观察者，往通知中心的调度表中添加观察者，这个观察者包括一个queue和一个block,并且会返回这个观察者对象。当接到通知时执行block所在的线程为添加观察者时传入的queue参数，queue也可以为nil，那么block就在通知所在的线程同步执行。

• 第三种是键值观察模式，具体请看http://www.cppblog.com/kesalin/archive/2012/11/17/kvo.html

  实现原理

  消息通知的原理简单来说就是有一个通知中心，它有一些属性，来存放那些注册了的观察者的信息，然后当通过通知中心发送通知的时候会去根据name，object去查找符合条件的观察者，将通知发送给它们。

  下面通过自己通过代码来实现NSNotiFication和NSNotificationCenter的方式来详细了解其中的原理，具体请看代码的注释。

//
//  EWNotificationCenter.h
//

#import <Foundation/Foundation.h>

@interface EWNotificationCenter : NSObject

+ (instancetype)defaultCenter;

- (void)addObserver:(id)observer selector:(SEL)aSelector name:(NSNotificationName)aName object:(id)anObject;

//发送通知
- (void)postNotificationName:(NSString *)aName object:(nullable id)anObject;
- (void)postNotificationName:(NSString *)aName object:(nullable id)anObject userInfo:(nullable NSDictionary *)aUserInfo;

@end

//
//  EWNotificationCenter.m
//

#import "EWNotificationCenter.h"
#import "EWNotificantionModel.h"
#import "EWObserverModel.h"

@interface EWNotificationCenter()

@property (nonatomic,strong) NSMutableArray *notificationArr;

@end

@implementation EWNotificationCenter

+ (instancetype)defaultCenter
{
    static EWNotificationCenter *defaultCenter;
    static dispatch_once_t onceToken;
    dispatch_once(&onceToken, ^{
        defaultCenter = [[EWNotificationCenter alloc] init];
    });
    return defaultCenter;
}

- (void)addObserver:(id)observer selector:(SEL)aSelector name:(NSNotificationName)aName object:(id)anObject
{
    EWObserverModel *observerModel = [[EWObserverModel alloc] init];
    observerModel.observer = observer;
    observerModel.selector = aSelector;
    observerModel.object = anObject;
    if (!_notificationArr) {
        _notificationArr = [NSMutableArray array];
    }
    for (int i = 0; i < _notificationArr.count; i++) {
        EWNotificantionModel *model = _notificationArr[i];
        if ([model.notificationName isEqualToString:aName]) {
            [model.observers addObject:observerModel];
            return;
        }
    }

    EWNotificantionModel *notificationModel = [[EWNotificantionModel alloc] init];
    notificationModel.notificationName = aName;
    notificationModel.observers = [NSMutableArray array];
    [notificationModel.observers addObject:observerModel];
    [_notificationArr addObject:notificationModel];
}

- (void)postNotificationName:(NSString *)aName object:(nullable id)anObject
{
    [self postNotificationName:aName object:anObject userInfo:nil];
}

- (void)postNotificationName:(NSString *)aName object:(nullable id)anObject userInfo:(nullable NSDictionary *)aUserInfo
{
    if (aName) {
        //遍历通知数组，通过通知名字匹配
        for (EWNotificantionModel *model in _notificationArr) {
            //如果找到对应的通知
            if ([aName isEqualToString:model.notificationName]) {
                //遍历该通知的观察者数组
                for (EWObserverModel *obserModel in model.observers) {
                    //判断object是否相等
                    if ((!anObject && !obserModel.object) || ([anObject isEqual:obserModel.object])) {
                        [obserModel.observer performSelector:obserModel.selector withObject:aUserInfo];
                    }
                }
            }
        }
    }
}

@end

//
//  EWNotificantionModel.h
//

#import <Foundation/Foundation.h>

@interface EWNotificantionModel : NSObject

@property (nonatomic,strong) NSString *notificationName;
@property (nonatomic,strong) id obejct;
@property (nonatomic,copy) NSDictionary *userInfos;
@property (nonatomic,strong) NSMutableArray *observers;

@end

//
//  EWObserverModel.h
//

#import <Foundation/Foundation.h>

@interface EWObserverModel : NSObject

@property (nonatomic,strong) id observer;
@property (nonatomic,assign) SEL selector;
@property (nonatomic,strong) id object;

@end

通过上面的代码就可以实现通知了（只实现了发送通知和注册观察者）