Merge #718: Refactor: use tower_http::request_id::MakeRequestUuid

eb8478d refactor: [#714] use tower_http::request_id::MakeRequestUuid (Jose Celano)

Pull request description:

  Instead of a custom request UUID generator. I did not know there was already an implementation for it when I implemented that.

  See: https://docs.rs/tower-http/latest/tower_http/request_id/struct.MakeRequestUuid.html

ACKs for top commit:
  josecelano:
    ACK eb8478d

Tree-SHA512: af605f267ab99db98b6f55885343328a7b77ec23275b3dc495a5aa6093ecae06b3f0e80c640e14c504fd40b3f6e8bdd6df2619f60785294827237aeb98262f88

Author: josecelano

src/servers/apis/routes.rs

@@ -8,18 +8,17 @@
 use std::sync::Arc;
 use std::time::Duration;
 
-use axum::http::{HeaderName, HeaderValue};
+use axum::http::HeaderName;
 use axum::response::Response;
 use axum::routing::get;
 use axum::{middleware, Router};
 use hyper::Request;
 use torrust_tracker_configuration::AccessTokens;
 use tower_http::compression::CompressionLayer;
 use tower_http::propagate_header::PropagateHeaderLayer;
-use tower_http::request_id::{MakeRequestId, RequestId, SetRequestIdLayer};
+use tower_http::request_id::{MakeRequestUuid, SetRequestIdLayer};
 use tower_http::trace::{DefaultMakeSpan, TraceLayer};
 use tracing::{Level, Span};
-use uuid::Uuid;
 
 use super::v1;
 use super::v1::context::health_check::handlers::health_check_handler;
@@ -41,7 +40,7 @@ pub fn router(tracker: Arc<Tracker>, access_tokens: Arc<AccessTokens>) -> Router
         .layer(middleware::from_fn_with_state(state, v1::middlewares::auth::auth))
         .route(&format!("{api_url_prefix}/health_check"), get(health_check_handler))
         .layer(CompressionLayer::new())
-        .layer(SetRequestIdLayer::x_request_id(RequestIdGenerator))
+        .layer(SetRequestIdLayer::x_request_id(MakeRequestUuid))
         .layer(PropagateHeaderLayer::new(HeaderName::from_static("x-request-id")))
         .layer(
             TraceLayer::new_for_http()
@@ -73,15 +72,5 @@ pub fn router(tracker: Arc<Tracker>, access_tokens: Arc<AccessTokens>) -> Router
                         tracing::Level::INFO, "response", latency = %latency_ms, status = %status_code, request_id = %request_id);
                 }),
         )
-        .layer(SetRequestIdLayer::x_request_id(RequestIdGenerator))
-}
-
-#[derive(Clone, Default)]
-struct RequestIdGenerator;
-
-impl MakeRequestId for RequestIdGenerator {
-    fn make_request_id<B>(&mut self, _request: &Request<B>) -> Option<RequestId> {
-        let id = HeaderValue::from_str(&Uuid::new_v4().to_string()).expect("UUID is a valid HTTP header value");
-        Some(RequestId::new(id))
-    }
+        .layer(SetRequestIdLayer::x_request_id(MakeRequestUuid))
 }

src/servers/health_check_api/server.rs

@@ -5,7 +5,7 @@
 use std::net::SocketAddr;
 use std::time::Duration;
 
-use axum::http::{HeaderName, HeaderValue};
+use axum::http::HeaderName;
 use axum::response::Response;
 use axum::routing::get;
 use axum::{Json, Router};
@@ -17,10 +17,9 @@ use serde_json::json;
 use tokio::sync::oneshot::{Receiver, Sender};
 use tower_http::compression::CompressionLayer;
 use tower_http::propagate_header::PropagateHeaderLayer;
-use tower_http::request_id::{MakeRequestId, RequestId, SetRequestIdLayer};
+use tower_http::request_id::{MakeRequestUuid, SetRequestIdLayer};
 use tower_http::trace::{DefaultMakeSpan, TraceLayer};
 use tracing::{Level, Span};
