scouter-server 0.1.2

Scouter server for model monitoring
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 std::sync::Arc;

use axum::{
    routing::{get, post},
    Router,
};

use crate::api::handler::{get_drift, health_check, insert_drift, insert_drift_profile};
use crate::sql::postgres::PostgresClient;
use axum::http::{
    header::{ACCEPT, AUTHORIZATION, CONTENT_TYPE},
    Method,
};
use tower_http::cors::CorsLayer;

pub struct AppState {
    pub db: PostgresClient,
}

pub fn create_router(app_state: Arc<AppState>) -> Router {
    let cors = CorsLayer::new()
        .allow_methods([Method::GET, Method::PUT, Method::DELETE])
        .allow_credentials(true)
        .allow_headers([AUTHORIZATION, ACCEPT, CONTENT_TYPE]);

    Router::new()
        .route("/healthcheck", get(health_check))
        .route("/drift", get(get_drift).post(insert_drift))
        .route("/profile", post(insert_drift_profile))
        .with_state(app_state)
        .layer(cors)
}