From 9085485a4f8a6be76d1ac8a8b7a7b65bdf83aa24 Mon Sep 17 00:00:00 2001
From: Conrad Ludgate <conrad.ludgate@truelayer.com>
Date: Thu, 21 Apr 2022 18:07:33 +0100
Subject: tracing (#315)

* enable tracing on server

* fmt

* instrument handlers
---
 atuin-server/src/router.rs | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

(limited to 'atuin-server/src/router.rs')

diff --git a/atuin-server/src/router.rs b/atuin-server/src/router.rs
index 146cc991..9b525e05 100644
--- a/atuin-server/src/router.rs
+++ b/atuin-server/src/router.rs
@@ -7,6 +7,8 @@ use axum::{
     Extension, Router,
 };
 use eyre::Result;
+use tower::ServiceBuilder;
+use tower_http::trace::TraceLayer;
 
 use super::{
     database::{Database, Postgres},
@@ -65,6 +67,10 @@ pub fn router(postgres: Postgres, settings: Settings) -> Router {
         .route("/register", post(handlers::user::register))
         .route("/login", post(handlers::user::login))
         .fallback(teapot.into_service())
-        .layer(Extension(postgres))
-        .layer(Extension(settings))
+        .layer(
+            ServiceBuilder::new()
+                .layer(TraceLayer::new_for_http())
+                .layer(Extension(postgres))
+                .layer(Extension(settings)),
+        )
 }
-- 
cgit v1.3.1