MessageThrottle 结合业务需求的修改

原文地址

MessageThrottle 是什么, 解决了什么痛点

MessageThrottle 是使用 Objective-C 实现的 iOS 函数节流(Throttle)和防抖(Debounce)的工具库. 在实现 App 交互的时候, 逻辑的刷新太频繁会导致性能上的问题, 比如不断有新的数据加载导致列表一直在刷新, 按钮的点击速度太快导致逻辑被多次执行, Slider 或者 滑动手势等频繁回调导致性能跟不上, 数据一直在变化导致耗费诸多性能在中间状态上等. 在业务层, 我们经常要结合业务来做限频或者定时刷新等逻辑, 来解决上面这些问题, 但是 MessageThrottle 提供了新的思路, 在不耦合业务逻辑的情况下, 实现间隔性消息调用, 即函数节流(Throttle)和防抖(Debounce).

使用 MessageThrottle, 只需如下简单的代码

#import "MessageThrottle.h"

// 设置 foo: 每秒调用一次, 默认是防抖模式, 此行代码调用一次即可
[object mt_limitSelector:@selector(foo:) oncePerDuration:1.0]; 

- (void)foo:
{
    // todo: xxx
}

// 业务代码, 正常调用 foo: 即可
- (void)otherFunction
{
    [self foo:params];
}

MessageThrottle 实现原理

MessageThrottle 是一个实现函数节流和防抖的工具库, 那么节流和防抖有什么不同呢? 以及 MessageThrottle 怎么来实现对消息的节流的?

节流和防抖有什么不同?

MessageThrottle 的思路源自 Underscore.js.
节流(Throttle) 在 Underscore.js 中的解释如下:

Creates and returns a new, throttled version of the passed function, that, when invoked repeatedly, will only actually call the original function at most once per every wait milliseconds. Useful for rate-limiting events that occur faster than you can keep up with.
By default, throttle will execute the function as soon as you call it for the first time, and, if you call it again any number of times during the wait period, as soon as that period is over.
创建并返回一个像节流阀一样的函数,当重复调用函数的时候,至少每隔 wait 毫秒调用一次该函数
默认情况下,throttle 将在你调用的第一时间尽快执行这个 function,并且,如果你在 wait 周期内调用任意次数的函数,都将尽快的被覆盖。

防抖(Debounce) 的解释如下:

Creates and returns a new debounced version of the passed function which will postpone its execution until after wait milliseconds have elapsed since the last time it was invoked. Useful for implementing behavior that should only happen after the input has stopped arriving. For example: rendering a preview of a Markdown comment, recalculating a layout after the window has stopped being resized, and so on.
创建并返回一个防抖动的函数, 在最后一次执行时间的 wait 毫秒之后才真正执行. 在需要不会再有输入(一般是用户行为等无法预测的改变)的情况下执行原有方法. 比如渲染一个 Markdown 格式的评论预览, 只有当窗口停止改变大小之后才重新计算, 等等.

简单说, 节流就像是公交车每 10 分钟发一班车; 一直没人上车就等待, 第一个人上车后立刻发车, 期间到的人就不管了, 直到 10 分钟后进入等待状态, 来人就立刻发车并进入下一个 10 分钟周期. 防抖是有人上车就开始 10 分钟计时, 计时期间有人上车的话就重新开始 10 分钟计时, 直到计时结束都没人上车的话才发车.

MessageThrottle 怎么对消息做节流

在我的上一篇文章iOS MessageThrottle 防抖与 RAC 的冲突中对此问题以及可能导致的问题有简单的解析.
MessageThrottle 将消息的转发流程(Runtime 相关知识)与限频逻辑做了结合, 将原有的 foo: 方法的IMP 换成了_objc_msgForward, 从而直接触发消息转发流程, 在消息转发中调用动态生成的 hook 方法, 实现了自己的__mt_foo:方法, 在自己的方法中处理了限频逻辑. 实现细节参考文章MessageThrottle Safety源码即可.

MessageThrottle 实现了哪些节流防抖的功能

MessageThrottle 的三个模式

