diff --git a/Cargo.lock b/Cargo.lock index 462d7a5..fc6231e 100644 --- a/Cargo.lock +++ b/Cargo.lock @@ -2363,6 +2363,7 @@ dependencies = [ "pin-project-lite", "tower-layer", "tower-service", + "tracing", ] [[package]] diff --git a/Cargo.toml b/Cargo.toml index e80324d..0c85e87 100644 --- a/Cargo.toml +++ b/Cargo.toml @@ -15,7 +15,7 @@ serde = { version = "1.0.219", features = ["derive"] } serde_json = "1.0.143" sqlx = { version = "0.8.6", features = ["postgres", "runtime-tokio-native-tls", "macros", "uuid", "chrono"] } tokio = { version = "1.47.1", features = ["rt-multi-thread", "macros"] } -tower-http = { version = "0.6.6", features = ["cors", "limit"] } +tower-http = { version = "0.6.6", features = ["cors", "limit", "trace"] } tower_governor = "0.8.0" tracing = "0.1.41" tracing-subscriber = "0.3.20" diff --git a/src/main.rs b/src/main.rs index 199b75b..f5f536a 100644 --- a/src/main.rs +++ b/src/main.rs @@ -4,7 +4,11 @@ use axum::{ }; use std::{env::var, net::SocketAddr, time::Duration}; use tower_governor::{GovernorLayer, governor::GovernorConfigBuilder}; -use tower_http::cors::{Any, CorsLayer}; +use tower_http::{ + cors::{Any, CorsLayer}, + trace::{DefaultMakeSpan, DefaultOnResponse, TraceLayer}, +}; +use tracing::Level; mod auth; mod db; @@ -13,8 +17,9 @@ mod routes; #[tokio::main] async fn main() -> anyhow::Result<()> { - let subscriber = tracing_subscriber::FmtSubscriber::new(); - tracing::subscriber::set_global_default(subscriber).unwrap(); + tracing_subscriber::fmt() + .with_max_level(tracing::Level::INFO) + .init(); tracing::info!("Connecting to database..."); let db_pool = db::init_db().await?; @@ -25,8 +30,8 @@ async fn main() -> anyhow::Result<()> { .allow_headers([header::AUTHORIZATION, header::CONTENT_TYPE]); let governor_conf = GovernorConfigBuilder::default() - .per_second(50) - .burst_size(200) + .burst_size(10) + .per_millisecond(500) .finish() .unwrap(); @@ -52,8 +57,13 @@ async fn main() -> anyhow::Result<()> { .merge(routes::ws::routes()) .layer(Extension(db_pool)) .layer(Extension(realtime)) - .layer(cors) - .layer(GovernorLayer::new(governor_conf)); + .layer(GovernorLayer::new(governor_conf)) + .layer( + TraceLayer::new_for_http() + .make_span_with(DefaultMakeSpan::new().level(Level::INFO)) + .on_response(DefaultOnResponse::new().level(Level::INFO)), + ) + .layer(cors); let port = var("CHATAPP_SERVER_PORT").unwrap_or_else(|_| "8080".to_string()); let addr = format!("127.0.0.1:{port}");