Add basic telemetry - tracing (#15)

* Add basic telemetry - tracing

* Update to tracing with requests id

Cargo.toml

@@ -14,6 +14,9 @@ serde = { version = "1.0.171", features = ["derive"] }
 config = "0.13.3"
 uuid = { version = "1.4.1", features = ["v4", "serde"] }
 chrono = { version = "0.4.21", features = ["serde"] }
+tracing = "0.1.37"
+tracing-subscriber = "0.3.17"
+tower-http = { version = "0.4.3", features = ["trace"] }
 
 [dependencies.sqlx]
 version = "0.7.1"

src/lib.rs

@@ -1,14 +1,17 @@
 pub mod configuration;
 pub mod model;
+pub mod request_tracing;
 pub mod routes;
 pub mod schema;
 
 use axum::routing::{get, post};
 use axum::Router;
+use request_tracing::RequestSpan;
 use routes::{full_url, health_check};
 use sqlx::{postgres::PgPoolOptions, Pool, Postgres};
 use std::net::TcpListener;
 use std::sync::Arc;
+use tower_http::trace::TraceLayer;
 
 pub async fn run(listener: TcpListener, db_connection: &str) {
     let pool = connect_to_database(db_connection)
@@ -32,6 +35,7 @@ pub fn app(app_state: Arc<AppState>) -> Router {
     Router::new()
         .route("/health_check", get(health_check))
         .route("/full_url", post(full_url))
+        .layer(TraceLayer::new_for_http().make_span_with(RequestSpan))
         .with_state(app_state)
 }
 

src/main.rs

@@ -1,8 +1,13 @@
 use briefly::{configuration::get_configuration, run};
 use std::net::TcpListener;
+use tracing_subscriber::{layer::SubscriberExt, util::SubscriberInitExt};
 
 #[tokio::main]
 async fn main() {
+    tracing_subscriber::registry()
+        .with(tracing_subscriber::fmt::layer())
+        .init();
+
     let configuration = get_configuration().expect("Failed to read configuration");
     let connection_string = configuration.database.connection_string();
     let address = format!("127.0.0.1:{}", configuration.application_port);

src/request_tracing.rs

@@ -0,0 +1,17 @@
+use axum::http;
+use uuid::Uuid;
+
+#[derive(Clone)]
+pub struct RequestSpan;
+
+impl<B> tower_http::trace::MakeSpan<B> for RequestSpan {
+    fn make_span(&mut self, request: &http::Request<B>) -> tracing::Span {
+        tracing::error_span!(
+            "rq",
+            id = %Uuid::new_v4().to_string(),
+            method = %request.method(),
+            uri = %request.uri(),
+            version = ?request.version(),
+        )
+    }
+}