nidrs 0.4.0

Nidrs is a web framework based on axum and tokio.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42

use nidrs_extern::{datasets::ServiceName, once_cell};
use once_cell::sync::OnceCell;
use std::{any::Any, sync::Arc};

use crate::{ImplMeta, ModuleCtx};

pub trait Service: ImplMeta {
    fn inject(&self, ctx: ModuleCtx, module_name: &str) -> ModuleCtx {
        ctx
    }
}

#[derive(Clone, Debug, Default)]
pub struct Inject<T> {
    value: OnceCell<Arc<T>>,
}

impl<T> Inject<T> {
    pub fn new() -> Self {
        Inject { value: OnceCell::new() }
    }

    pub fn inject(&self, value: Arc<T>) {
        let _ = self.value.set(value);
    }

    pub fn extract(&self) -> Arc<T> {
        self.value.get().unwrap().clone()
    }
}

impl<T> std::ops::Deref for Inject<T> {
    type Target = Arc<T>;
    fn deref(&self) -> &Self::Target {
        self.value.get().unwrap_or_else(|| panic!("{} not inject.", std::any::type_name::<T>()))
    }
}

pub fn provider<T: Service + 'static>(service: T) -> (String, Box<dyn Any>) {
    let name = service.__meta().get_data::<ServiceName>().unwrap().value().clone();
    (name, Box::new(Arc::new(service)) as Box<dyn Any>)
}