derust 0.2.6

Easy way to start your Rust asynchronous application server using Tokio and Axum frameworks.
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

use axum::Router;
use std::net::{Ipv4Addr, SocketAddr};
use tokio::net::TcpListener;

use crate::httpx::extension::apply_middlewares;
use crate::httpx::AppContext;
use crate::shutdown_signal;
use tracing::info;
use wg::WaitGroup;

#[cfg(feature = "outbox")]
use crate::outboxx;
#[cfg(feature = "outbox")]
use outbox_pattern_processor::outbox_resources::OutboxProcessorResources;

pub async fn start<T>(
    port: u16,
    context: AppContext<T>,
    router: Router<AppContext<T>>,
    #[cfg(feature = "outbox")] outbox_processor_resources: OutboxProcessorResources,
) -> Result<(), Box<dyn std::error::Error>>
where
    T: Clone + Send + Sync + 'static,
{
    let wg = WaitGroup::new();
    let http_router = apply_middlewares(router, context);
    tokio::spawn(start_http_server(wg.add(1), port, http_router));

    #[cfg(feature = "outbox")]
    tokio::spawn(outboxx::run(wg.add(1), outbox_processor_resources));

    wg.wait();

    info!("Shutdown completed!");

    Ok(())
}

#[cfg(feature = "start_test")]
pub async fn start_test<T>(
    context: AppContext<T>,
    router: Router<AppContext<T>>,
    listener: TcpListener,
) -> std::io::Result<()>
where
    T: Clone + Send + Sync + 'static,
{
    let http_router = apply_middlewares(router, context);

    axum::serve(listener, http_router.into_make_service())
        .with_graceful_shutdown(shutdown_signal())
        .await
}

async fn start_http_server(wg: WaitGroup, port: u16, router: Router<()>) {
    info!("Starting http server on port {}", port);

    let addr = SocketAddr::from((Ipv4Addr::UNSPECIFIED, port));

    if let Ok(listener) = TcpListener::bind(addr).await {
        let _ = axum::serve(listener, router.into_make_service())
            .with_graceful_shutdown(shutdown_signal())
            .await;
    }

    wg.done();

    info!("Http server stopped!");
}