feat(tracing): trace http response & pretty (#49)

* refactor(backend): trace http response

* refactor(tracing): pretty

Cargo.toml

@@ -40,11 +40,12 @@ panic = "abort"
 strip = "symbols"
 
 [features]
-default = ["snmalloc", "hatsu_tracing/console"]
+default = ["snmalloc", "hatsu_tracing/console", "hatsu_tracing/pretty"]
 snmalloc = ["dep:snmalloc-rs"]
 # hatsu_tracing
-tracing-conosle = ["hatsu_tracing/console"]
-tracing-json = ["hatsu_tracing/json"]
+tracing_conosle = ["hatsu_tracing/console"]
+tracing_json = ["hatsu_tracing/json"]
+tracing_pretty = ["hatsu_tracing/pretty"]
 
 [workspace]
 members = [

crates/backend/src/lib.rs

@@ -3,7 +3,11 @@ use std::net::SocketAddr;
 use activitypub_federation::config::{FederationConfig, FederationMiddleware};
 use hatsu_utils::{AppData, AppError};
 use tokio_graceful_shutdown::{IntoSubsystem, SubsystemHandle};
-use tower_http::{cors::CorsLayer, trace::TraceLayer};
+use tower_http::{
+    cors::CorsLayer,
+    trace::{self, TraceLayer},
+};
+use tracing::Level;
 
 mod favicon;
 mod routes;
@@ -31,7 +35,11 @@ impl IntoSubsystem<AppError, AppError> for Server {
         let app = routes::routes()
             .layer(FederationMiddleware::new(self.federation_config))
             .layer(CorsLayer::permissive())
-            .layer(TraceLayer::new_for_http());
+            .layer(
+                TraceLayer::new_for_http()
+                    .make_span_with(trace::DefaultMakeSpan::new().level(Level::INFO))
+                    .on_response(trace::DefaultOnResponse::new().level(Level::INFO)),
+            );
 
         // axum 0.6
         // run our app with hyper

crates/tracing/Cargo.toml

@@ -19,6 +19,7 @@ path = "src/lib.rs"
 default = []
 console = ["dep:console-subscriber"]
 json = ["tracing-subscriber/json"]
+pretty = []
 # TODO: opentelemetry
 
 

crates/tracing/src/lib.rs

@@ -9,9 +9,14 @@ use tracing_subscriber::{
 };
 
 pub fn init() -> Result<(), AppError> {
+    let fmt_layer = fmt_layer();
+
+    #[cfg(feature = "pretty")]
+    let fmt_layer = fmt_layer.pretty();
+
     let registry = tracing_subscriber::registry()
         .with(filter_layer())
-        .with(fmt_layer())
+        .with(fmt_layer)
         .with(tracing_error::ErrorLayer::default());
 
     #[cfg(feature = "console")]