bakbon 0.1.2

BakBon is an infrastructure microkernel library in Rust that provides generic building blocks (Router, Registry, Balancer, Queue, Gateway, Cache, Middleware, Service/Processor) for message‑driven distributed systems.
Documentation 
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
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111

mod builder;
mod eviction;

pub use eviction::Eviction;
use {
    crate::Envelope,
    builder::Builder,
    std::collections::HashMap,
};

type Store = HashMap<String, Envelope>;

#[derive(Default)]
pub struct Cache {
    store:    Store,
    eviction: Eviction,
}

impl Cache {
    pub fn builder() -> Builder { Builder::default() }

    pub fn get(&self, k: &str) -> Option<&Envelope> { self.store.get(k) }

    pub fn set(&mut self, k: &str, v: Envelope) {
        self.store
            .insert(k.to_string(), v);
    }

    pub fn clear(&mut self) { self.store.clear(); }

    pub fn eviction(&self) -> Eviction { self.eviction }
}

//  +------------+
//  | UNIT TESTS |
//  +------------+

#[cfg(test)]
mod tests {
    use {
        super::*,
        crate::{
            Address,
            Payload,
            Result,
        },
    };
    const SRC: &str = "http://source.com";
    const DST: &str = "http://destination.com";

    #[test]
    fn default_cache() {
        let cache = Cache::default();
        assert!(cache.store.is_empty());
    }

    #[test]
    fn cache_store() -> Result<()> {
        let src = Address::parse(SRC)?;
        let dst = Address::parse(DST)?;
        let payload = Payload::default();
        let msg = Envelope::new(src.clone(), dst.clone(), payload.clone());

        let k = "default";
        let mut cache = Cache::default();
        cache.set(k, msg);

        let msg = cache.get(k);
        assert!(msg.is_some());

        let msg = msg.unwrap();
        assert_eq!(msg.source(), &src);
        assert_eq!(msg.destination(), &dst);
        assert_eq!(msg.payload(), &payload);
        Ok(())
    }

    #[test]
    fn get_from_empty_cache() {
        let k = "phantom";
        let cache = Cache::default();
        let msg = cache.get(k);
        assert!(msg.is_none());
    }

    #[test]
    fn clear_cache() -> Result<()> {
        let src = Address::parse(SRC)?;
        let dst = Address::parse(DST)?;
        let payload = Payload::default();

        let msg1 = Envelope::new(src.clone(), dst.clone(), payload.clone());
        let msg2 = Envelope::new(src.clone(), dst.clone(), payload.clone());
        let msg3 = Envelope::new(src.clone(), dst.clone(), payload.clone());
        let msg4 = Envelope::new(src.clone(), dst.clone(), payload.clone());
        let msg5 = Envelope::new(src.clone(), dst.clone(), payload.clone());

        let mut cache = Cache::default();
        cache.set("msg1", msg1);
        cache.set("msg2", msg2);
        cache.set("msg3", msg3);
        cache.set("msg4", msg4);
        cache.set("msg5", msg5);

        assert_eq!(cache.store.len(), 5);

        cache.clear();
        assert!(cache.store.is_empty());
        Ok(())
    }
}