fix build

predibase · Jan 16, 2025 · 5751b1d · 5751b1d
1 parent f0a7d9f
commit 5751b1d
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/router/src/server.rs b/router/src/server.rs
@@ -2079,6 +2079,15 @@ async fn tokenize(
     }
 }
 
+// Implements Tracing Value for Header Value, to support recording header values
+impl tracing::Value for HeaderValue {
+    fn record(&self, key: &tracing::field::Field, visitor: &mut dyn Visit) {
+        // Convert HeaderValue to a string representation safely
+        let value_str = self.to_str().unwrap_or("<invalid utf-8>");
+        key.record_str(value_str, visitor);
+    }
+}
+
 fn trace_headers(headers: HeaderMap, span: &tracing::Span) {
     headers
         .get("x-predibase-tenant")