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
use crate::loom::sync::atomic::AtomicUsize;
use crate::util::slab::Address;
use std::fmt;
use std::sync::atomic::Ordering;
use std::usize;
pub(super) struct TransferStack {
head: AtomicUsize,
}
impl TransferStack {
pub(super) fn new() -> Self {
Self {
head: AtomicUsize::new(Address::NULL),
}
}
pub(super) fn pop_all(&self) -> Option<usize> {
let val = self.head.swap(Address::NULL, Ordering::Acquire);
if val == Address::NULL {
None
} else {
Some(val)
}
}
pub(super) fn push(&self, value: usize, before: impl Fn(usize)) {
let mut next = self.head.load(Ordering::Relaxed);
loop {
before(next);
match self
.head
.compare_exchange(next, value, Ordering::AcqRel, Ordering::Acquire)
{
Err(actual) => next = actual,
Ok(_) => return,
}
}
}
}
impl fmt::Debug for TransferStack {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
f.debug_struct("TransferStack")
.field(
"head",
&format_args!("{:#x}", self.head.load(Ordering::Relaxed)),
)
.finish()
}
}