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
use crate::rt;
use std::sync::atomic::AtomicBool;
use std::sync::atomic::Ordering::SeqCst;
#[derive(Debug)]
pub struct Notify {
object: rt::Notify,
waiting: AtomicBool,
}
impl Notify {
pub fn new() -> Notify {
Notify {
object: rt::Notify::new(false, true),
waiting: AtomicBool::new(false),
}
}
pub fn notify(&self) {
self.object.notify();
}
pub fn wait(&self) {
let actual = self.waiting.compare_and_swap(false, true, SeqCst);
assert!(!actual, "only a single thread may wait on `Notify`");
self.object.wait();
self.waiting.store(false, SeqCst);
}
}