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}