目录
- 摘要
- 1. 架构与文件构成
- 2. TimerTask
- 3. bug1:构造顺序
- 4. bug2:TimerAdd 与 Refresh
- 5. bug5:TickRun
- 6. bug3、bug4:两套索引
- 7. 总结
- 8. 常见问题
- 9. 标签
# timewheel.cpp 源码走查:TimerTask 与 TimeWheel
摘要
单文件时间轮:TimerTask 持 id、超时与回调,析构中执行任务;TimeWheel 用 _timers 与 _wheel
两套索引,外部每秒 TickRun 驱动。
1. 架构与文件构成
两个类:TimerTask、TimeWheel。_timers 管 id,_wheel 管时间位置。main、TimerAdd、每秒 TickRun;到期时 clear() 释放该位置 PtrTask,无其他强引用则 ~TimerTask 执行任务,_release 回调 RemoveTimer。
【流程图占位:全链路;节点:main 循环 sleep → TickRun → (_tick+1)%_capacity → clear → ~TimerTask → RemoveTimer】
2. TimerTask
_cancel 决定两处:析构中为 true 时提前 return,回调均不执行;SetRealese 拒绝设置。SetCancel 状态未变时返回 false。
摘录(timewheel.cpp,TimerTask::SetCancel):
boolSetCancel(boolb){if(b==_cancel){spdlog::info("SetCancel error 已经set过");returnfalse;}_cancel=b;returntrue;}3. bug1:构造顺序
声明序 _tick、_capacity、_timers、_wheel;构造列表 _wheel(_capacity) 依赖 _capacity 先声明。
摘录(timewheel.cpp,TimeWheel 私有成员):
private:int_tick;// 滴答下标uint64_t_capacity;// 表盘最大数量---其实就是最大延迟时间// Hash-通过id进行管理 blog分析🔍销毁(shared_ptr与weak_ptr)std::unordered_map<uint64_t,WeakTask>_timers;// <id,weak_ptr> 管理销毁与创建std::vector<std::vector<PtrTask>>_wheel;// 二维时间轮【bug1】:未顾及初始化顺序,错误的以“读顺序优先”
4. bug2:TimerAdd 与 Refresh
TimerAdd:查 _timers 去重,SetRealese 绑 RemoveTimer+this+id,插入 _wheel[delay]。Refresh:pos = 超时 + _tick 取模 _capacity,再头插同一 PtrTask。
摘录(timewheel.cpp,TimeWheel::TimerAdd):
pt->SetRealese(std::bind(&TimeWheel::RemoveTimer,this,id));// _wheel[delay].push_back (pt);_wheel[delay].insert(_wheel[delay].begin(),pt);【bug2】:对结构不熟悉,写代码纯粹的不细心。
5. bug5:TickRun
TickRun:遍历 _wheel[(_tick + 1) % _capacity],逐个 SetCancel(false);返回 true 记 flag–,flag < 0 时打印并 return,_tick 不前进;否则 _tick 前进一格并 clear()。
摘录(timewheel.cpp,TimeWheel::TickRun):
for(auto&e:_wheel[(_tick+1)%_capacity]){// if (!e->SetCancel (true))可以吗? bug5 不可以!!if(e->SetCancel(false))【bug5】:想要复用逻辑,复用仅存在一种有效方法
【流程图占位:TickRun 单次推进;节点:定下标 (_tick+1)%_capacity → 遍历 SetCancel → 失败计数 → _tick 前进 → clear()】
6. bug3、bug4:两套索引
RemoveTimer 经 _release 调用:find 失败打印 remove error,成功则 erase(id)。SetCanCel:lock() 取 GetTimeOut() 得 pos,再遍历 _wheel[pos] 逐个 SetCancel(true)。
摘录(timewheel.cpp,TimeWheel::RemoveTimer / SetCanCel):
autoit=_timers.find(id);if(it==_timers.end()){spdlog::error("remove error");return;}uint32_tpos=_timers[id].lock()->GetTimeOut();【bug3】
【bug4】
7. 总结
结构:任务状态 + 两套索引 + 外部驱动;
8. 常见问题
- lock() 为何有效?(_wheel 强引用持有)
- SetCancel 返回 false?(b == _cancel 时)
- _wheel[id] 能定位吗?(不能,下标是延迟距离)
9. 标签
时间轮 / C++ / 智能指针 / weak_ptr