muzzman_lib/
storage.rs

1#![allow(dead_code)]
2
3use std::any::Any;
4
5#[derive(Debug, Default)]
6pub struct Storage {
7    pub data: Vec<Box<dyn Any + Send>>,
8}
9
10impl Storage {
11    pub fn get<T: 'static>(&self) -> Option<&T> {
12        for element in self.data.iter() {
13            if let Some(data) = element.downcast_ref::<T>() {
14                return Some(data);
15            }
16        }
17
18        None
19    }
20
21    pub fn get_mut<T: 'static>(&mut self) -> Option<&mut T> {
22        for element in self.data.iter_mut() {
23            if let Some(data) = element.downcast_mut::<T>() {
24                return Some(data);
25            }
26        }
27
28        None
29    }
30
31    pub fn set<T: 'static + Send>(&mut self, data: T) -> Option<Box<T>> {
32        let res = self.remove();
33        self.data.push(Box::new(data));
34        res
35    }
36
37    pub fn remove<T: 'static>(&mut self) -> Option<Box<T>> {
38        let mut finded = None;
39        for (i, element) in self.data.iter().enumerate() {
40            if element.downcast_ref::<T>().is_some() {
41                finded = Some(i);
42                break;
43            }
44        }
45
46        if let Some(finded) = finded {
47            Some(self.data.remove(finded).downcast::<T>().unwrap())
48        } else {
49            None
50        }
51    }
52}
53
54#[cfg(test)]
55mod test {
56    use super::*;
57
58    struct Data {
59        id: usize,
60    }
61
62    #[test]
63    fn storage() {
64        let mut storage = Storage::default();
65
66        let res = storage.set(Data { id: 21 });
67
68        assert!(res.is_none());
69
70        assert_eq!(storage.get::<Data>().unwrap().id, 21);
71
72        let res = storage.set(Data { id: 92 });
73
74        assert!(res.is_some());
75
76        assert_eq!(res.unwrap().id, 21);
77
78        assert_eq!(storage.get::<Data>().unwrap().id, 92);
79    }
80}
muzzman_lib/storage.rs

muzzman_lib/
storage.rs
