rok-container 0.3.4

IoC service container and dependency injection for the rok ecosystem
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
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57

use std::any::Any;
use std::sync::Arc;

use axum::extract::FromRequestParts;
use axum::http::request::Parts;

use crate::container::Container;
use crate::error::ContainerError;

/// Axum extractor that resolves `Arc<T>` from an [`Arc<Container>`] extension.
///
/// Requires [`Arc<Container>`] to be installed as an Axum [`Extension`](axum::Extension)
/// layer.
///
/// # Example
///
/// ```rust,ignore
/// use std::sync::Arc;
/// use axum::{routing::post, Extension, Router};
/// use rok_container::{Container, Inject};
///
/// async fn store(Inject(svc): Inject<UserService>) -> impl IntoResponse {
///     svc.create_user().await
/// }
///
/// let container = Arc::new(Container::new());
/// container.singleton(UserService::new());
///
/// let app = Router::new()
///     .route("/users", post(store))
///     .layer(Extension(container));
/// ```
pub struct Inject<T>(pub Arc<T>);

impl<T> Inject<T> {
    /// Unwrap the inner `Arc<T>`.
    pub fn into_inner(self) -> Arc<T> {
        self.0
    }
}

impl<T, S> FromRequestParts<S> for Inject<T>
where
    T: Any + Send + Sync + 'static,
    S: Send + Sync,
{
    type Rejection = ContainerError;

    async fn from_request_parts(parts: &mut Parts, _state: &S) -> Result<Self, Self::Rejection> {
        let container = parts
            .extensions
            .get::<Arc<Container>>()
            .cloned()
            .ok_or(ContainerError::NotRegistered(std::any::type_name::<T>()))?;
        Ok(Inject(container.make::<T>()?))
    }
}