之前面试的时候被面试官问到 “如何使用block实现一对多回调”，那会儿只知道回调分两种，第一种是一对一的，就是常用的代理和block；另一种是一对多的，就是通知。
在网上查阅资料后记录下
设计思路
1、使用了NSMapTable这个类来持有block，NSMapTable有个特性是可以对其key和value弱引用，在这种情况下当key或者value被释放的时候，此entry会自动从NSMapTable中移除。
2、block 持有者为单例中的 NSMapTable ，而非由注册 block 回调对象 observer 持有，并且单例本身仅维护 block 映射关系。
3、为了解决 block 自动释放问题，由 NSMapTable 来持有 block ，通过给 observer 绑定一个对象 DeallocWatcher ，利用objc_setAssociatedObject把 observer 与绑定对象 DeallocWatcher 进行关联，以此监听 DeallocWatcher 的 dealloc 释放，从而间接得知 observer 释放时机，达到 block 自动释放目的。
文章中提到的间接监听释放时机，在 ReactiveCocoa 中的onExit方法也是类似的思路来实现。
实现步骤
1、创建 NSMapTable 映射表
// key为 observer 注册对象，用 weak 属性表示不持有 observer，仅指向 observer// value 为 observer 注册的 block 回调，使用 strong 属性意味着映射表要持有 blockself.blockTable = [[NSMapTable alloc] initWithKeyOptions:NSPointerFunctionsWeakMemory valueOptions:NSPointerFunctionsStrongMemory capacity:1];
2、声明 observer 要绑定的对象 DeallocWatcher 类实现方法
@interface DeallocWatcher: NSObject
@property (nonatomic, copy) dispatch_block_t deallocCallback;
- (instancetype)initWithDeallocCallback:(dispatch_block_t)callback;
@end
@implementation DeallocWatcher
- (instancetype)initWithDeallocCallback:(dispatch_block_t)callback {
self = [super init];
if (self) {
self.deallocCallback = callback;
}
return self;
}
// 关键代码，当该对象释放触发 dealloc 方法时，会去执行 callback 回调
- (void)dealloc{
if (self.deallocCallback) {
self.deallocCallback();
}
}
@end
3、给 observer 添加关联绑定对象 watch，并添加至映射表中。
- (void)addObserver:(id)observer callback:(isPlayingChangedBlock)callback {
// 这里要打破循环引用，因为关联代码中 watch 被 observer 持有，而 watch 中的 callback 去调用了 observer
__weak typeof (observer) weakObserver = observer;
DeallocWatcher *watch = [[DeallocWatcher alloc] initWithDeallocCallback:^{
__strong typeof (observer) strongObserver = weakObserver;
[self removeObserver:strongObserver];
}];
[self.blockTable setObject:callback forKey:observer];
// 将 observer 与 watch 进行绑定关联，key 则使用 observer 指针指向的内存地址
objc_setAssociatedObject(observer,(__bridge const void * _Nonnull)([NSString stringWithFormat:@"%p", observer]), watch, OBJC_ASSOCIATION_RETAIN_NONATOMIC);}
在调用者看来，self 并没有持有 callback，即使 callback 中直接使用 self 也不会造成循环引用。但这是一个误区，在单例中，我们实际将 callback 绑定给了 self。所以 callback 中还是必须使用 weakSelf 才行。目前没想到好的方案。
4、observer 自动释放（后续更正：此处代码已经没必要执行。原因：当 watch 的 block 中执行 remove 方法时，这里 observer 已经释放了，手动关联的watch对象只是触发了dealloc回调，映射表中因为以observer为key弱引用存储，也在表中删除了该key值以及value值）
- (void)removeObserver:(id)observer {
[self.blockTable removeObjectForKey:observer];
objc_setAssociatedObject(observer, (__bridge const void * _Nonnull)([NSString stringWithFormat:@"%p", observer]), nil, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
}
5、当需要去执行映射表中的block代码块时，遍历映射表并执行已有的block块，最终达到单例中的 block 实现一对多的目的。
- (void)runBlockMethod {
// 当需要去执行映射表中的block代码块时，遍历映射表并执行已有的block块
[[[self.blockTable objectEnumerator] allObjects] enumerateObjectsUsingBlock:^(multiObserverBlock callback, NSUInteger idx, BOOL * _Nonnull stop) {
callback();
}];
}
网友评论