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
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
use std::prelude::v1::*;
use std::convert::TryInto;
use std::fmt::Debug;
use std::ops::Add;
use std::sync::atomic::AtomicU64;
use std::sync::atomic::Ordering;
use std::sync::Arc;
use std::time::Duration;
use crate::nanos::Nanos;
pub trait Reference:
Sized + Add<Nanos, Output = Self> + PartialEq + Eq + Ord + Copy + Clone + Send + Sync + Debug
{
fn duration_since(&self, earlier: Self) -> Nanos;
fn saturating_sub(&self, duration: Nanos) -> Self;
}
pub trait Clock: Clone {
type Instant: Reference;
fn now(&self) -> Self::Instant;
}
impl Reference for Duration {
fn duration_since(&self, earlier: Self) -> Nanos {
self.checked_sub(earlier)
.unwrap_or_else(|| Duration::new(0, 0))
.into()
}
fn saturating_sub(&self, duration: Nanos) -> Self {
self.checked_sub(duration.into()).unwrap_or(*self)
}
}
impl Add<Nanos> for Duration {
type Output = Self;
fn add(self, other: Nanos) -> Self {
let other: Duration = other.into();
self + other
}
}
#[derive(Debug, Clone, Default)]
pub struct FakeRelativeClock {
now: Arc<AtomicU64>,
}
impl FakeRelativeClock {
pub fn advance(&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 = Nanos;
fn now(&self) -> Self::Instant {
self.now.load(Ordering::Relaxed).into()
}
}
#[cfg(feature = "std")]
mod with_std;
#[cfg(feature = "std")]
pub use with_std::*;
#[cfg(all(feature = "std", feature = "quanta"))]
mod quanta;
#[cfg(all(feature = "std", feature = "quanta"))]
pub use self::quanta::*;
mod default;
pub use default::*;