NSNotificationCenter实现原理

NSNotificationCenter 内部实现

通知怎么使用就不介绍了,相信大家都会使用。

参考cocotron的源代码,观察者模式的通知一种实现方式如下:

主要有下列各类组成

  • NSNotification:通知消息携带的载体,通过它,可以把消息内容传递给观察者。
  • NSNotificationCenter: 通知中心,是一个单例,来管理观察者和发送通知。
  • NSNotificationObserver:观察者。由注册观察者的类和通知selector来初始化,主要就是绑定观察者和实现的通知方法selector。
  • NSObjectToObservers:通知调度表。记录通知name相同的一组观察者。
  • NSNotificationQueue:通知队列。

实现过程:通知name和观察者集合有NSMutableDictionary来维护key和value的对应关系。

  • 添加观察者,从字典中找对应的到NSObjectToObservers通知调度表,然后添加观察者。
  • 发送通知,找到对应的观察者集合,然后循环通过performSelector方法来执行注册的selector。
  • 移除观察者,也是找到相应的通知调度表,然后删除相应的观察者。

NSNotificationCenter 用单例来初始化

+(NSNotificationCenter *)defaultCenter {
   return NSThreadSharedInstance(@"NSNotificationCenter");
}

添加观察者

NSNotificationCenter中代码

-(void)addObserver:anObserver selector:(SEL)selector name:(NSString *)name
   object:object {
   NSNotificationObserver *observer=[[[NSNotificationObserver allocWithZone:[self zone]] 
        initWithObserver:anObserver selector:selector] autorelease];
   NSObjectToObservers *registry;

   if(name==nil)//name为空时,直接对应一个观察者集合
    registry=_noNameRegistry;
   else{
    registry=[_nameToRegistry objectForKey:name];//取出对应的观察者集合

    if(registry==nil){
     registry=[[[NSObjectToObservers allocWithZone:[self zone]] init]
       autorelease];//初始化观察者集合
     [_nameToRegistry setObject:registry forKey:name];//观察者集合和name放在字典中对应起来
    }
   }

   [registry addObserver:observer object:object];
}

NSObjectToObservers中代码

-(void)addObserver:(NSNotificationObserver *)observer object:objectOrNil {
   id              object=(objectOrNil!=nil)?objectOrNil:(id)[NSNull null];
   NSMutableArray *observers=NSMapGet(_objectToObservers,object);

   if(observers==nil){
    observers=[NSMutableArray array];
    NSMapInsert(_objectToObservers,object,observers);
   }
   [observers addObject:observer];
}

通知调度表是由NSMapTable的哈希表维护来记录每一个观察者的。其中防止添加观察者时的相同的object(发送者),也就是hash表里的key把之前的value给覆盖了,所以hash表里的value是一个装有观察者的数组集合。

其中_nameToRegistry和_noNameRegistry的声明和初始化如下:

@interface NSNotificationCenter : NSObject {
   NSMutableDictionary *_nameToRegistry;
   id                   _noNameRegistry;
}
-init {
   _nameToRegistry=[[NSMutableDictionary allocWithZone:[self zone]] init];
   _noNameRegistry=[[NSObjectToObservers allocWithZone:[self zone]] init];
   return self;
}

NSNotificationCenter中的发送通知实现

static inline void postNotification(NSNotificationCenter *self,NSNotification *note){
   NSAutoreleasePool       *pool=[NSAutoreleasePool new];
   NSObjectToObservers *registry=self->_noNameRegistry;

   [registry postNotification:note];
   registry=[self->_nameToRegistry objectForKey:[note name]];
   [registry postNotification:note];

   [pool release];
}

-(void)postNotification:(NSNotification *)note {
   postNotification(self,note);
}

-(void)postNotificationName:(NSString *)name object:object userInfo:(NSDictionary *)userInfo {
    NSNotification *note = NSNotification_concreteNew(NULL,name,object,userInfo);
    postNotification(self,note);
    [note release];
}

-(void)postNotificationName:(NSString *)name object:object {
    NSNotification *note = NSNotification_concreteNew(NULL,name,object,nil);
    postNotification(self,note);
    [note release];
}

作为通知中心来说,只是告知哪一个观察者集合需要发送哪一个name的通知,具体执行还是得靠通知调度表分发下去。

-(void)postNotification:(NSNotification *)note {
// FIXME: NSNotificationCenter sends notifications in the order they are added for observation regardless of
// the object registered. This implementation stores objects for observation seperately so if you observe nil
// and a particular object you will always get the particular object notifications before the nil one instead
// of in the order they are registered.

// The copy and double check for presence is to deal with observers being removed during notification
   id         object=[note object];
   NSArray   *observers;
   NSInteger  count;
    
   if(object!=nil){
    observers=[NSArray arrayWithArray:(id)NSMapGet(_objectToObservers,object)];

    count=[observers count];
    while(--count>=0){
     id observer=[observers objectAtIndex:count];

     if([(NSArray *)NSMapGet(_objectToObservers,object) indexOfObjectIdenticalTo:observer]!=NSNotFound)
      [observer performSelector:_cmd withObject:note];
    }
   }

   observers=[NSArray arrayWithArray:(id)NSMapGet(_objectToObservers,[NSNull null])];

   count=[observers count];
   while(--count>=0){
    id observer=[observers objectAtIndex:count];

    if([(NSArray *)NSMapGet(_objectToObservers,[NSNull null]) indexOfObjectIdenticalTo:observer]!=NSNotFound)
     [observer performSelector:_cmd withObject:note];
   }
}

