Configured TraceLayer in web server to log info about request/response

agama-project · Apr 11, 2024 · c700129 · c700129
1 parent 1bf4dbb
commit c700129
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/rust/agama-server/src/web/service.rs b/rust/agama-server/src/web/service.rs
@@ -1,14 +1,18 @@
 use super::{auth::TokenClaims, config::ServiceConfig, state::ServiceState, EventsSender};
 use axum::{
+    body::Body,
     extract::Request,
     middleware,
     response::IntoResponse,
+    response::Response,
     routing::{get, post},
     Router,
 };
 use std::convert::Infallible;
+use std::time::Duration;
 use tower::Service;
 use tower_http::{compression::CompressionLayer, trace::TraceLayer};
+use tracing::Span;
 
 pub struct MainServiceBuilder {
     config: ServiceConfig,
@@ -55,7 +59,14 @@ impl MainServiceBuilder {
             ))
             .route("/ping", get(super::http::ping))
             .route("/authenticate", post(super::http::authenticate))
-            .layer(TraceLayer::new_for_http())
+            .layer(TraceLayer::new_for_http()
+                .on_request(|request: &Request<Body>, _span: &Span| {
+                    tracing::info!("request: {} {}", request.method(), request.uri().path())
+                })
+                .on_response(|response: &Response<Body>, latency: Duration, _span: &Span| {
+                    tracing::info!("response: {} {:?}", response.status(), latency)
+                })
+            )
             .layer(CompressionLayer::new().br(true))
             .with_state(state)
     }