From 34584514652a77559c3d7491a9587ebf9be0b340 Mon Sep 17 00:00:00 2001 From: amigin Date: Mon, 14 Oct 2024 14:37:01 +0300 Subject: [PATCH] Fixes --- .../subscriber/delivered_message_telemetry.rs | 17 ++++++----------- 1 file changed, 6 insertions(+), 11 deletions(-) diff --git a/my-service-bus-abstractions/src/subscriber/delivered_message_telemetry.rs b/my-service-bus-abstractions/src/subscriber/delivered_message_telemetry.rs index c34ccee..53af597 100644 --- a/my-service-bus-abstractions/src/subscriber/delivered_message_telemetry.rs +++ b/my-service-bus-abstractions/src/subscriber/delivered_message_telemetry.rs @@ -64,12 +64,10 @@ impl DeliveredMessageTelemetry { if self.event_duration_tracker.is_none() { if let Some(telemetry_event_name) = self.telemetry_event_name.take() { let mut event_duration_tracker = - self.ctx.start_event_tracking(telemetry_event_name); - - event_duration_tracker.add_tag( - "msg_id".to_string(), - self.message_id.get_value().to_string(), - ); + self.ctx.start_event_tracking(telemetry_event_name).add_tag( + "msg_id".to_string(), + self.message_id.get_value().to_string(), + ); event_duration_tracker.ignore_this_event(); @@ -85,12 +83,9 @@ impl DeliveredMessageTelemetry { key: impl Into>, value: impl Into>, ) { - match self.event_duration_tracker.as_mut() { + match self.event_duration_tracker.take() { Some(event_duration_tracker) => { - let key: StrOrString<'static> = key.into(); - let value: StrOrString<'static> = value.into(); - - event_duration_tracker.add_tag(key.to_string(), value.to_string()); + self.event_duration_tracker = Some(event_duration_tracker.add_tag(key, value)); } None => { panic!("Telemetry is not engaged. Please call engage_telemetry before adding tags")