面试题 - iOS中如何用Block实现一对多回调

        之前面试的时候被面试官问到 “如何使用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();   
    }];
 }

文章内容引用自《iOS单例中 Block 回调一对多设计》
                         《一个关于单例的 Block 回调设计》