tracing

1 files changed, 13 insertions, 2 deletions

diff --git a/src/main.rs b/src/main.rs
index 54c73c1..ab5a9cc 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,7 +1,9 @@
 use std::net::SocketAddr;
 use axum::{Router, response::Response, http::Request, middleware::{Next, self}, extract::ConnectInfo, RequestExt, body::HttpBody, Extension};
 use tower_cookies::CookieManagerLayer;
-use types::{response::ResponseCode};
+use tracing::metadata::LevelFilter;
+use tracing_subscriber::{prelude::__tracing_subscriber_SubscriberExt, util::SubscriberInitExt, Layer, filter::filter_fn};
+use types::response::ResponseCode;
 
 use crate::{api::{pages, auth, users, posts}, types::extract::RouterURI};
 
@@ -42,6 +44,15 @@ async fn main() {
 
     database::init().unwrap();
 
+    let fmt_layer = tracing_subscriber::fmt::layer();
+    tracing_subscriber::registry() 
+        .with(
+            fmt_layer.with_filter(LevelFilter::TRACE).with_filter(filter_fn(|metadata| {
+                metadata.target().starts_with("xssbook")
+            }))
+        )
+        .init();
+
     let app = Router::new()
     .fallback(not_found)
     .nest("/", pages::router())
@@ -56,7 +67,7 @@ async fn main() {
     ).layer(CookieManagerLayer::new());
 
     let addr = "[::]:8080".parse::<std::net::SocketAddr>().unwrap();
-    println!("Listening on {}", addr);
+    tracing::info!("listening on {}", addr);
     
     axum::Server::bind(&addr)
         .serve(app.into_make_service_with_connect_info::<SocketAddr>())