/**
 消息节流模式

 - MTPerformModeFirstly: Throttle 模式:执行最靠前发送的消息,后面发送的消息会被忽略
 - MTPerformModeLast: Throttle 模式:执行最靠后发送的消息,前面发送的消息会被忽略,执行时间会有延时
 - MTPerformModeDebounce: Debounce 模式:消息发送后延迟一段时间执行,如果在这段时间内继续发送消息,则重新计时
 */
typedef NS_ENUM(NSUInteger, MTPerformMode) {
    MTPerformModeFirstly,
    MTPerformModeLast,
    MTPerformModeDebounce
};

具体到代码以及注释如下

// MTPerformModeFirstly
// 如果当前的时间减去上一次执行的时间, 大于规定的周期时长, 才执行本次逻辑
if (now - rule.lastTimeRequest > rule.durationThreshold) {
    // 将生成的 __mt_foo: 方法覆盖原有的 selector
    invocation.selector = rule.aliasSelector;
    // 调用 __mt_foo:
    [invocation invoke];
    // 修改这个方法最后一次被执行的时间
    rule.lastTimeRequest = now;
    dispatch_async(rule.messageQueue, ^{
        // May switch from other modes, set nil just in case.
        rule.lastInvocation = nil;
    });
}
// MTPerformModeLast
// 不判断条件, 直接覆盖原有方法
invocation.selector = rule.aliasSelector;
// 引用调用参数, 防止异步执行 invoke 时候 crash, 参数会在 invoke 后自动释放,
[invocation retainArguments];
dispatch_async(rule.messageQueue, ^{
    // invocation 准备异步执行, 放到 rule 中以免被释放, 注意rule.lastInvocation 在周期内会被再次替换
    rule.lastInvocation = invocation;
    // 如果这次的时间距离上一次超过了一个周期才准备执行
    if (now - rule.lastTimeRequest > rule.durationThreshold) {
        rule.lastTimeRequest = now;
        // 会被推迟到这个周期的最后时刻来执行
        dispatch_after(dispatch_time(DISPATCH_TIME_NOW, (int64_t)(rule.durationThreshold * NSEC_PER_SEC)), rule.messageQueue, ^{
            if (!rule.isActive) {
                rule.lastInvocation.selector = rule.selector;
            }
            // 执行此次周期内最后一个 invocation
            [rule.lastInvocation invoke];
            rule.lastInvocation = nil;
        });
    }
});
// MTPerformModeDebounce
// 不判断条件, 直接覆盖原有方法
invocation.selector = rule.aliasSelector;
// 引用调用参数, 防止异步执行 invoke 时候 crash, 参数会在 invoke 后自动释放,
[invocation retainArguments];
dispatch_async(rule.messageQueue, ^{
    // 替换 lastInvocation, 注意只要不被执行, 并且有新的 invocation, 就会不断被替换
    rule.lastInvocation = invocation;
    // 延期一个周期
    dispatch_after(dispatch_time(DISPATCH_TIME_NOW, (int64_t)(rule.durationThreshold * NSEC_PER_SEC)), rule.messageQueue, ^{
        // 如果 invocation 被替换了, 就不执行了
        if (rule.lastInvocation == invocation) {
            if (!rule.isActive) {
                rule.lastInvocation.selector = rule.selector;
            }
            // 没有被替换, 说明这一个周期内没有来新的调用, 直接执行
            [rule.lastInvocation invoke];
            rule.lastInvocation = nil;
        }
    });
});

我们通常需要一个怎么样的节流器

如上面代码注释所言, MessageThrottle 提供的两个 Throttle 模式均会丢弃最前面或者最后面一次的消息, 而 Debounce 模式则是会一直延时执行, 但是在很多场景中, 我们希望节流器可以做到以下三点:

  1. 第一次消息能立即调用, 交互上可以让用户知道已经在开始变化了.
  2. 最后一次消息不应该被丢弃, 因为最后一次的消息是最新的, 丢弃会使状态处在一个中间状态, 这会给用户传递错误的信息.
  3. 中间状态应该有序均匀调用, 比如在频繁刷新的场景保持一秒刷一次.

