Skip to main content

ach_util/
op.rs

1#[derive(Debug, Clone, Copy, PartialEq, Eq)]
2pub enum Op {
3    None = 0,
4    Write,
5    Take,
6    Replace,
7    Update,
8    Remove,
9}
10impl From<u8> for Op {
11    fn from(s: u8) -> Self {
12        match s {
13            s if s == Op::None as u8 => Op::None,
14            s if s == Op::Write as u8 => Op::Write,
15            s if s == Op::Take as u8 => Op::Take,
16            s if s == Op::Replace as u8 => Op::Replace,
17            s if s == Op::Remove as u8 => Op::Remove,
18            _ => Op::None,
19        }
20    }
21}
22impl From<Op> for u8 {
23    fn from(s: Op) -> Self {
24        s as u8
25    }
26}
27
28pub type AtomicMemoryOp = atomic::Atomic<MemoryOp>;
29
30#[derive(Debug, Clone, Copy, PartialEq, Eq, PartialOrd, Ord)]
31pub struct MemoryOp(u32);
32impl Default for MemoryOp {
33    fn default() -> Self {
34        Self::new()
35    }
36}
37impl MemoryOp {
38    pub const fn new() -> MemoryOp {
39        Self(0)
40    }
41    pub fn op(&self) -> Op {
42        Op::from((self.0 & 0xFF) as u8)
43    }
44    pub fn cur_version(&self) -> u16 {
45        ((self.0 & 0x000FFF00) >> 8) as u16
46    }
47    pub fn next_version(&mut self) -> u16 {
48        ((self.0 & 0xFFF00000) >> 20) as u16
49    }
50    pub fn finish(&mut self) {
51        self.0 = (self.0 & 0xFFF00000) | ((self.next_version() as u32) << 8)
52    }
53    pub fn is_finished(&mut self) -> bool {
54        self.cur_version() == self.next_version()
55    }
56    pub fn set_op(&mut self, op: Op) -> u16 {
57        self.0 = (self.0.wrapping_add(0x00100000) & 0xFFFFFF00) | (op as u32);
58        self.next_version()
59    }
60}
61impl From<u32> for MemoryOp {
62    fn from(s: u32) -> Self {
63        Self(s)
64    }
65}
66impl From<MemoryOp> for u32 {
67    fn from(s: MemoryOp) -> Self {
68        s.0
69    }
70}