-use uuid::Uuid;
 
 use crate::bootstrap::jobs::Started;
 use crate::servers::health_check_api::handlers::health_check_handler;
@@ -43,7 +42,7 @@ pub fn start(
         .route("/health_check", get(health_check_handler))
         .with_state(register)
         .layer(CompressionLayer::new())
-        .layer(SetRequestIdLayer::x_request_id(RequestIdGenerator))
+        .layer(SetRequestIdLayer::x_request_id(MakeRequestUuid))
         .layer(PropagateHeaderLayer::new(HeaderName::from_static("x-request-id")))
         .layer(
             TraceLayer::new_for_http()
@@ -75,7 +74,7 @@ pub fn start(
                         tracing::Level::INFO, "response", latency = %latency_ms, status = %status_code, request_id = %request_id);
                 }),
         )
-        .layer(SetRequestIdLayer::x_request_id(RequestIdGenerator));
+        .layer(SetRequestIdLayer::x_request_id(MakeRequestUuid));
 
     let socket = std::net::TcpListener::bind(bind_to).expect("Could not bind tcp_listener to address.");
     let address = socket.local_addr().expect("Could not get local_addr from tcp_listener.");
@@ -99,13 +98,3 @@ pub fn start(
 
     running
 }
-
-#[derive(Clone, Default)]
-struct RequestIdGenerator;
-
-impl MakeRequestId for RequestIdGenerator {
-    fn make_request_id<B>(&mut self, _request: &Request<B>) -> Option<RequestId> {
-        let id = HeaderValue::from_str(&Uuid::new_v4().to_string()).expect("UUID is a valid HTTP header value");
-        Some(RequestId::new(id))
-    }
-}

src/servers/http/v1/routes.rs

@@ -3,18 +3,17 @@ use std::net::SocketAddr;
 use std::sync::Arc;
 use std::time::Duration;
 
-use axum::http::{HeaderName, HeaderValue};
+use axum::http::HeaderName;
 use axum::response::Response;
 use axum::routing::get;
 use axum::Router;
 use axum_client_ip::SecureClientIpSource;
 use hyper::Request;
 use tower_http::compression::CompressionLayer;
 use tower_http::propagate_header::PropagateHeaderLayer;
-use tower_http::request_id::{MakeRequestId, RequestId, SetRequestIdLayer};
+use tower_http::request_id::{MakeRequestUuid, SetRequestIdLayer};
 use tower_http::trace::{DefaultMakeSpan, TraceLayer};
 use tracing::{Level, Span};
-use uuid::Uuid;
 
 use super::handlers::{announce, health_check, scrape};
 use crate::core::Tracker;
@@ -37,7 +36,7 @@ pub fn router(tracker: Arc<Tracker>, server_socket_addr: SocketAddr) -> Router {
         // Add extension to get the client IP from the connection info
         .layer(SecureClientIpSource::ConnectInfo.into_extension())
         .layer(CompressionLayer::new())
-        .layer(SetRequestIdLayer::x_request_id(RequestIdGenerator))
+        .layer(SetRequestIdLayer::x_request_id(MakeRequestUuid))
         .layer(PropagateHeaderLayer::new(HeaderName::from_static("x-request-id")))
         .layer(
             TraceLayer::new_for_http()
@@ -69,15 +68,5 @@ pub fn router(tracker: Arc<Tracker>, server_socket_addr: SocketAddr) -> Router {
                         tracing::Level::INFO, "response", server_socket_addr= %server_socket_addr, latency = %latency_ms, status = %status_code, request_id = %request_id);
                 }),
         )
-        .layer(SetRequestIdLayer::x_request_id(RequestIdGenerator))
-}
-
-#[derive(Clone, Default)]
-struct RequestIdGenerator;
-
-impl MakeRequestId for RequestIdGenerator {
-    fn make_request_id<B>(&mut self, _request: &Request<B>) -> Option<RequestId> {
-        let id = HeaderValue::from_str(&Uuid::new_v4().to_string()).expect("UUID is a valid HTTP header value");
-        Some(RequestId::new(id))
-    }
+        .layer(SetRequestIdLayer::x_request_id(MakeRequestUuid))
 }