怎么改造 MessageThrottle

基于 MessageThrottle, 我增加了第四种模式 MTPerformModeFirstlyPerDuration. 周期内第一条消息会被立刻执行, 最后一条消息会被延迟到下一个周期的开始执行, 然后中间的消息会被忽略. 这样满足了上面的三个要求.
下面直接上代码

case MTPerformModeFirstlyPerDuration: {
    if (now - rule.lastTimeRequest > rule.durationThreshold) {
        invocation.selector = rule.aliasSelector;
        [invocation invoke];
        rule.lastTimeRequest = now;
        dispatch_async(rule.messageQueue, ^{
            // May switch from other modes, set nil just in case.
            rule.lastInvocation = nil;
            // 一个周期后 检查是否执行最后一个被忽略的调用
            checkWhenNextDuration(rule);
        });
    } else {
        dispatch_async(rule.messageQueue, ^{
            if (rule.lastInvocation)
            {
                rule.lastInvocation = nil; // 清空之前设置的
            }
            invocation.selector = rule.aliasSelector;
            [invocation retainArguments];
            rule.lastInvocation = invocation; // 设置为需要延时执行
        });
    }
    break;
}

static void checkWhenNextDuration(MTRule *rule)
{
    dispatch_after(dispatch_time(DISPATCH_TIME_NOW, (int64_t)(rule.durationThreshold * NSEC_PER_SEC)), rule.messageQueue, ^{
        if (rule.lastInvocation)
        {
            if (!rule.isActive) {
                rule.lastInvocation.selector = rule.selector;
            }
            [rule.lastInvocation invoke];
            rule.lastInvocation = nil;
            
            NSTimeInterval now = [[NSDate date] timeIntervalSince1970];
            now += MTEngine.defaultEngine.correctionForSystemTime;
            rule.lastTimeRequest = now; // 重新计时
            
            checkWhenNextDuration(rule); // 再次延时检查是否需要执行
        }
    });
}

最后用一张图表示这四个模式:

duration 为 1 秒, 演示四个模式
x / ignore 表示忽略此次消息
delayTo 表示延时处理此次消息
√ / invoke 表示执行消息

MTPerformModeFirstly
0______________1______________2______________3______________
√______x________√_____x______x______√___x______x___
|      |        |     |      |      |   |      |
|      ignore   |     ignore ignore |   ignore ignore
invoke          invoke              invoke

MTPerformModeLast
0______________1______________2______________3______________
x______x_____√___x______√________√_________________
|      |     |   |      |        |
ignore ignore|   ignore |        |
             |          |        |
      delayTo->1   delayTo--->2  delayTo---->3
               |              |              |
               Invoke         Invoke         Invoke

MTModePerformDebounce
0______________1______________2______________3______________
x________x_____x__x________√______________________
|        |     |  |        |
ignore   ignore   ignore   |
                           |
                    delayTo--------------->Here
                              duration     |
                                           invoke

MTPerformModeFirstlyPerDuration
0______________1______________2______________3______________
√_____x______√_____x___x___√______x__x__√___________
|     |      |     |   |   |      |  |  |
|     ignore |     ignore  |      ignore|
|            |             |            |
|     delayTo->1    delayTo-->2  delayTo---->3
|              |              |              |
invoke         invoke         invoke         invoke

参考文章

  1. Objective-C Message Throttle and Debounce : http://yulingtianxia.com/blog/2017/11/05/Objective-C-Message-Throttle-and-Debounce/
  2. MessageThrottle Safety : http://yulingtianxia.com/blog/2018/07/31/MessageThrottle-Safety/
  3. Underscore.js : https://underscorejs.org/#throttle
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 203,098评论 5 476
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 85,213评论 2 380
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 149,960评论 0 336
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,519评论 1 273
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,512评论 5 364
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,533评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 37,914评论 3 395
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,574评论 0 256
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,804评论 1 296
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,563评论 2 319
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,644评论 1 329
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,350评论 4 318
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,933评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,908评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,146评论 1 259
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 42,847评论 2 349
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,361评论 2 342

推荐阅读更多精彩内容