From 1c46b68581213ca8ae6200daa32f626b5389b4b0 Mon Sep 17 00:00:00 2001 From: HampusM Date: Thu, 25 Aug 2022 20:21:49 +0200 Subject: refactor!: make DI container have single get function BREAKING CHANGE: The DI container get_singleton & get_factory functions have been replaced by the get function now returning a enum --- src/di_container.rs | 163 +++++++++++++++++----------------------------------- 1 file changed, 52 insertions(+), 111 deletions(-) (limited to 'src/di_container.rs') diff --git a/src/di_container.rs b/src/di_container.rs index cc2a930..1be570b 100644 --- a/src/di_container.rs +++ b/src/di_container.rs @@ -59,7 +59,7 @@ use crate::errors::di_container::{ use crate::interfaces::injectable::Injectable; use crate::libs::intertrait::cast::{CastBox, CastRc}; use crate::provider::{Providable, SingletonProvider, TransientTypeProvider}; -use crate::ptr::{SingletonPtr, TransientPtr}; +use crate::ptr::{SingletonPtr, SomePtr}; /// Scope configurator for a binding for type 'Interface' inside a [`DIContainer`]. pub struct BindingScopeConfigurator<'di_container, Interface, Implementation> @@ -177,7 +177,7 @@ where #[cfg(feature = "factory")] pub fn to_factory( &mut self, - factory_func: &'static dyn Fn>, + factory_func: &'static dyn Fn>, ) -> Result<(), BindingBuilderError> where Args: 'static, @@ -212,7 +212,7 @@ where #[cfg(feature = "factory")] pub fn to_default_factory( &mut self, - factory_func: &'static dyn Fn<(), Output = TransientPtr>, + factory_func: &'static dyn Fn<(), Output = crate::ptr::TransientPtr>, ) -> Result<(), BindingBuilderError> where Return: 'static + ?Sized, @@ -260,127 +260,67 @@ impl DIContainer BindingBuilder::::new(self) } - /// Returns a new instance of the type bound with `Interface`. + /// Returns the type bound with `Interface`. /// /// # Errors /// Will return `Err` if: /// - No binding for `Interface` exists /// - Resolving the binding for `Interface` fails /// - Casting the binding for `Interface` fails - /// - The binding for `Interface` is not transient - pub fn get(&self) -> Result, DIContainerError> + pub fn get(&self) -> Result, DIContainerError> where Interface: 'static + ?Sized, { - self.get_with_history::(Vec::new()) - } - - /// Returns the singleton instance bound with `Interface`. - /// - /// # Errors - /// Will return `Err` if: - /// - No binding for `Interface` exists - /// - Resolving the binding for `Interface` fails - /// - Casting the binding for `Interface` fails - /// - The binding for `Interface` is not a singleton - pub fn get_singleton( - &self, - ) -> Result, DIContainerError> - where - Interface: 'static + ?Sized, - { - self.get_singleton_with_history(Vec::new()) - } - - /// Returns the factory bound with factory type `Interface`. - /// - /// *This function is only available if Syrette is built with the "factory" feature.* - /// - /// # Errors - /// Will return `Err` if: - /// - No binding for `Interface` exists - /// - Resolving the binding for `Interface` fails - /// - Casting the binding for `Interface` fails - /// - The binding for `Interface` is not a factory - #[cfg(feature = "factory")] - pub fn get_factory( - &self, - ) -> Result, DIContainerError> - where - Interface: 'static + ?Sized, - { - let binding_providable = self.get_binding_providable::(Vec::new())?; - - if let Providable::Factory(binding_factory) = binding_providable { - return binding_factory - .cast::() - .map_err(|_| DIContainerError::CastFailed(type_name::())); - } - - Err(DIContainerError::WrongBindingType { - interface: type_name::(), - expected: "factory", - found: binding_providable.to_string().to_lowercase(), - }) - } - - #[doc(hidden)] - pub fn get_with_history( - &self, - dependency_history: Vec<&'static str>, - ) -> Result, DIContainerError> - where - Interface: 'static + ?Sized, - { - let binding_providable = - self.get_binding_providable::(dependency_history)?; - - if let Providable::Transient(binding_transient) = binding_providable { - return binding_transient - .cast::() - .map_err(|_| DIContainerError::CastFailed(type_name::())); - } - - #[cfg(feature = "factory")] - if let Providable::Factory(binding_factory) = binding_providable { - use crate::interfaces::factory::IFactory; - - let factory = binding_factory - .cast::>() - .map_err(|_| DIContainerError::CastFailed(type_name::()))?; - - return Ok(factory()); - } - - Err(DIContainerError::WrongBindingType { - interface: type_name::(), - expected: "transient", - found: binding_providable.to_string().to_lowercase(), - }) + self.get_bound::(Vec::new()) } #[doc(hidden)] - pub fn get_singleton_with_history( + pub fn get_bound( &self, dependency_history: Vec<&'static str>, - ) -> Result, DIContainerError> + ) -> Result, DIContainerError> where Interface: 'static + ?Sized, { let binding_providable = self.get_binding_providable::(dependency_history)?; - if let Providable::Singleton(binding_singleton) = binding_providable { - return binding_singleton - .cast::() - .map_err(|_| DIContainerError::CastFailed(type_name::())); + match binding_providable { + Providable::Transient(transient_binding) => Ok(SomePtr::Transient( + transient_binding.cast::().map_err(|_| { + DIContainerError::CastFailed(type_name::()) + })?, + )), + Providable::Singleton(singleton_binding) => Ok(SomePtr::Singleton( + singleton_binding.cast::().map_err(|_| { + DIContainerError::CastFailed(type_name::()) + })?, + )), + #[cfg(feature = "factory")] + Providable::Factory(factory_binding) => { + match factory_binding.clone().cast::() { + Ok(factory) => Ok(SomePtr::Factory(factory)), + Err(_err) => { + use crate::interfaces::factory::IFactory; + + let default_factory = factory_binding + .cast::>() + .map_err(|_| { + DIContainerError::CastFailed(type_name::()) + })?; + + Ok(SomePtr::Transient(default_factory())) + } + } + } + #[cfg(not(feature = "factory"))] + Providable::Factory(_) => { + return Err(DIContainerError::CantHandleFactoryBinding(type_name::< + Interface, + >( + ))); + } } - - Err(DIContainerError::WrongBindingType { - interface: type_name::(), - expected: "singleton", - found: binding_providable.to_string().to_lowercase(), - }) } fn get_binding_providable( @@ -545,7 +485,7 @@ mod tests } #[test] - fn can_bind_to() -> Result<(), BindingBuilderError> + fn can_bind_to() -> Result<(), Box> { let mut di_container: DIContainer = DIContainer::new(); @@ -579,7 +519,7 @@ mod tests #[test] #[cfg(feature = "factory")] - fn can_bind_to_factory() -> Result<(), BindingBuilderError> + fn can_bind_to_factory() -> Result<(), Box> { type IUserManagerFactory = dyn crate::interfaces::factory::IFactory<(), dyn subjects::IUserManager>; @@ -601,7 +541,7 @@ mod tests } #[test] - fn can_get() -> Result<(), DIContainerError> + fn can_get() -> Result<(), Box> { mock! { Provider {} @@ -636,7 +576,7 @@ mod tests } #[test] - fn can_get_singleton() -> Result<(), DIContainerError> + fn can_get_singleton() -> Result<(), Box> { mock! { Provider {} @@ -667,11 +607,12 @@ mod tests .bindings .set::(Box::new(mock_provider)); - let first_number_rc = di_container.get_singleton::()?; + let first_number_rc = di_container.get::()?.singleton()?; assert_eq!(first_number_rc.get(), 2820); - let second_number_rc = di_container.get_singleton::()?; + let second_number_rc = + di_container.get::()?.singleton()?; assert_eq!(first_number_rc.as_ref(), second_number_rc.as_ref()); @@ -680,7 +621,7 @@ mod tests #[test] #[cfg(feature = "factory")] - fn can_get_factory() -> Result<(), DIContainerError> + fn can_get_factory() -> Result<(), Box> { trait IUserManager { @@ -756,7 +697,7 @@ mod tests .bindings .set::(Box::new(mock_provider)); - di_container.get_factory::()?; + di_container.get::()?; Ok(()) } -- cgit v1.2.3-18-g5258