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
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
use super::{Clock, Reference};
use parking_lot::Mutex;
use std::sync::Arc;
use std::time::{Duration, Instant, SystemTime};
pub type DefaultClock = MonotonicClock;
#[derive(Debug, Clone)]
pub struct FakeAbsoluteClock {
now: Arc<Mutex<Instant>>,
}
impl Default for FakeAbsoluteClock {
fn default() -> Self {
FakeAbsoluteClock {
now: Arc::new(Mutex::new(Instant::now())),
}
}
}
impl FakeAbsoluteClock {
pub fn advance(&mut self, by: Duration) {
*(self.now.lock()) += by
}
}
impl Clock for FakeAbsoluteClock {
type Instant = Instant;
fn now(&self) -> Self::Instant {
*self.now.lock()
}
}
#[derive(Clone, Debug, Default)]
pub struct MonotonicClock();
impl Reference for Instant {
fn duration_since(&self, earlier: Self) -> Duration {
if earlier < *self {
*self - earlier
} else {
Duration::new(0, 0)
}
}
fn saturating_sub(&self, duration: Duration) -> Self {
self.checked_sub(duration).unwrap_or(*self)
}
}
impl Clock for MonotonicClock {
type Instant = Instant;
fn now(&self) -> Self::Instant {
Instant::now()
}
}
#[derive(Clone, Debug, Default)]
pub struct SystemClock();
impl Reference for SystemTime {
fn duration_since(&self, earlier: Self) -> Duration {
self.duration_since(earlier)
.unwrap_or_else(|_| Duration::new(0, 0))
}
fn saturating_sub(&self, duration: Duration) -> Self {
self.checked_sub(duration).unwrap_or(*self)
}
}
impl Clock for SystemClock {
type Instant = SystemTime;
fn now(&self) -> Self::Instant {
SystemTime::now()
}
}