diff --git a/gateway/src/node/client_handling/websocket/connection_handler/authenticated.rs b/gateway/src/node/client_handling/websocket/connection_handler/authenticated.rs index ebbb79a974..d1f7fe4808 100644 --- a/gateway/src/node/client_handling/websocket/connection_handler/authenticated.rs +++ b/gateway/src/node/client_handling/websocket/connection_handler/authenticated.rs @@ -236,6 +236,7 @@ impl AuthenticatedHandler { /// # Arguments /// /// * `mix_packet`: packet received from the client that should get forwarded into the network. + #[instrument(skip_all)] fn forward_packet(&self, mix_packet: MixPacket) { let herited_span = self.root_span.as_ref().cloned().unwrap_or_else(|| tracing::Span::none()); let span = info_span!(parent: &herited_span, "forwarding_packet"); @@ -295,6 +296,7 @@ impl AuthenticatedHandler { /// # Arguments /// /// * `mix_packet`: packet received from the client that should get forwarded into the network. + #[instrument(skip_all)] async fn handle_forward_sphinx( &mut self, mix_packet: MixPacket, @@ -321,6 +323,7 @@ impl AuthenticatedHandler { /// # Arguments /// /// * `bin_msg`: raw message to handle. + #[instrument(skip_all)] async fn handle_binary(&mut self, bin_msg: Vec) -> Message { trace!("binary request"); let herited_span = self.root_span.as_ref().cloned().unwrap_or_else(|| tracing::Span::none()); diff --git a/gateway/src/node/client_handling/websocket/connection_handler/fresh.rs b/gateway/src/node/client_handling/websocket/connection_handler/fresh.rs index 90e7821ef9..3910021c27 100644 --- a/gateway/src/node/client_handling/websocket/connection_handler/fresh.rs +++ b/gateway/src/node/client_handling/websocket/connection_handler/fresh.rs @@ -894,12 +894,12 @@ impl FreshHandler { }; let span_cx = new_span_context_with_id(trace_id); let _context_guard = span_cx.clone().attach(); - warn!("==== Context propagation successful ===="); // Build root_span with extracted context as parent let span = info_span!("=== Manual context propagation starting point ===", %trace_id); span.set_parent(span_cx.clone()); let context_ext = context_ext.set_root_span(span); + warn!("==== Context propagation successful ===="); context_ext } else { warn!("No OpenTelemetry context provided in the request");