iOS实录7：iOS通知中心的替换方案
[这是第7篇]
导语：在某些场景下，会选择使用通知中心去传递数据，传递的数据格式是字典，放到userInfo参数中，对于习惯传递数据对象的我们来说，希望有类似通知中心的方案，可以支持传递任意数据对象。基于此写了QSMessageCenter方案。
一、NSNotificationCenter的使用体会####
-
通知中心的使用分为三步：第一步，注册观察者，可以为它指定一个处理方法，名字和（传递的）字典对象；第二步，发送通知的消息，观察者收到通知后，会执行处理方法；第三步，移除观察者。
-
只要注册了观察者身份，就必须要移除。有些情况下，如在Controller内存泄漏的情况下，即时成对注册和移除了，注册观察的处理方法会被执行多次。所以，如果发现注册观察的处理方法被执行多次，先检查是否成对注册和移除观察者，其次就是检查是否发生内存泄漏。
-
NSNotificationCenter不能直接传递对象，就算想传递，也是封装在字典里，然后放到postNotificationName:object:userInfo:的userInfo参数中，放进去也麻烦，取出来解析也麻烦。
二、QSMessageCenter方案####
1、QSMessageCenter方案的背景
-
由于项目中大量存在列表页和详情页，使用了UITableView构建UI界面，(思路详参iOS实录1：使用UITableView构建UI界面)。为了复用某些视图，将一个大的视图拆散分到若干个UITableViewCell视图中。
-
因为若干个UITableViewCell视图组成业务需要的大视图，业务上这几个UITableViewCell视图是有相互关联的。如点击某子视图(UITableViewCell)，另一个子视图(UITableViewCell)的数据展示发生变化。这就导致了需要选择方案去实现部分UITableViewCell之间的通信。
-
最开始的方案就是选用了通知中心。因为它不仅可以在不同的UITableViewCell之间传递数据的，还降低了代码耦合。但是通知中心的使用特点导致了UITableViewCell大量存在注册观察，移除观察这样的代码（移除的工作还非常重要）。通知中心传递的参数又不可以灵活地使用model对象。当时的考虑是：重写通知中心，或者为找到iOS通知中心的替换方案。这是搞QSMessageCenter方案的最初的出发点和落地点。
-
综上，QSMessageCenter方案的目标很直接：1）实现不同对象之间的数据传递；2）传递的参数可以是任何的对象,包括数据model，数组,字典等；3）注册后不需要手动移除； 4）使用简单和代码耦合低。
2、QSMessageCenter方案的具体实现#####
-
主要思路：指定观察者时，将观察者和key成对存放在字典结构中，根据指定key发出消息时，就可以从这个字典结构中取出观察者对象，然后执行观察者的处理办法。
-
具体实现：字典结构使用NSMapTable，因为NSMapTable不仅有着和NSDictionary相似的数据结构，还可以指定key是强引用，value是弱引用这个特性。当value被dealloc时，NSMapTable中存的value是nil；而NSDictionary会复制keys并且通过强引用values来实现存储，不考虑使用。
-
代码组成：QSMessageCenter类 + NSObject+QSMessageCenter分类
1）QSMessageCenter类的实现
//QSMessageCenter.h
@protocol QSMessageCenterDelegate <NSObject>
@optional
/**
接收到message对象
*/
- (void)qsReceiveMessage:(id)message messageId:(NSString *)msgId;
@end
@interface QSMessageCenter : NSObject
/**
添加消息接收者
@param msgReceiver 消息接收对象
@param receiverKey 消息接收对象key
*/
+ (void)addMessageReceiver:(id<QSMessageCenterDelegate>)msgReceiver receiverKey:(NSString *)receiverKey;
/**
发送消息
@param msg 消息对象
@param msgId 消息id
@param receiverKey 消息接收对象key
*/
+ (void)sendMessage:(id)msg messageId:(NSString *)msgId receiverKey:(NSString *)receiverKey;
@end
//QSMessageCenter.m
static NSMapTable *weakMessageCenterMapTable = nil;
@implementation QSMessageCenter
+ (void)initialize{
if (self == [QSMessageCenter class]){
// 强引用key & 值弱引用object
weakMessageCenterMapTable = [NSMapTable strongToWeakObjectsMapTable];
}
}
#pragma mark - 添加 && 删除
/**
添加消息接收者
*/
+ (void)addMessageReceiver:(id<QSMessageCenterDelegate>)msgReceiver receiverKey:(NSString *)receiverKey{
if ([weakMessageCenterMapTable objectForKey:receiverKey] == nil){
// 添加对象进weak集合
[weakMessageCenterMapTable setObject:msgReceiver forKey:receiverKey];
}
}
/**
发送消息
*/
+ (void)sendMessage:(id)msg messageId:(NSString *)msgId receiverKey:(NSString *)receiverKey{
if (!CHECK_VALID_STRING(receiverKey)) {
return;
}
id<QSMessageCenterDelegate> object = [weakMessageCenterMapTable objectForKey:receiverKey];
if (!object){
//对象未添加或对象已被释放,则移除掉这个键值
[weakMessageCenterMapTable removeObjectForKey:receiverKey];
}else{
if ([object respondsToSelector:@selector(qsReceiveMessage:messageId:)]) {
if (!CHECK_VALID_STRING(msgId)) {
msgId = DEFAULT_MESSAGE_ID;
}
[object qsReceiveMessage:msg messageId:msgId];
}
}
}
#pragma mark - private methods
+ (void)printMessageCenterMapTable{
NSArray *keys = [[weakMessageCenterMapTable keyEnumerator] allObjects];
[keys enumerateObjectsUsingBlock:^(NSString *key, NSUInteger idx, BOOL * _Nonnull stop) {
id obj = [weakMessageCenterMapTable objectForKey:key];
NSLog(@"(key)%@-->(value)%@",key,obj);
}];
}
@end
2）NSObject+QSMessageCenter分类的实现
//NSObject+QSMessageCenter.h
@interface NSObject (QSMessageCenter)
/**
注册成为消息接受者
@param receiverKey 接受者key,值为nil时，key值选用注册者对象的类名
*/
- (void)registerMessageReceiverWithKey:(NSString *)receiverKey;
/**
发送者发送消息，并指定接收者
@param msg 消息对象，可以是任意对象
@param msgId 消息对象id
@param receiverKey 接受者key，值为nil时,receiverKey选用发送者对象的类名
*/
- (void)sendMessage:(id)msg messageId:(NSString *)msgId receiverKey:(NSString *)receiverKey;
@end
//NSObject+QSMessageCenter.m
@interface NSObject ()<QSMessageCenterDelegate>
@property (nonatomic, weak) id<QSMessageCenterDelegate> QSMessageCenterDelegate;
@end
@implementation NSObject (QSMessageCenter)
- (void)setQSMessageCenterDelegate:(id<QSMessageCenterDelegate>)QSMessageCenterDelegate{
objc_setAssociatedObject(self, @selector(QSMessageCenterDelegate), QSMessageCenterDelegate, OBJC_ASSOCIATION_ASSIGN);
}
- (id<QSMessageCenterDelegate>)QSMessageCenterDelegate{
return objc_getAssociatedObject(self, @selector(QSMessageCenterDelegate));
}
#pragma mark - 成为 & 取消成为 消息接受者
/**
成为消息接受者
*/
- (void)registerMessageReceiverWithKey:(NSString *)key{
//将自己设置成为代理
self.QSMessageCenterDelegate = self;
if (!CHECK_VALID_STRING(key)) {
key = [self getClassName];
}
[QSMessageCenter addMessageReceiver:self.QSMessageCenterDelegate
receiverKey:key];
}
/**
发送者发送消息，并指定接收者
*/
- (void)sendMessage:(id)msg messageId:(NSString *)msgId receiverKey:(NSString *)receiverKey{
if (!CHECK_VALID_STRING(receiverKey)) {
receiverKey = [self getClassName];
}
[QSMessageCenter sendMessage:msg messageId:msgId receiverKey:receiverKey];
}
#pragma mark - private methods
- (NSString *)getClassName{
return NSStringFromClass([self class]);
}
@end
3、QSMessageCenter方案的使用#####
核心使用步骤如下：
1、在观察者中注册，指定receiverKey
[self registerMessageReceiverWithKey:receiverKey];
2、发送消息，指定receiverKey，当注册方法和发送消息中的receiverKey相同，才可以把消息发送给观察者
[self sendMessage:message messageId:messageId receiverKey:receiverKey];
3、在观察者类中实现qsReceiveMessage:messageId:方法
- (void)qsReceiveMessage:(id)message messageId:(NSString *)msgId{
//可以根据msgId去做不同处理
}
补充：在实现中，还是加上了移除观察者的办法，但是可以不使用。因为weakMessageCenterMapTable中存的value的对象引用是弱引用。在观察者被dealloc的情况下，还给这个观察者发消息，由于观察者已经dealloc，weakMessageCenterMapTable根据key值找不到观察者对象，会删除key-value这个object。
在Demo中使用了QSMessageCenter方案， 完成数据的跨UI传输。UI界面如下：
QSUseMessageCenterDemo.png
注解：
-
在Controller中，内容展示视图contentView含有四个Label对象，contentView没有更新Label展示数据的接口。使用QSMessageCenter实现数据的传输。
-
contentView在初始化的时候，会将自己注册成为receiverKey值是QSShowContentView的观察者，当接收到QSMessageCenter发来的receiverKey为QSShowContentView的消息会根据messageID来处理。messageID代表更新数据，就更新数据展示，代表重置就重置数据展示。
-
在contentView的下面有两个上下排列的输入框中，分别对应contentView中label的索引号(0,1,2,3) 和需要显示的数据。输入数据后，点击“更新数据”按钮，对数据进行基本的校验，通过之后，就会把数据封装成自定义的model，交给已经注册receiverKey为QSShowContentView的观察者。点击“重置数据”按钮处理类似
-
此时，由于contentView已经注册为观察者，所以它会去处理收到的消息对象。
-
具体代码见源码
源码直通车：QSUseMessageCenterDemo
感觉还可以的话，欢迎star