观察者执行通知

-(void)postNotification:(NSNotification *)note {
   [_observer performSelector:_selector withObject:note];
}

移除观察者

在对象被释放前需要移除掉观察者,避免已经被释放的对象还接收到通知导致崩溃。

NSNotificationCenter中代码

-(void)removeObserver:anObserver {
   NSMutableArray          *removeRegistries=[NSMutableArray array];
   NSEnumerator            *keyEnumerator=[_nameToRegistry keyEnumerator];
   NSString                *key;
   NSObjectToObservers *registry;
   NSInteger                count;

   while((key=[keyEnumerator nextObject])!=nil){
    registry=[_nameToRegistry objectForKey:key];
    [registry removeObserver:anObserver object:nil];
    if([registry count]==0)
     [removeRegistries addObject:key];
   }

   [_noNameRegistry removeObserver:anObserver object:nil];

   count=[removeRegistries count];
   while(--count>=0)
    [_nameToRegistry removeObjectForKey:[removeRegistries objectAtIndex:count]];
}

-(void)removeObserver:anObserver name:(NSString *)name object:object {
   NSMutableArray          *removeRegistries=[NSMutableArray array];
   NSObjectToObservers *registry;
   NSInteger               count;

   if(name!=nil){
    registry=[_nameToRegistry objectForKey:name];

    [registry removeObserver:anObserver object:object];

    if([registry count]==0)
     [removeRegistries addObject:name];
   }
   else {
    NSEnumerator *keyEnumerator=[_nameToRegistry keyEnumerator];
    NSString     *key;

    while((key=[keyEnumerator nextObject])!=nil){
     registry=[_nameToRegistry objectForKey:key];

     [registry removeObserver:anObserver object:object];

     if([registry count]==0)
      [removeRegistries addObject:key];
    }

    [_noNameRegistry removeObserver:anObserver object:object];
   }

   count=[removeRegistries count];
   while(--count>=0){
    NSString                *key=[removeRegistries objectAtIndex:count];
    NSObjectToObservers *remove=[_nameToRegistry objectForKey:key];

    [[remove retain] autorelease];
    [remove invalidate];

    [_nameToRegistry removeObjectForKey:key];
   }
}

NSObjectToObservers实现如下:

-(void)removeObserver:anObserver object:object {
   id  removeKeys[NSCountMapTable(_objectToObservers)];
   int removeCount=0;

   if(object!=nil){
    NSMutableArray *observers=NSMapGet(_objectToObservers,object);
    NSInteger             count=[observers count];

    while(--count>=0)
     if(anObserver==[[observers objectAtIndex:count] observer])
      [observers removeObjectAtIndex:count];

    if([observers count]==0)
     removeKeys[removeCount++]=object;
   }
   else {
    NSMapEnumerator state=NSEnumerateMapTable(_objectToObservers);
    id              key;
    NSMutableArray *observers;

    while(NSNextMapEnumeratorPair(&state,(void **)&key,(void **)&observers)){
     NSInteger count=[observers count];

     while(--count>=0)
      if(anObserver==[[observers objectAtIndex:count] observer])
       [observers removeObjectAtIndex:count];

     if([observers count]==0)
      removeKeys[removeCount++]=object;
    }
   }

   while(--removeCount>=0)
    NSMapRemove(_objectToObservers,removeKeys[removeCount]);
}

从hash表中删除相应的观察者。

通知中心用到的概念还有NSNotificationNSNotificationQueue等。

疑问:若是一个类中注册了多个观察者,则在通知调度表中会记录几次呢?

结合通知实现和NSMapTable的hash规则可以看到,每一通知调度表维护一张hash表,每一个hash表示根据key(通知的object)来插入装有value(观察者)的数组。所以观察者的存储次数会根据通知名字和keyobject)来发送通知而定,有可能记录多次。

以上的通知实现方式应该是最接近苹果官方的实现。
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 157,198评论 4 359
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 66,663评论 1 290
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 106,985评论 0 237
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 43,673评论 0 202
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 51,994评论 3 285
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 40,399评论 1 211
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 31,717评论 2 310
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 30,407评论 0 194
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 34,112评论 1 239
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 30,371评论 2 241
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 31,891评论 1 256
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 28,255评论 2 250
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 32,881评论 3 233
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 26,010评论 0 8
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 26,764评论 0 192
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 35,412评论 2 269
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 35,299评论 2 260

推荐阅读更多精彩内容