splitrc 0.1.12

Thread-Safe Split Reference Counts
Documentation 
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

use std::fmt;
use std::sync::atomic::Ordering;

#[cfg(loom)]
use loom::sync::atomic::AtomicBool;

#[cfg(not(loom))]
use std::sync::atomic::AtomicBool;

#[derive(Debug)]
pub struct Unit;
impl splitrc::Notify for Unit {}

impl fmt::Display for Unit {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        fmt::Display::fmt("Unit", f)
    }
}

#[derive(Default)]
pub struct TrackNotify {
    pub tx_did_drop: AtomicBool,
    pub rx_did_drop: AtomicBool,
}

impl splitrc::Notify for TrackNotify {
    fn last_tx_did_drop(&self) {
        self.tx_did_drop.store(true, Ordering::Release);
    }
    fn last_rx_did_drop(&self) {
        self.rx_did_drop.store(true, Ordering::Release);
    }
}

#[allow(dead_code)]
impl TrackNotify {
    // ASAN, TSAN, and Loom tests must read the stored values.
    pub fn access(&self) -> (bool, bool) {
        (
            self.tx_did_drop.load(Ordering::Acquire),
            self.rx_did_drop.load(Ordering::Acquire),
        )
    }
}