dfir_rs/scheduled/handoff/
vector.rs

Help
1use std::any::Any;
2use std::cell::{RefCell, RefMut};
3use std::rc::Rc;
4
5use super::{CanReceive, Handoff, HandoffMeta, Iter};
6
7/// A [Vec]-based FIFO handoff.
8pub struct VecHandoff<T>
9where
10    T: 'static,
11{
12    pub(crate) input: Rc<RefCell<Vec<T>>>,
13    pub(crate) output: Rc<RefCell<Vec<T>>>,
14}
15impl<T> Default for VecHandoff<T>
16where
17    T: 'static,
18{
19    fn default() -> Self {
20        Self {
21            input: Default::default(),
22            output: Default::default(),
23        }
24    }
25}
26impl<T> Handoff for VecHandoff<T> {
27    type Inner = Vec<T>;
28
29    fn take_inner(&self) -> Self::Inner {
30        self.input.take()
31    }
32
33    fn borrow_mut_swap(&self) -> RefMut<Self::Inner> {
34        let mut input = self.input.borrow_mut();
35        let mut output = self.output.borrow_mut();
36
37        std::mem::swap(&mut *input, &mut *output);
38
39        output
40    }
41}
42
43impl<T> CanReceive<Option<T>> for VecHandoff<T> {
44    fn give(&self, mut item: Option<T>) -> Option<T> {
45        if let Some(item) = item.take() {
46            (*self.input).borrow_mut().push(item)
47        }
48        None
49    }
50}
51impl<T, I> CanReceive<Iter<I>> for VecHandoff<T>
52where
53    I: Iterator<Item = T>,
54{
55    fn give(&self, mut iter: Iter<I>) -> Iter<I> {
56        (*self.input).borrow_mut().extend(&mut iter.0);
57        iter
58    }
59}
60impl<T> CanReceive<Vec<T>> for VecHandoff<T> {
61    fn give(&self, mut vec: Vec<T>) -> Vec<T> {
62        (*self.input).borrow_mut().extend(vec.drain(..));
63        vec
64    }
65}
66
67impl<T> HandoffMeta for VecHandoff<T> {
68    fn any_ref(&self) -> &dyn Any {
69        self
70    }
71
72    fn is_bottom(&self) -> bool {
73        (*self.input).borrow_mut().is_empty()
74    }
75}
dfir_rs/scheduled/handoff/vector.rs

dfir_rs/scheduled/handoff/
vector.rs
