timer

Author: tickbh

examples/timer.rs

@@ -2,9 +2,9 @@ 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.append_timer_wheel(60, "SecondWheel");
+    timer.append_timer_wheel(60, "MinuteWheel");
+    timer.append_timer_wheel(12, "HourWheel");
 
     timer.add_timer(30);
     assert_eq!(timer.get_delay_id(), 30);

src/lib.rs

@@ -13,7 +13,7 @@ pub mod buf;
 pub use cache::{LruCache, LruKCache, LfuCache, ArcCache, Slab, Reinit};
 pub use tree::RBTree;
 pub use map::{BitMap, RoaringBitMap};
-pub use timer::{TimerWheel, Timer, TimerRBTree, StampTimer};
+pub use timer::{TimerWheel, Timer, TimerRBTree, StampTimer, StepTimer};
 pub use arr::{CircularBuffer, FixedVec};
 pub use util::*;
 

src/timer/mod.rs

@@ -37,7 +37,9 @@ impl_primitive_timer!(usize);
 mod timer_rbtree;
 mod timer_wheel;
 mod stamp_timer;
+mod step_timer;
 
 pub use timer_wheel::TimerWheel;
 pub use timer_rbtree::TimerRBTree;
 pub use stamp_timer::StampTimer;
+pub use step_timer::StepTimer;

src/timer/stamp_timer.rs

@@ -14,18 +14,18 @@ impl<T> StampTimer<T> {
         Self { val, duration, is_sec }
     }
 
-    pub fn new_second(val: T, interval: u64) -> Self {
+    pub fn new_second(val: T, duration: Duration) -> Self {
         Self {
             val,
-            duration: Duration::from_secs(interval),
+            duration,
             is_sec: true,
         }
     }
 
-    pub fn new_millis(val: T, interval: u64) -> Self {
+    pub fn new_millis(val: T, duration: Duration) -> Self {
         Self {
             val,
-            duration: Duration::from_millis(interval),
+            duration,
             is_sec: false,
         }
     }

src/timer/step_timer.rs

@@ -0,0 +1,18 @@
+use super::Timer;
+
+pub struct StepTimer<T: Timer, V> {
+    pub step: T,
+    pub val: V,
+}
+
+impl<T: Timer, V> StepTimer<T, V> {
+    pub fn new(val: V, step: T) -> Self {
+        Self { val, step }
+    }
+}
+
+impl<T: Timer, V> Timer for StepTimer<T, V> {
+    fn when(&self) -> u64 {
+        self.step.when()
+    }
+}