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
94
95
96
97
98
99
use crate::lib::*;
pub trait Reference:
Sized + Add<Duration, Output = Self> + PartialEq + Eq + Ord + Copy + Clone + Send + Sync + Debug
{
fn duration_since(&self, earlier: Self) -> Duration;
fn saturating_sub(&self, duration: Duration) -> Self;
}
pub trait Clock: Clone {
type Instant: Reference;
fn now(&self) -> Self::Instant;
}
impl Reference for Duration {
fn duration_since(&self, earlier: Self) -> Duration {
self.checked_sub(earlier)
.unwrap_or_else(|| Duration::new(0, 0))
}
fn saturating_sub(&self, duration: Duration) -> Self {
self.checked_sub(duration).unwrap_or(*self)
}
}
#[derive(Debug, Clone, Default)]
pub struct FakeRelativeClock {
now: Arc<AtomicU64>,
}
impl FakeRelativeClock {
pub fn advance(&mut self, by: Duration) {
let by: u64 = by
.as_nanos()
.try_into()
.expect("Can not represent times past ~584 years");
let mut prev = self.now.load(Ordering::Acquire);
let mut next = prev + by;
while let Err(next_prev) =
self.now
.compare_exchange_weak(prev, next, Ordering::Release, Ordering::Relaxed)
{
prev = next_prev;
next = prev + by;
}
}
}
impl PartialEq for FakeRelativeClock {
fn eq(&self, other: &Self) -> bool {
self.now.load(Ordering::Relaxed) == other.now.load(Ordering::Relaxed)
}
}
impl Clock for FakeRelativeClock {
type Instant = Duration;
fn now(&self) -> Self::Instant {
Duration::from_nanos(self.now.load(Ordering::Relaxed))
}
}
#[cfg(not(feature = "std"))]
mod no_std;
#[cfg(not(feature = "std"))]
pub use no_std::*;
#[cfg(feature = "std")]
mod with_std;
#[cfg(feature = "std")]
pub use with_std::*;