iOS中消息通知的另一种方式

原理

iOS中消息通知是以notificationName来做为标识,发通知时,监听了同一个notificationName的实例会收到通知。

而这里所说的方式是以protocol为标识,发通知时,以protocol+selector,来判断要发送给哪些实例。

做法如下:

  1. 将需要通知的方法定义成protocol,通常是将一组有关联的方法定义在同个protocol中。

  2. 以protocol为key,监听该protocol的实例(可能有多个)为value,组成映射关系,即protocol->observer list。

  3. 防止强引用实例,使用包装对象,用weak属性存储该实例。

  4. 发送通知时,根据protocol+selector,拿到对应的实例,进行调用,同时传递参数。

看图会比较清晰:

1.png

相比自带的消息通知,好处是:

  • 自带的消息通知,添加同一个notificationName多次,会收到多个通知,而这种方式会去重。
  • 参数灵活,可直接传递,不用包装成dict和解dict。
  • 避免移除系统通知的问题
  • 实例对象以weak修饰,其释放后,会自动置为nil。避免野指针问题。

添加观察者

// 可在内部判断该protocol对应的object是否重复添加
#define ADD_SERVICE_CLIENT(protocolName, object) 
[[NotificationServiceCenter defaultCenter] addServiceClient:object withKey:@protocol(protocolName)]

添加时,需带上要监听的protocol。

发送通知

#define NOTIFY_SERVICE_CLIENT(protocolName, selector, func) \
{ \
NSArray *__clients__ = [[NotificationServiceCenter defaultCenter] serviceClientsWithKey:@protocol(protocolName)]; \
for (NotificationClient *client in __clients__) \
{ \
id obj = client.object; \
if ([obj respondsToSelector:selector]) \
{ \
[obj func]; \
} \
} \

发送时需要加上selector和func。selector用来判断实例是否实现了该selector,func用来直接调用方法。

if ([obj respondsToSelector:selector])
{ 
    [obj func]; 
} 

Example:

NOTIFY_SERVICE_CLIENT(UserProtocol, @selector(userLogin:), userLogin:19000);

有个比较巧妙的地方,就是方法的调用和参数的传递。使用宏替换的特性,直接写成[obj func],在编译时不会报错。比如上面的例子中,func指userLogin:19000,展开调用就是[obj userLogin:900]。如果我们要将NOTIFY_SERVICE_CLIENT写成方法,是会编译不过的,因为这里的func根本就不是一种类型。所以采用宏,比较tricky的解决这个问题。

另外,如果在处理通知过程中,又调用NOTIFY_SERVICE_CLIENT给同一个protocol+selector发送通知消息,是会引起死循环的。

比如A监听了ProtocolA的test方法,在test方法中调用NOTIFY_SERVICE_CLIENT(ProtocolA, @selector(test), test),A又会收到通知,循环往复,就会造成死循环。所以,需要注意这种情况。

// 在A中
- (void)addClient {
     ADD_SERVICE_CLIENT(ProtocolA, self)
}

#pragma mark - ProtocolA

// 收到通知
- (void)test {
    NSLog("receive ProtocolA test");
    
    // 会引起死循环
    NOTIFY_SERVICE_CLIENT(ProtocolA, @selector(test), test)
}

移除观察者

#define REMOVE_SERVICE_CLIENT(protocolName, object) 
[[NotificationServiceCenter defaultCenter] removeServiceClient:object withKey:@protocol(protocolName)]

#define REMOVE_ALL_SERVICE_CLIENT(object) 
[[NotificationServiceCenter defaultCenter] removeServiceClient:object]

移除有两种方式:移除单个protocol和移除该object监听的所有protocol。

使用

定义protocol

@protocol UserProtocol <NSObject>

@optional
- (void)logout;

- (void)userLogin:(NSUInteger)userID;

@end

添加监听,实现通知的方法,移除监听。注意需让其遵循协议,因为在ADD_SERVICE_CLIENT会判断object是否遵循了protocol。

@interface ViewController ()<UserProtocol>

@end

- (void)viewDidLoad {
    [super viewDidLoad];
    // Do any additional setup after loading the view, typically from a nib.
        
    // 添加监听    
    ADD_SERVICE_CLIENT(UserProtocol, self);
}

- (void)dealloc {
    // 移除监听
    REMOVE_ALL_SERVICE_CLIENT(self);
}

#pragma mark - UserProtocol

// 通知过来后的具体实现
- (void)logout {
    NSLog(@"logout");
}

- (void)userLogin:(NSUInteger)userId {
    NSLog(@"loginSuccess");
}

发送通知:

- (IBAction)login:(id)sender {
    NOTIFY_SERVICE_CLIENT(UserProtocol, @selector(userLogin:), userLogin:19000);
}

- (IBAction)logout:(id)sender {
    NOTIFY_SERVICE_CLIENT(UserProtocol, @selector(logout), logout);
}

如果只关心protocol的某个方法通知,只需要实现该方法即可,不需要实现protocol中的所有方法。

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

推荐阅读更多精彩内容