From 1fef24a6ba425eb6908a281b5f0dead74fcf847c Mon Sep 17 00:00:00 2001
From: Floriane TUERNAL SABOTINOV <ftuertinov@proton.me>
Date: Mon, 8 Sep 2025 13:57:28 +0200
Subject: [PATCH] try debug panic when using parent span

---
 .../client_handling/websocket/connection_handler/fresh.rs  | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

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 5966ed0eaa..2bbf0b465d 100644
--- a/gateway/src/node/client_handling/websocket/connection_handler/fresh.rs
+++ b/gateway/src/node/client_handling/websocket/connection_handler/fresh.rs
@@ -990,8 +990,7 @@ impl<R, S> FreshHandler<R, S> {
                 }
             };
 
-            if let (Some(registration_details), Some(herited_span)) = (maybe_auth_res, herited_span) {
-
+            if let (Some(registration_details), Some(ref herited_span)) = (maybe_auth_res, herited_span) {
                 let span = info_span!(parent: herited_span, "upgrading_to_authenticated_handler");
                 let _enter = span.enter();
                 let (mix_sender, mix_receiver) = mpsc::unbounded();
@@ -1004,7 +1003,8 @@ impl<R, S> FreshHandler<R, S> {
                     registration_details.session_request_timestamp,
                 );
 
-                return AuthenticatedHandler::upgrade(
+                let exit_span = info_span!("upgraded_to_authenticated_handler");
+                let auth_handle = AuthenticatedHandler::upgrade(
                     self,
                     registration_details,
                     mix_receiver,
@@ -1013,6 +1013,7 @@ impl<R, S> FreshHandler<R, S> {
                 .await
                 .inspect_err(|err| error!("failed to upgrade client handler: {err}"))
                 .ok();
+                return auth_handle;
             }
         }
     }