關於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

通過上面的代碼就可以實現通知了（只實現了發送通知和註冊觀察者）