Fixes

MyJetTools · Feb 21, 2025 · ad22dd2 · ad22dd2
1 parent 4c79d4e
commit ad22dd2
Showing 1 changed file with 4 additions and 2 deletions.
diff --git 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
     }
 }