简单实现iOS通知中心的功能

iOS自带的NSNotification在日常使用中如果注册完不removeObserver的话会引发crash,开发者在开发过程中往往由于自己的疏忽而忘记写removeObserver,尤其在代码重构过程中很容易忽略掉.那么有没有好的办法自己实现一套通知中心的功能,而不用在observer的dealloc函数中removeObserver呢?

答案是有的,iOS开发中经常用weak来修饰类的属性,当weak指针指向的对象销毁时,weak指针指向的对象变为nil, 向nil发送任何消息都不会崩溃,有兴趣的同学可以上网查查Appleweak底层的实现原理,粗略的图如下:

iOS中有一个类NSPointerArray弱引用数组,对添加到其中的类不强引用,如果对象销毁,那么NSPointerArray数组中对应的项变为nil,到这里你有想法了没有?

根据这个思路我简单写了点代码实现iOS NSNotification的功能,PWFNotification源码地址代码已经提交到github上.下面简单介绍一个如何使用,用户可以在自己的工程podfile中加上:

pod 'PWFNotification'

就可以把源码加到自己的工程中.你可以通过下面的方法注册PWFNotification:

- (void)pwf_addObserver:(id)observer selector:(SEL)aSelector name:(NSString *)aName object:(id)anObject;

在发通知的时候可以调用PWFNotificationCenter的下面三个方法之一:

- (void)pwf_postNotificationName:(NSString *)aName object:(id)object userInfo:(NSDictionary *)aUserInfo;

- (void)pwf_postNotificationName:(NSString *)aName object:(id)object;

- (void)pwf_postNotificationName:(NSString *)aName;

PWFNotification同时支持在主线程发通知:

[PWFNotificationCenter defaultCenter].postNotificationInMainThread = YES;

如果不设置该属性默认从当前线程发送通知.由于通知中心是一个单例的形式,时间长了通知中心会保存很多无效的observer对象(nil),我也作了相应的处理,在App内收到内存警告时,会清理一些无效的observer及相关信息.

感兴趣的同学可以去github上下载源码看一下,如果有问题请联系674423263@qq.com.