计时器轮(TimerWheel),模拟时钟格式组成的高效计时器
TimerWheel算法原理
-
环形数据结构:TimerWheel,即时间轮,是一个环形的数据结构,类似于时钟的面,被等分为多个格子或槽位(slot)。
-
槽位时间间隔:每个槽位代表一个固定的时间间隔,例如1毫秒、1秒等。这个时间间隔决定了定时器的精度。
-
初始化:在算法开始时,需要初始化时间轮,包括设定时间轮的大小(即槽位的数量)和每个槽位代表的时间间隔。即当插入数据后即不允许修改时轮信息。
TimerWheel算法的优缺点
- 优点:
- 高效性:通过利用循环数组和指针的特性,时间轮算法可以以固定的时间复杂度(通常是O(1))来处理任务的调度和触发,这提供了一种高效的定时任务管理方式。
- 可扩展性:如果单个时间轮无法满足需求,可以通过增加时间轮的层数来扩展其表示的时间范围。多层时间轮可以处理更长时间跨度的定时任务。
- 缺点:
- 精度越高资源耗费多:时间轮的精度取决于每个槽位代表的时间间隔。通过调整这个时间间隔,可以在精度和资源消耗之间找到平衡。
- 删除效率:通常时轮未索引key值存储位置,删除需要遍历整个列表,复杂度为O(n)
结构设计
在TimerWheel的结构中,我们需要有两层结构,一层TimerWheel
管理整个类,用OneTimerWheel
来管理每个轮,此外定义了trait为Timer
,类型必须实现该接口以管理类的递进时长。
管理类
管理类里定义了时轮的最大最小及其它的信息
pub struct TimerWheel<T:Timer> {
/// 时轮的最大轮,以时钟为例就是时针
greatest: *mut OneTimerWheel<T>,
/// 时轮的最小轮,以时钟为例就是秒针
lessest: *mut OneTimerWheel<T>,
/// 时轮的最小间隔,以时间为例就是秒
min_step: usize,
/// 维护定时器id
next_timer_id: usize,
/// 离的最近的id
delay_id: usize,
/// 总共的递进步长,缓存优化触发
all_deltatime: usize,
/// 当时时轮里的元素个数
len: usize,
}
单轮管理
当前轮管理的当前指针及每个糟里的元素
pub struct OneTimerWheel<T:Timer> {
/// 当时指针指向的位置,如秒针指向3点钟方向
index: usize,
/// 当前结构的容量,如表示秒的为60的容量
capation: usize,
/// 当前结构步长,如分钟就表示60s的
step: usize,
/// 当前槽位容纳的元素
slots: Vec<Vec<Entry<T>>>,
/// 当前轮结构的父轮,如当前是分的,那父轮为时轮
parent: *mut OneTimerWheel<T>,
/// 当前轮结构的子轮,如当前是分的,那父轮为秒轮
child: *mut OneTimerWheel<T>,
/// 当前轮的名字,辅助定位
name: &'static str,
}
接口Timer
类必须时间Timer
结构,以统一知道他对当前时间的间隔。必须覆写when
以确定当前的offset。
pub trait Timer {
/// 当时与现在的间隔,以确定插入确定的槽
fn when(&self) -> usize;
/// 可能需要修改对象,此处用可变值
fn when_mut(&mut self) -> usize {
self.when()
}
}
源码分析
初始化
因为时轮是后续设置,最小刻度由最小数额确定。故初始化不需要任何参数:
impl<T:Timer> TimerWheel<T> {
pub fn new() -> Self {
Self {
greatest: ptr::null_mut(),
lessest: ptr::null_mut(),
next_timer_id: 0,
delay_id: 0,
min_step: 0,
all_deltatime: 0,
len: 0,
}
}
}
设置时轮
时轮从大到小添加,如时钟的模型先添加12小时每个刻度为3600秒。
let mut timer = TimerWheel::new();
timer.append_timer_wheel(12, 60*60, "HourWheel");
timer.append_timer_wheel(60, 60, "MinuteWheel");
接下来添加分钟的轮,槽位60个,每个槽位60秒:
timer.append_timer_wheel(60, 60, "MinuteWheel");
接下来添加秒钟的轮,槽位60个,每个槽位1秒:
timer.append_timer_wheel(60, 1, "SecondWheel");
这里必须保证,每个小的刻度的总和必须为上一个模型的一个小刻度。
完整源码,将在添加的时候自动维护最小刻度及下一个delay_id防止空转。
pub fn append_timer_wheel(&mut self, slots: usize, step: usize, name: &'static str) {
debug_assert!(self.len == 0, "必须时轮为空才可改变时轮");
let one = Box::into_raw(Box::new(OneTimerWheel::new(slots, step, name)));
self.delay_id = self.delay_id.max(slots * step);
self.lessest = one;
if self.greatest.is_null() {
self.greatest = one;
} else {
unsafe {
(*self.greatest).append(one);
}
}
self.min_step = step;
}
添加元素
元素必须实现Timer
,以确定放在指定的槽位中,将会返回定时器的id。以方便操作定时器,删除、获取、修改等操作。并且如果当前的时移更小,会重新设置delay_id的大小,以确定指定时间内能触发定时器。
pub fn add_timer(&mut self, mut val: T) -> usize {
debug_assert!(!self.greatest.is_null(), "必须设置时轮才能添加元素");
let timer_id = self.next_timer_id;
self.next_timer_id += 1;
self.delay_id = self.delay_id.min(val.when_mut());
let entry = Entry { val, id: timer_id };
unsafe {
(*self.greatest).add_timer(entry);
}
self.len += 1;
timer_id
}
确定时移
时移通过外部系统传入进去,可以方便的设置自己是否根据时间来进行判定。或者自己有另一套时移模式均可通用。
pub fn update_deltatime(&mut self, delta: usize) -> Option<Vec<T>>
如果当前的时移未大于delay_id
,那么即表示还未触发任何的定时器。此时处理的仅仅为all_deltatime
的自加,效率极高。
当时移大于delay_id
后表示将触发定时器,与最小的min_step
相除得出时轮中的时移offset
self.all_deltatime += delta;
let mut offset = self.all_deltatime / self.min_step;
if offset < self.delay_id {
return None;
}
self.all_deltatime -= offset * self.min_step;
确定后将开始自身的时移,这里用比较高效的方式维护各个时轮:
比如以标准时钟为例,当前offset为156,即表示此时已过了156秒,那么秒轮的将会产生156 % 60 = 36
的偏差值,那么秒轮的index=index+36
为新的秒轮,且因为156 / 60 > 2
,已经完整的转了一圈,表示秒轮上的所有节点全部被触发了。
此时分轮上移动了2分钟,比如此时为3,也就是index=index+2
,那么移动的两个位置(3和4)所有元素将会被触发。此时分针指向了5,因为该位置上的最大偏移已经不超过60秒,为了使接下来在60秒内能正确触发该对象的值。那么将5身上的所有元素移除重新加入到了秒轮。
let mut result = vec![];
let mut wheel = self.lessest;
while !wheel.is_null() {
unsafe {
(offset, remainder) = (*wheel).update_index(offset, remainder, &mut result);
if offset == 0 {
break;
}
wheel = (*wheel).parent;
}
}
更新索引值update_index
,因为每次递进的的梯度可能会非常的大的,也可能会非常的小,这边需要准确的更新索引的值,并且包括降维的数据也能正确的降维。其中余数则是高维往低维降维的关键,例如149,从秒轮上来可以看出分轮有2分钟29秒,将会把第3分钟的进行降维,其中第三分钟如果在29内的标签应该是均命中,大于29的应该-29后的值插入到秒轮。
pub fn update_index(&mut self, offset: usize, remainder: usize, result: &mut Vec<T>) -> (usize, usize) {
let next = self.index + offset;
let mut all = 0;
for idx in self.index..next {
if all > self.capation {
break;
}
all += 1;
let idx = idx % self.capation;
let list = &mut self.slots[idx];
for val in list.drain(..) {
result.push(val.val);
}
}
self.index = next % self.capation;
if !self.child.is_null() {
unsafe {
let list = &mut self.slots[self.index];
for mut val in list.drain(..) {
val.when = (val.when % self.step).saturating_sub(remainder);
if val.when <= 0 {
result.push(val.val);
} else {
(*self.child).add_step_timer(val);
}
}
}
}
(next / self.capation, next % self.capation + remainder)
}
此时我们收集到了时移产生156秒触发的所有元素,我们将其打包进行返回,与此同时因为时轮上的指针发生变化,及元素被重新拆解,所以此时需要重新维护delay_id
,此时需调用
pub fn calc_delay_id(&mut self)
密度高的基本为O(1)的复杂度,最差情况为O(n)的复杂度
总刻度数以时钟为计秒轮遍历60次,分轮遍历60次,时轮遍历12次,即最高遍历132次。
确定时移的时候也可以调用回调函数版本直接处理时间回调
pub fn update_deltatime_with_callback<F>(&mut self, delta: usize, f: &mut F)
where
F: FnMut(&mut Self, T),
移除定时器
当我们对某个定时器不需要的时候可以提前进行删除
pub fn del_timer(&mut self, timer_id: usize) -> Option<T>
该模型删除不具备优势,时间复杂度为O(n),n为当前容器内的个数。
需要频繁删除请选用其它时间框架。如红黑树模型(nginx)的删除复杂度仅为O(logn)。
参考示例
以下是按时钟的一个计时轮:
use algorithm::TimerWheel;
fn main() {
let mut timer = TimerWheel::new();
timer.append_timer_wheel(12, 60 * 60, "HourWheel");
timer.append_timer_wheel(60, 60, "MinuteWheel");
timer.append_timer_wheel(60, 1, "SecondWheel");
timer.add_timer(30);
assert_eq!(timer.get_delay_id(), 30);
timer.add_timer(149);
assert_eq!(timer.get_delay_id(), 30);
let t = timer.add_timer(600);
assert_eq!(timer.get_delay_id(), 30);
timer.add_timer(1);
assert_eq!(timer.get_delay_id(), 1);
timer.del_timer(t);
timer.add_timer(150);
assert_eq!(timer.get_delay_id(), 1);
let val = timer.update_deltatime(30).unwrap();
assert_eq!(val, vec![1, 30]);
timer.add_timer(2);
let val = timer.update_deltatime(119).unwrap();
assert_eq!(val, vec![2, 149]);
let val = timer.update_deltatime(1).unwrap();
assert_eq!(val, vec![150]);
assert!(timer.is_empty());
}
完整项目地址
https://github.com/tickbh/algorithm-rs
结语
TimerWheel算法通过其独特的数据结构和运行原理,实现了高效、可扩展且灵活的定时任务管理。该结构用于对高性能的定时器框架,尤其密集程度越高的定时器效率越高。
标签:mut,self,timer,60,源码,usize,TimerWheel,id,Rust From: https://www.cnblogs.com/wmproxy/p/18243240/TimerWheel