aboutsummaryrefslogtreecommitdiff
path: root/src/castable_function/threadsafe.rs
diff options
context:
space:
mode:
Diffstat (limited to 'src/castable_function/threadsafe.rs')
-rw-r--r--src/castable_function/threadsafe.rs105
1 files changed, 105 insertions, 0 deletions
diff --git a/src/castable_function/threadsafe.rs b/src/castable_function/threadsafe.rs
new file mode 100644
index 0000000..7543396
--- /dev/null
+++ b/src/castable_function/threadsafe.rs
@@ -0,0 +1,105 @@
+use std::any::{type_name, Any};
+use std::fmt::Debug;
+
+use crate::castable_function::AnyCastableFunction;
+use crate::ptr::TransientPtr;
+
+/// Interface for any threadsafe castable function.
+pub trait AnyThreadsafeCastableFunction:
+ AnyCastableFunction + Send + Sync + Debug
+{
+}
+
+pub struct ThreadsafeCastableFunction<ReturnInterface, DIContainerT>
+where
+ DIContainerT: 'static,
+ ReturnInterface: 'static + ?Sized,
+{
+ func: &'static (dyn Fn(&DIContainerT) -> TransientPtr<ReturnInterface> + Send + Sync),
+}
+
+impl<ReturnInterface, DIContainerT>
+ ThreadsafeCastableFunction<ReturnInterface, DIContainerT>
+where
+ DIContainerT: 'static,
+ ReturnInterface: 'static + ?Sized,
+{
+ pub fn new(
+ func: &'static (dyn Fn(&DIContainerT) -> TransientPtr<ReturnInterface>
+ + Send
+ + Sync),
+ ) -> Self
+ {
+ Self { func }
+ }
+
+ pub fn call(&self, di_container: &DIContainerT) -> TransientPtr<ReturnInterface>
+ {
+ (self.func)(di_container)
+ }
+}
+
+impl<ReturnInterface, DIContainerT> AnyCastableFunction
+ for ThreadsafeCastableFunction<ReturnInterface, DIContainerT>
+where
+ DIContainerT: 'static,
+ ReturnInterface: 'static + ?Sized,
+{
+ fn as_any(&self) -> &dyn Any
+ {
+ self
+ }
+}
+
+impl<ReturnInterface, DIContainerT> AnyThreadsafeCastableFunction
+ for ThreadsafeCastableFunction<ReturnInterface, DIContainerT>
+where
+ DIContainerT: 'static,
+ ReturnInterface: 'static + ?Sized,
+{
+}
+
+impl<ReturnInterface, DIContainerT> Debug
+ for ThreadsafeCastableFunction<ReturnInterface, DIContainerT>
+where
+ DIContainerT: 'static,
+ ReturnInterface: 'static + ?Sized,
+{
+ #[cfg(not(tarpaulin_include))]
+ fn fmt(&self, formatter: &mut std::fmt::Formatter<'_>) -> std::fmt::Result
+ {
+ let ret = type_name::<TransientPtr<ReturnInterface>>();
+
+ formatter.write_fmt(format_args!(
+ "ThreadsafeCastableFunction(&AsyncDIContainer) -> {ret} {{ ... }}",
+ ))
+ }
+}
+
+#[cfg(test)]
+mod tests
+{
+ use super::*;
+ use crate::di_container::asynchronous::MockAsyncDIContainer;
+
+ #[derive(Debug, PartialEq, Eq)]
+ struct Bacon
+ {
+ heal_amount: u32,
+ }
+
+ #[test]
+ fn can_call()
+ {
+ let castable_function =
+ ThreadsafeCastableFunction::new(&|_: &MockAsyncDIContainer| {
+ TransientPtr::new(Bacon { heal_amount: 27 })
+ });
+
+ let mock_di_container = MockAsyncDIContainer::new();
+
+ let output = castable_function.call(&mock_di_container);
+
+ assert_eq!(output, TransientPtr::new(Bacon { heal_amount: 27 }));
+ }
+}