roadster 0.9.0-alpha.5

A "Batteries Included" web framework for rust designed to get you moving fast.
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
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84

use crate::app::context::AppContext;
use crate::error::RoadsterResult;
use crate::service::Service;
use async_trait::async_trait;
use axum_core::extract::FromRef;
use std::future::Future;
use std::marker::PhantomData;
use tokio_util::sync::CancellationToken;

/// A generic [Service] to allow creating a service from an async function.
///
/// # Examples
/// ```rust
/// # use async_trait::async_trait;
/// # use tokio_util::sync::CancellationToken;
/// # use roadster::app::context::AppContext;
/// # use roadster::error::RoadsterResult;
/// # use roadster::service::function::service::FunctionService;
/// # use roadster::service::registry::ServiceRegistry;
/// # use roadster::app::RoadsterApp;
///
/// async fn example_service(
///     _state: AppContext,
///     _cancel_token: CancellationToken,
/// ) -> RoadsterResult<()> {
///     // Service logic here
/// #    unimplemented!()
/// }
///
/// type App = RoadsterApp<AppContext>;
///
/// let service = FunctionService::builder()
///             .name("example".to_string())
///             .enabled(true)
///             .function(example_service)
///             .build();
///
/// let app: App = RoadsterApp::builder()
///     .add_service(service)
///     .build();
/// ```
#[derive(bon::Builder)]
pub struct FunctionService<S, F, Fut>
where
    S: 'static + Send + Sync + Clone,
    AppContext: FromRef<S>,
    F: Send + Sync + Fn(S, CancellationToken) -> Fut,
    Fut: Send + Future<Output = RoadsterResult<()>>,
{
    #[builder(into)]
    name: String,
    enabled: Option<bool>,
    function: F,
    #[builder(skip)]
    _state: PhantomData<S>,
}

#[async_trait]
impl<S, F, Fut> Service<S> for FunctionService<S, F, Fut>
where
    S: 'static + Send + Sync + Clone,
    AppContext: FromRef<S>,
    F: 'static + Send + Sync + Fn(S, CancellationToken) -> Fut,
    Fut: 'static + Send + Future<Output = RoadsterResult<()>>,
{
    type Error = crate::error::Error;

    fn name(&self) -> String {
        self.name.clone()
    }

    fn enabled(&self, state: &S) -> bool {
        self.enabled
            .unwrap_or(AppContext::from_ref(state).config().service.default_enable)
    }

    async fn run(
        self: Box<Self>,
        state: &S,
        cancel_token: CancellationToken,
    ) -> RoadsterResult<()> {
        (self.function)(state.clone(), cancel_token).await
    }
}