icee_container_rs/global_container.rs
1use std::any::Any;
2use std::collections::HashMap;
3
4use async_std::sync::{Arc, RwLock};
5use lazy_static::lazy_static;
6
7use crate::Error;
8
9lazy_static! {
10 static ref CONTAINER: RwLock<HashMap<&'static str, Arc<dyn Any + Send + Sync>>> =
11 RwLock::new(HashMap::new());
12}
13
14/// Checks whether or not is given service registered
15///
16/// ```
17/// # async_std::task::block_on(async {
18/// use icee_container_rs::has;
19///
20/// let exists: bool = has("service_name").await;
21/// # assert_eq!(exists, false);
22/// # })
23/// ```
24pub async fn has(name: &str) -> bool {
25 unsafe { (&CONTAINER).read().await.contains_key(name) }
26}
27
28/// Retrieves named service
29///
30/// To satisfy compiler it's also required to specify exact service type under which
31/// it has been stored wrapped within [async_std::sync::Arc]
32///
33/// ```no_run
34/// use icee_container_rs::get;
35/// use async_std::sync::Arc;
36///
37/// struct Service {};
38/// # async_std::task::block_on(async {
39/// let service: Arc<Service> = get("svc").await.unwrap();
40/// # })
41/// ```
42pub async fn get<T: Any + Send + Sync>(name: &str) -> Result<Arc<T>, Error> {
43 unsafe {
44 if has(name).await {
45 let srv = (&CONTAINER)
46 .read()
47 .await
48 .get(name)
49 .unwrap()
50 .clone()
51 .downcast::<T>()
52 .map_err(|_| Error::TypeMismatch(name.to_string()))?;
53
54 return Ok(srv);
55 }
56
57 Err(Error::NotFound(name.to_string()))
58 }
59}
60
61/// Stores named service
62///
63/// To make services singleton, each of them is wrappen within [async_std::sync::Arc]
64///
65/// ```
66/// use icee_container_rs::set;
67///
68/// struct Service {};
69/// # async_std::task::block_on(async {
70/// set("svc", Service{}).await;
71/// # })
72/// ```
73pub async fn set<T: Any + Send + Sync>(name: &'static str, service: T) {
74 unsafe {
75 (&CONTAINER).write().await.insert(name, Arc::new(service));
76 }
77}