From 4e47e19b128d5be18a0c4acc47964f1fd0b806f6 Mon Sep 17 00:00:00 2001
From: Paul-Christian Volkmer <volkmer_p@ukw.de>
Date: Fri, 8 Mar 2024 14:44:18 +0100
Subject: [PATCH] build: use tracing in debug mode

---
 Cargo.toml  | 11 +++++++++++
 src/main.rs | 12 ++++++++++++
 2 files changed, 23 insertions(+)

diff --git a/Cargo.toml b/Cargo.toml
index 9f07025..57f3164 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -16,11 +16,22 @@ version = "1.6"
 
 [dependencies.axum]
 version = "0.7"
+features = ["tracing"]
+
+[dependencies.tracing-subscriber]
+version = "0.3"
+
+[dependencies.tracing]
+version = "0.1"
 
 [dependencies.tokio]
 version = "1.36"
 features = ["rt-multi-thread"]
 
+[dependencies.tower-http]
+version = "0.5"
+features = ["trace"]
+
 [dependencies.serde]
 version = "1.0"
 features = ["derive"]
diff --git a/src/main.rs b/src/main.rs
index d2c0104..e8f9e0a 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -14,6 +14,8 @@ use rdkafka::ClientConfig;
 use rdkafka::message::{Header, OwnedHeaders};
 use rdkafka::producer::{FutureProducer, FutureRecord};
 use serde::{Deserialize, Serialize};
+#[cfg(debug_assertions)]
+use tower_http::trace::TraceLayer;
 use uuid::Uuid;
 
 mod auth;
@@ -30,6 +32,13 @@ struct RecordKey {
 async fn main() {
     let _ = bcrypt_hashed_token();
 
+    #[cfg(debug_assertions)]
+    {
+        tracing_subscriber::fmt()
+            .with_max_level(tracing::Level::DEBUG)
+            .init();
+    }
+
     let boostrap_servers = env::var("KAFKA_BOOTSTRAP_SERVERS").unwrap_or("kafka:9094".into());
 
     let producer: FutureProducer = ClientConfig::new()
@@ -44,6 +53,9 @@ async fn main() {
         .layer(Extension(producer))
         .layer(from_fn(check_basic_auth));
 
+    #[cfg(debug_assertions)]
+    let app = app.layer(TraceLayer::new_for_http());
+
     let listener = tokio::net::TcpListener::bind("0.0.0.0:8000").await.unwrap();
     axum::serve(listener, app).await.unwrap();
 }