dfir_rs/scheduled/handoff/
vector.rs

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

dfir_rs/scheduled/handoff/
vector.rs
