src/di_container_binding_map.rs


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

use std::any::{type_name, TypeId};

use ahash::AHashMap;
use error_stack::report;

use crate::{errors::di_container::DIContainerError, provider::IProvider};

pub struct DIContainerBindingMap
{
    bindings: AHashMap<TypeId, Box<dyn IProvider>>,
}

impl DIContainerBindingMap
{
    pub fn new() -> Self
    {
        Self {
            bindings: AHashMap::new(),
        }
    }

    pub fn get<Interface>(&self) -> error_stack::Result<&dyn IProvider, DIContainerError>
    where
        Interface: 'static + ?Sized,
    {
        let interface_typeid = TypeId::of::<Interface>();

        Ok(self
            .bindings
            .get(&interface_typeid)
            .ok_or_else(|| {
                report!(DIContainerError).attach_printable(format!(
                    "No binding exists for {}",
                    type_name::<Interface>()
                ))
            })?
            .as_ref())
    }

    pub fn set<Interface>(&mut self, provider: Box<dyn IProvider>)
    where
        Interface: 'static + ?Sized,
    {
        let interface_typeid = TypeId::of::<Interface>();

        self.bindings.insert(interface_typeid, provider);
    }

    /// Only used by tests in the ``di_container`` module.
    #[cfg(test)]
    pub fn count(&self) -> usize
    {
        self.bindings.len()
    }
}