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
use std::thread::sleep;
use std::time::{Duration, Instant};
use failure::{Error, Fail, Fallible};
#[derive(Debug, Fail)]
#[fail(display = "The event waited for never came")]
pub struct Timeout;
#[derive(Debug)]
pub struct Wait {
timeout: Duration,
sleep: Duration,
}
impl Default for Wait {
fn default() -> Self {
Self {
timeout: Duration::from_secs(10),
sleep: Duration::from_millis(100),
}
}
}
impl Wait {
pub fn new(timeout: Duration, sleep: Duration) -> Self {
Self { timeout, sleep }
}
pub fn with_timeout(timeout: Duration) -> Self {
Self {
timeout,
..Self::default()
}
}
pub fn with_sleep(sleep: Duration) -> Self {
Self {
sleep,
..Self::default()
}
}
pub fn forever() -> Self {
Self {
timeout: Duration::from_secs(u64::max_value()),
..Self::default()
}
}
pub fn until<F, G>(&self, predicate: F) -> Result<G, Timeout>
where
F: FnMut() -> Option<G>,
{
let mut predicate = predicate;
let start = Instant::now();
loop {
if let Some(v) = predicate() {
return Ok(v);
}
if start.elapsed() > self.timeout {
return Err(Timeout);
}
sleep(self.sleep);
}
}
pub fn strict_until<F, D, E, G>(&self, predicate: F, downcast: D) -> Fallible<G>
where
F: FnMut() -> Fallible<G>,
D: FnMut(Error) -> Fallible<E>,
E: Fail,
{
let mut predicate = predicate;
let mut downcast = downcast;
let start = Instant::now();
loop {
match predicate() {
Ok(value) => return Ok(value),
Err(error) => downcast(error)?,
};
if start.elapsed() > self.timeout {
return Err(Timeout.into());
}
sleep(self.sleep);
}
}
}