deadnews-template-rust 1.2.1

Rust Project Template
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

use axum::{Json, Router, extract::State, response::IntoResponse, routing::get};
use sqlx::PgPool;
use tower::ServiceBuilder;
use tower_http::{
    catch_panic::CatchPanicLayer,
    request_id::{MakeRequestUuid, SetRequestIdLayer},
    trace::TraceLayer,
};

use crate::db::get_database_info;
use crate::error::AppError;

#[derive(Clone)]
pub struct AppState {
    pub db: PgPool,
}

pub fn create_router(state: AppState) -> Router {
    Router::new()
        .route("/health", get(health_check))
        .route("/test", get(database_test))
        .layer(
            ServiceBuilder::new()
                .layer(SetRequestIdLayer::x_request_id(MakeRequestUuid))
                .layer(TraceLayer::new_for_http())
                .layer(CatchPanicLayer::new()),
        )
        .with_state(state)
}

async fn health_check() -> &'static str {
    "Healthy"
}

pub async fn database_test(State(state): State<AppState>) -> Result<impl IntoResponse, AppError> {
    let info = get_database_info(&state.db).await?;
    Ok(Json(info))
}