From ad22dd2e61d26cc306591a105d74b847bb159f8d Mon Sep 17 00:00:00 2001
From: amigin <amigin@gmail.com>
Date: Fri, 21 Feb 2025 19:42:14 +0200
Subject: [PATCH] Fixes

---
 .../src/subscriber/delivered_message.rs                     | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/my-service-bus-abstractions/src/subscriber/delivered_message.rs b/my-service-bus-abstractions/src/subscriber/delivered_message.rs
index e585e1c..a27100b 100644
--- a/my-service-bus-abstractions/src/subscriber/delivered_message.rs
+++ b/my-service-bus-abstractions/src/subscriber/delivered_message.rs
@@ -53,12 +53,14 @@ impl<TMessageModel: MySbMessageDeserializer<Item = TMessageModel>>
         }
     }
     #[cfg(feature = "with-telemetry")]
-    pub async fn engage_telemetry(&self) {
+    pub async fn engage_telemetry(&self) -> my_telemetry::MyTelemetryContext {
         let inner = self.inner.as_ref().unwrap();
         let mut inner = inner.lock().await;
 
         if let Some(my_telemetry) = inner.current_message_telemetry.as_mut() {
-            my_telemetry.engage_telemetry();
+            return my_telemetry.engage_telemetry();
         }
+
+        my_telemetry::MyTelemetryContext::Empty
     }
 }