Log incoming HTTP headers in WebSocket handlers

Added logging of all incoming HTTP headers in both mixed_handler and v2_mixed_handler functions for improved request tracing and debugging. Each header is logged with its name and value, handling non-UTF8 values gracefully.

Author: DarumaDocker

src/services/mod.rs

@@ -2,6 +2,7 @@ use std::sync::Arc;
 
 use axum::{
     extract::{Path, Query, WebSocketUpgrade},
+    http::HeaderMap,
     response::{IntoResponse, Response},
     Extension,
 };
@@ -22,10 +23,16 @@ pub struct ConnectQueryParams {
 pub async fn mixed_handler(
     Extension(pool): Extension<Arc<ws::WsSetting>>,
     Extension(record_setting): Extension<Arc<ws_record::WsRecordSetting>>,
+    headers: HeaderMap,
     ws: WebSocketUpgrade,
     Path(id): Path<String>,
     Query(params): Query<ConnectQueryParams>,
 ) -> Response {
+    for (name, value) in headers.iter() {
+        let value = value.to_str().unwrap_or("<non-utf8>");
+        log::info!("Incoming header: {}: {}", name, value);
+    }
+
     if params.record {
         ws_record::ws_handler(Extension(record_setting), ws, Path(id))
             .await
@@ -47,10 +54,16 @@ pub async fn mixed_handler(
 pub async fn v2_mixed_handler(
     Extension(record_setting): Extension<Arc<ws_record::WsRecordSetting>>,
     Extension(pool): Extension<Arc<ws::stable::StableWsSetting>>,
+    headers: HeaderMap,
     ws: WebSocketUpgrade,
     Path(id): Path<String>,
     Query(params): Query<ConnectQueryParams>,
 ) -> Response {
+    for (name, value) in headers.iter() {
+        let value = value.to_str().unwrap_or("<non-utf8>");
+        log::info!("Incoming header: {}: {}", name, value);
+    }
+
     if params.record {
         ws_record::ws_handler(Extension(record_setting), ws, Path(id))
             .await