test

Farfetch · Oct 17, 2023 · 9c38849 · 9c38849
1 parent 2e2a718
commit 9c38849
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/src/KafkaFlow.OpenTelemetry/OpenTelemetryConsumerEventsHandler.cs b/src/KafkaFlow.OpenTelemetry/OpenTelemetryConsumerEventsHandler.cs
@@ -36,6 +36,8 @@ public static Task OnConsumeStarted(IMessageContext context)
                     activity?.AddBaggage(item.Key, item.Value);
                 }
 
+                Activity.Current = activity;
+
                 context?.Items.Add(ActivitySourceAccessor.ActivityString, activity);
 
                 ActivitySourceAccessor.SetGenericTags(activity);
@@ -57,7 +59,7 @@ public static Task OnConsumeCompleted(IMessageContext context)
         {
             if (context.Items.TryGetValue(ActivitySourceAccessor.ActivityString, out var value) && value is Activity activity)
             {
-                //activity?.Stop();
+                activity?.Stop();
             }
 
             return Task.CompletedTask;
@@ -71,7 +73,7 @@ public static Task OnConsumeError(IMessageContext context, Exception ex)
 
                 activity?.AddEvent(exceptionEvent);
 
-                //activity?.Stop();
+                activity?.Stop();
             }
 
             return Task.CompletedTask;

diff --git a/src/KafkaFlow.OpenTelemetry/OpenTelemetryProducerEventsHandler.cs b/src/KafkaFlow.OpenTelemetry/OpenTelemetryProducerEventsHandler.cs
@@ -67,7 +67,7 @@ public static Task OnProducerCompleted(IMessageContext context)
         {
             if (context.Items.TryGetValue(ActivitySourceAccessor.ActivityString, out var value) && value is Activity activity)
             {
-                //activity?.Stop();
+                activity?.Stop();
             }
 
             return Task.CompletedTask;
@@ -81,7 +81,7 @@ public static Task OnProducerError(IMessageContext context, Exception ex)
 
                 activity?.AddEvent(exceptionEvent);
 
-                //activity?.Stop();
+                activity?.Stop();
             }
 
             return Task.CompletedTask;