From 494d905bcbc50b536de892ecb3fa285dc6b8a727 Mon Sep 17 00:00:00 2001 From: HampusM Date: Mon, 24 Oct 2022 21:53:14 +0200 Subject: refactor: rename DI container binding map to DI container storage --- src/di_container/binding_storage.rs | 293 ++++++++++++++++++++++++++++++++++++ 1 file changed, 293 insertions(+) create mode 100644 src/di_container/binding_storage.rs (limited to 'src/di_container/binding_storage.rs') diff --git a/src/di_container/binding_storage.rs b/src/di_container/binding_storage.rs new file mode 100644 index 0000000..2bc208f --- /dev/null +++ b/src/di_container/binding_storage.rs @@ -0,0 +1,293 @@ +use std::any::TypeId; + +use ahash::AHashMap; + +pub struct DIContainerBindingStorage +where + Provider: 'static + ?Sized, +{ + inner: AHashMap>, +} + +impl DIContainerBindingStorage +where + Provider: 'static + ?Sized, +{ + pub fn new() -> Self + { + Self { + inner: AHashMap::new(), + } + } + + #[allow(clippy::borrowed_box)] + pub fn get(&self, name: Option<&'static str>) -> Option<&Box> + where + Interface: 'static + ?Sized, + { + let interface_typeid = TypeId::of::(); + + self.inner.get(&BindingIdentification { + type_id: interface_typeid, + name, + }) + } + + pub fn set(&mut self, name: Option<&'static str>, provider: Box) + where + Interface: 'static + ?Sized, + { + let interface_typeid = TypeId::of::(); + + self.inner.insert( + BindingIdentification { + type_id: interface_typeid, + name, + }, + provider, + ); + } + + pub fn remove( + &mut self, + name: Option<&'static str>, + ) -> Option> + where + Interface: 'static + ?Sized, + { + let interface_typeid = TypeId::of::(); + + self.inner.remove(&BindingIdentification { + type_id: interface_typeid, + name, + }) + } + + pub fn has(&self, name: Option<&'static str>) -> bool + where + Interface: 'static + ?Sized, + { + let interface_typeid = TypeId::of::(); + + self.inner.contains_key(&BindingIdentification { + type_id: interface_typeid, + name, + }) + } +} + +#[derive(Debug, PartialEq, Eq, Hash)] +struct BindingIdentification +{ + type_id: TypeId, + name: Option<&'static str>, +} + +#[cfg(test)] +mod tests +{ + use super::*; + + mod subjects + { + pub trait SomeProvider + { + fn get_id(&self) -> u8; + } + + pub struct SomeProviderImpl + { + pub id: u8, + } + + impl SomeProvider for SomeProviderImpl + { + fn get_id(&self) -> u8 + { + self.id + } + } + } + + #[test] + fn can_get() + { + type Interface = (); + + let mut binding_map = + DIContainerBindingStorage::::new(); + + binding_map.inner.insert( + BindingIdentification { + type_id: TypeId::of::(), + name: None, + }, + Box::new(subjects::SomeProviderImpl { id: 20 }), + ); + + assert!(binding_map + .get::(None) + .map_or_else(|| false, |provider| provider.get_id() == 20)); + } + + #[test] + fn can_get_with_name() + { + type Interface = (); + + let mut binding_map = + DIContainerBindingStorage::::new(); + + binding_map.inner.insert( + BindingIdentification { + type_id: TypeId::of::(), + name: Some("hello"), + }, + Box::new(subjects::SomeProviderImpl { id: 11 }), + ); + + assert!(binding_map + .get::(Some("hello")) + .map_or_else(|| false, |provider| provider.get_id() == 11)); + + assert!(binding_map.get::(None).is_none()); + } + + #[test] + fn can_set() + { + type Interface = (); + + let mut binding_map = + DIContainerBindingStorage::::new(); + + binding_map + .set::(None, Box::new(subjects::SomeProviderImpl { id: 65 })); + + let expected_key = &BindingIdentification { + type_id: TypeId::of::(), + name: None, + }; + + assert!(binding_map.inner.contains_key(expected_key)); + + assert_eq!(binding_map.inner[expected_key].get_id(), 65); + } + + #[test] + fn can_set_with_name() + { + type Interface = (); + + let mut binding_map = + DIContainerBindingStorage::::new(); + + binding_map.set::( + Some("special"), + Box::new(subjects::SomeProviderImpl { id: 3 }), + ); + + let expected_key = &BindingIdentification { + type_id: TypeId::of::(), + name: Some("special"), + }; + + assert!(binding_map.inner.contains_key(expected_key)); + + assert_eq!(binding_map.inner[expected_key].get_id(), 3); + } + + #[test] + fn can_remove() + { + type Interface = (); + + let mut binding_map = + DIContainerBindingStorage::::new(); + + binding_map.inner.insert( + BindingIdentification { + type_id: TypeId::of::(), + name: None, + }, + Box::new(subjects::SomeProviderImpl { id: 103 }), + ); + + binding_map.remove::(None); + + let expected_key = &BindingIdentification { + type_id: TypeId::of::(), + name: None, + }; + + assert!(!binding_map.inner.contains_key(expected_key)); + } + + #[test] + fn can_remove_with_name() + { + type Interface = (); + + let mut binding_map = + DIContainerBindingStorage::::new(); + + binding_map.inner.insert( + BindingIdentification { + type_id: TypeId::of::(), + name: Some("cool"), + }, + Box::new(subjects::SomeProviderImpl { id: 42 }), + ); + + binding_map.remove::(Some("cool")); + + let expected_key = &BindingIdentification { + type_id: TypeId::of::(), + name: Some("cool"), + }; + + assert!(!binding_map.inner.contains_key(expected_key)); + } + + #[test] + fn can_get_has() + { + type Interface = (); + + let mut binding_map = + DIContainerBindingStorage::::new(); + + assert!(!binding_map.has::(None)); + + binding_map.inner.insert( + BindingIdentification { + type_id: TypeId::of::(), + name: None, + }, + Box::new(subjects::SomeProviderImpl { id: 103 }), + ); + + assert!(binding_map.has::(None)); + } + + #[test] + fn can_get_has_with_name() + { + type Interface = (); + + let mut binding_map = + DIContainerBindingStorage::::new(); + + assert!(!binding_map.has::(Some("awesome"))); + + binding_map.inner.insert( + BindingIdentification { + type_id: TypeId::of::(), + name: Some("awesome"), + }, + Box::new(subjects::SomeProviderImpl { id: 101 }), + ); + + assert!(binding_map.has::(Some("awesome"))); + } +} -- cgit v1.2.3-18-g5258