面试题 - 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 回调设计》

你可能感兴趣的:(面试题 - iOS中如何用Block实现一对多回调)