By

文章目录
  1. 1. 问题的由来
  2. 2. 实现原理
  3. 3. MessageThrottle 的特殊定制
  4. 4. Reference

我的 Objective-C 消息节流防抖库 MessageThrottle 需要实现一个特性：当 MTRule 的 target 释放后，自动调用 MTRule 的 discard 方法。后来使用了业界很早就已有的方案：Associated Object，在这里整理下相关的知识点。

问题的由来

起初的思路是考虑到 MTRule 的 target 属性是 weak 的，想在其释放之前，也就是 target 变成 nil 之前调用 MTRule 对象的 discard 方法。然而 target 被释放赋值为 nil 的操作并不能通过 KVO 之类来监听，因为其并不是在外部通过 set 方法，这涉及到 weak 的实现原理（PS: 可以查看源码中 weak_clear_no_lock() 函数的实现）。于是问题转而变成了『在对象销毁前得到通知』。

接着我在 MacRumors 上找到了一篇 2005 年的贴子。大概内容就是讲通过 KVO 监听 retainCount 属性纯属失了智，众所周知 retainCount 不能真实反映对象内存管理的情况，即便 retainCount 为 1 的时候收到了 release 消息，也会直接 dealloc 掉，并不会变成 0。接着又有人说干脆 hook 下 dealloc 方法，然后抛通知，但是这样不安全。直到 2008 年 DenNukem 回帖说他直到咋办啦，用 Associated Object！

实现原理

当一个对象（Host）释放后，其关联的对象（Associated Object）也会被解除。可以在 Host 对象上添加 Associated Object，策略用 OBJC_ASSOCIATION_RETAIN。由于只有 Host 持有了这个 Associated Object，当 Host 释放后 Associated Object 也会被释放。在 Associated Object 的 dealloc 方法中告知外界其 Host 对象已经释放。Perfect！

dealloc 方法的调用顺序是从子类到父类直至 NSObject 的，NSObject 的 dealloc 会调用 object_dispose() 函数，进而移除 Associated Object。具体的实现如下:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
id 
object_dispose(id obj)
{
    if (!obj) return nil;
	// 销毁对象
    objc_destructInstance(obj);    
  // 释放内存
    free(obj);

    return nil;
}

void *objc_destructInstance(id obj) 
{
    if (obj) {
        // Read all of the flags at once for performance.
        bool cxx = obj->hasCxxDtor();
        bool assoc = obj->hasAssociatedObjects();

        // This order is important.
        // C++ 析构
        if (cxx) object_cxxDestruct(obj);
        // 移除 Associated Object
        if (assoc) _object_remove_assocations(obj);
        // ARC 下调用实例变量的 release 方法，移除 weak 引用
        obj->clearDeallocating();
    }

    return obj;
}

也就是说 Associated Object 的 dealloc 调用时 Host 已经释放了，无法拿到关于 Host 对象的任何信息了。但这其实对于大部分场景已经足够了，给外界一个 block/delegate callback，甚至是抛通知都 OK 的。实现起来很简单，代码很少，网上也可以找到一些 MRC 或 ARC 下实现的示例代码。

虽说 Runtime 帮我们自动移除了 Associated Object，但对我这种平常几乎一直写 MRC 代码的人来说还真有点不适应，毕竟脑子里时刻警惕着：每一次 retain 都要配套来一次 release 或 autorelease。

MessageThrottle 的特殊定制

好，又回归到文章最开头的问题。现在解决了『MTRule 的 target 释放后，自动调用 MTRule 的 discard 方法』的问题。但是，要注意到此时 target 属性都释放了，于是就无法提供 discard 方法正确执行做需要的信息。所以需要在 Associated Object 中加入一些属性来保存一些执行 discard 时所需必要的信息。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
@interface MTDealloc : NSObject
// 这三个属性就是 discardRule:whenTargetDealloc: 方法将要用到的信息。这个方法会把 rule 从 MTEngine 列表中移除，并按需要还原之前的 hook 操作。这些细节不是重点，重点就是调用 discard 需要这仨属性。
@property (nonatomic, weak) MTRule *rule;
@property (nonatomic, copy) NSString *methodDescription;
@property (nonatomic) Class cls;

@end

@implementation MTDealloc

- (void)dealloc
{
// 我只是觉得这样写代码就没警告了，而且还骚
    SEL selector = NSSelectorFromString(@"discardRule:whenTargetDealloc:");
    ((void (*)(id, SEL, MTRule *, MTDealloc *))[MTEngine.defaultEngine methodForSelector:selector])(MTEngine.defaultEngine, selector, self.rule, self);
}

@end

然后只需要在 applyRule 的时候初始化和配置好 MTDealloc 对象，并将其关联到 target 上即可：

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
// applyRule 时调用这个方法
static void mt_configureTargetDealloc(MTRule *rule)
{
    if (mt_object_isClass(rule.target)) {
        return;
    }
    else {
        Class cls = object_getClass(rule.target);
        MTDealloc *mtDealloc = objc_getAssociatedObject(rule.target, rule.selector);
        if (!mtDealloc) {
            mtDealloc = [MTDealloc new];
            mtDealloc.rule = rule;
            mtDealloc.methodDescription = mt_methodDescription(rule.target, rule.selector);
            mtDealloc.cls = cls;
            objc_setAssociatedObject(rule.target, rule.selector, mtDealloc, OBJC_ASSOCIATION_RETAIN);
        }
    }
}

使用 rule.selector 作为 Key 的目的就是让 target 对象上的每一个方法都对应一个关联对象，不会搞混。

感兴趣的可以查看 MessageThrottle 的源码，或者阅读我的上一篇文章 Objective-C Message Throttle and Debounce，更详细地讲述了 Objective-C 消息节流防抖的实现原理。这里只是对其实现自动 discard 原理的补充。

Reference

文章目录
  1. 1. 问题的由来
  2. 2. 实现原理
  3. 3. MessageThrottle 的特殊定制
  4. 4. Reference