1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40
use std::time::{Instant, Duration}; pub trait Interval : Default { fn now(&self) -> Instant { Instant::now() } fn elapsed(&self, instant: Instant) -> Duration { instant.elapsed() } } #[derive(Default)] pub struct RealInterval; impl Interval for RealInterval { } #[derive(Default)] #[cfg(test)] pub struct FixedIntervalSpawner { step_millis: u64, } #[cfg(test)] impl FixedIntervalSpawner { pub fn new(step_millis: u64) -> Self { FixedIntervalSpawner { step_millis : step_millis } } } #[cfg(test)] impl Interval for FixedIntervalSpawner { fn now(&self) -> Instant { Instant::now() } fn elapsed(&self, instant: Instant) -> Duration { (instant - Duration::from_millis(self.step_millis)).elapsed() } }