原文首发链接:Swoole 源码分析之 Timer 定时器模块
大家好,我是码农先森。
引言
Swoole 中的毫秒精度的定时器。底层基于 epoll_wait
和 setitimer
实现,数据结构使用最小堆,可支持添加大量定时器。
在同步 IO 进程中使用 setitimer
和信号实现,如 Manager
和 TaskWorker
进程,在异步 IO 进程中使用 epoll_wait/kevent/poll/select
超时时间实现。
定时器的添加和删除,全部为内存操作。在官方的基准测试脚本中,添加或删除 10 万个随机时间的定时器耗时为 0.08s 左右,因此性能是非常高效的。
源码拆解
我们在分析源代码之前,先看这段使用定时器的代码。Timer::after
函数是设置一个一次性的定时器,也就是执行一次就结束了,常用于执行一次性任务的场景。Timer::tick
函数会每间隔一段时间执行一次,类似一个闹钟的机制,常用于需要定时执行任务的场景。
<?php
// 设置一个一次性定时器
Swoole\Timer::after(1000, function(){echo " timer after timeout\n";
});// 设置一个间隔时钟定时器
Swoole\Timer::tick(1000, function(){echo "timer tick timeout\n";
});
按照之前分析源代码的策略,先对整个源码的调用流程进行梳理,以便于让我们有个整体的印象,调用流程如下图所示。
swoole_timer.cc
这个源码文件中定义了两个函数 swoole_timer_after
、swoole_timer_tick
。从这段代码中可以看出唯一的区别是,在调用 timer_add
函数时的传参有所不同,一个是 false
,一个是 true
,表示的是是否需要持久化的执行任务。另外 timer_add
函数实现了一些根据细化的逻辑,例如:参数的解析、一些检查判断的工作。最后,根据 persistent
参数判断是否执行持久化的操作。
// 定义 PHP 函数 swoole_timer_after
// swoole-src/ext-src/swoole_timer.cc:221
static PHP_FUNCTION(swoole_timer_after) {timer_add(INTERNAL_FUNCTION_PARAM_PASSTHRU, false);
}// 定义 PHP 函数 swoole_timer_tick
// swoole-src/ext-src/swoole_timer.cc:225
static PHP_FUNCTION(swoole_timer_tick) {timer_add(INTERNAL_FUNCTION_PARAM_PASSTHRU, true);
}// 添加定时任务到定时器中, 并根据持久性标志判断是否需要一直执行
// swoole-src/ext-src/swoole_timer.cc:155
static void timer_add(INTERNAL_FUNCTION_PARAMETERS, bool persistent) {zend_long ms;Function *fci = (Function *) ecalloc(1, sizeof(Function));TimerNode *tnode;// 解析参数ZEND_PARSE_PARAMETERS_START(2, -1)Z_PARAM_LONG(ms)Z_PARAM_FUNC(fci->fci, fci->fci_cache)Z_PARAM_VARIADIC('*', fci->fci.params, fci->fci.param_count)ZEND_PARSE_PARAMETERS_END_EX(goto _failed);// 检查定时器值 ms 是否小于预定义的最小值 SW_TIMER_MIN_MSif (UNEXPECTED(ms < SW_TIMER_MIN_MS)) {php_swoole_fatal_error(E_WARNING, "Timer must be greater than or equal to " ZEND_TOSTR(SW_TIMER_MIN_MS));_failed:efree(fci);RETURN_FALSE;}// 进行额外的检查// no server || user worker || task process with async modeif (!sw_server() || sw_server()->is_user_worker() ||(sw_server()->is_task_worker() && sw_server()->task_enable_coroutine)) {php_swoole_check_reactor();}// 使用指定的毫秒数、持久性标志、回调函数 timer_callback 和函数指针 fci 添加一个定时器tnode = swoole_timer_add((long) ms, persistent, timer_callback, fci);if (UNEXPECTED(!tnode)) {php_swoole_fatal_error(E_WARNING, "add timer failed");goto _failed;}// 为定时器节点 tnode 设置类型和析构函数tnode->type = TimerNode::TYPE_PHP;tnode->destructor = timer_dtor;// 根据持久性标志,会一直执行定时的任务if (persistent) {if (fci->fci.param_count > 0) {uint32_t i;zval *params = (zval *) ecalloc(fci->fci.param_count + 1, sizeof(zval));for (i = 0; i < fci->fci.param_count; i++) {ZVAL_COPY(¶ms[i + 1], &fci->fci.params[i]);}fci->fci.params = params;} else {fci->fci.params = (zval *) emalloc(sizeof(zval));}fci->fci.param_count += 1;ZVAL_LONG(fci->fci.params, tnode->id);} else {// 只会执行一次sw_zend_fci_params_persist(&fci->fci);}sw_zend_fci_cache_persist(&fci->fci_cache);RETURN_LONG(tnode->id);
}
在 timer.cc
源码文件中 swoole_timer_add
这个函数会检查是否已经有可用的定时器管理对象,如果没有的话会进行实例化创建一个,然后通过 SwooleTG.timer->add()
方法添加一个定时器任务。
// 这段代码用于添加一个定时器到 Swoole 框架中的定时器管理器中
// swoole-src/src/wrapper/timer.cc:40
TimerNode *swoole_timer_add(long ms, bool persistent, const TimerCallback &callback, void *private_data) {// 这里检查定时器是否可用if (sw_unlikely(!swoole_timer_is_available())) {// 如果定时器不可用,则会创建一个新的对象SwooleTG.timer = new Timer();// 并对其进行初始化if (sw_unlikely(!SwooleTG.timer->init())) {// 若初始化失败,就会释放内存delete SwooleTG.timer;SwooleTG.timer = nullptr;return nullptr;}}// 调用定时器对象的 add 方法,向定时器中添加一个定时器return SwooleTG.timer->add(ms, persistent, private_data, callback);
}
这个函数 *Timer::add
会构建一个新的定时器节点,并且设置一些属性值,例如:类型、执行时间、回调函数等。最后,会将定时器节点加入到最小堆的数据结构中。
// 用于向定时器管理器中添加一个新的定时器节点
// swoole-src/src/core/timer.cc:106
TimerNode *Timer::add(long _msec, bool persistent, void *data, const TimerCallback &callback) {// 检查传入的毫秒数 _msec 是否小于等于 0if (sw_unlikely(_msec <= 0)) {swoole_error_log(SW_LOG_WARNING, SW_ERROR_INVALID_PARAMS, "msec value[%ld] is invalid", _msec);return nullptr;}// 获取当前相对毫秒数,并检查其是否小于 0int64_t now_msec = get_relative_msec();if (sw_unlikely(now_msec < 0)) {return nullptr;}// 创建一个新的定时器节点 tnode// 并设置节点的数据、类型、执行时间、间隔、状态、回调函数、轮数以及析构函数TimerNode *tnode = new TimerNode();tnode->data = data;tnode->type = TimerNode::TYPE_KERNEL;tnode->exec_msec = now_msec + _msec;tnode->interval = persistent ? _msec : 0;tnode->removed = false;tnode->callback = callback;tnode->round = round;tnode->destructor = nullptr;// 更新下一个计划触发时间// 如果当前没有下一个计划或者新的时间比当前下一个计划更早// 则更新为新的时间。if (next_msec_ < 0 || next_msec_ > _msec) {set(this, _msec);next_msec_ = _msec;}// 给定时器节点分配一个唯一的IDtnode->id = _next_id++;if (sw_unlikely(tnode->id < 0)) {tnode->id = 1;_next_id = 2;}// 将节点加入堆中,同时更新堆的索引tnode->heap_node = heap.push(tnode->exec_msec, tnode);if (sw_unlikely(tnode->heap_node == nullptr)) {delete tnode;return nullptr;}// 记录节点信息map.emplace(std::make_pair(tnode->id, tnode));swoole_trace_log(SW_TRACE_TIMER,"id=%ld, exec_msec=%" PRId64 ", msec=%ld, round=%" PRIu64 ", exist=%lu",tnode->id,tnode->exec_msec,_msec,tnode->round,count());// 返回新添加的定时器节点return tnode;
}
总结
- Swoole 中实现了毫秒精度的定时器,而原生的 PHP 中只支持到秒级别。
- 数据结构使用最小堆支持添加大量定时器,全部为内存操作且十分高效。
- 定时器在实际的业务场景中应用也是非常广泛,常用于延时或定时执行的任务中,例如:订单超时未付款自动取消等场景。