by-loco 0.15.1

The one-person framework for Rust
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

use async_trait::async_trait;
use axum::{Extension, Router as AxumRouter};

use crate::{
    app::{AppContext, Initializer},
    db, Error, Result,
};

#[allow(clippy::module_name_repetitions)]
pub struct MultiDbInitializer;

#[async_trait]
impl Initializer for MultiDbInitializer {
    fn name(&self) -> String {
        "multi_db".to_string()
    }

    async fn after_routes(&self, router: AxumRouter, ctx: &AppContext) -> Result<AxumRouter> {
        let settings = ctx
            .config
            .initializers
            .clone()
            .ok_or_else(|| Error::Message("settings config not configured".to_string()))?;

        let multi_db = settings
            .get("multi_db")
            .ok_or_else(|| Error::Message("multi_db not configured".to_string()))?;

        let multi_db = db::MultiDb::new(serde_json::from_value(multi_db.clone())?).await?;
        Ok(router.layer(Extension(multi_db)))
    }
}