prns-core 0.3.4

Pure Reticulum engine and wire contract for Personal Reticulum
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

use crate::interfaces::InterfaceId;

pub trait DirtyInterfaceSet {
    fn mark(&mut self, interface: InterfaceId);
    fn drain(&mut self, visit: impl FnMut(InterfaceId));
}

#[cfg(feature = "alloc")]
impl DirtyInterfaceSet for alloc::collections::BTreeSet<InterfaceId> {
    fn mark(&mut self, interface: InterfaceId) {
        self.insert(interface);
    }

    fn drain(&mut self, mut visit: impl FnMut(InterfaceId)) {
        for interface in core::mem::take(self) {
            visit(interface);
        }
    }
}

impl<const N: usize> DirtyInterfaceSet for heapless::Vec<InterfaceId, N> {
    fn mark(&mut self, interface: InterfaceId) {
        if !self.contains(&interface) {
            let pushed = self.push(interface);
            debug_assert!(
                pushed.is_ok(),
                "dirty interface set capacity is smaller than the live interface count"
            );
        }
    }

    fn drain(&mut self, mut visit: impl FnMut(InterfaceId)) {
        for interface in core::mem::take(self) {
            visit(interface);
        }
    }
}