Update AppInsightsConsumerEventsHandler.cs

agoda-com · Dec 21, 2023 · 77be5a2 · 77be5a2
1 parent 9a23a02
commit 77be5a2
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/src/KafkaFlow.ApplicationInsights/AppInsightsConsumerEventsHandler.cs b/src/KafkaFlow.ApplicationInsights/AppInsightsConsumerEventsHandler.cs
@@ -14,9 +14,10 @@ public static Task OnConsumeStarted(IMessageContext eventContextMessageContext,
         return Task.CompletedTask;
     }
 
-    public static Task OnConsumeError(IMessageContext eventContextMessageContext, Exception eventContextException,
-        TelemetryClient telemetryClient)
+    public static Task OnConsumeError(IMessageContext eventContextMessageContext, Exception eventContextException)
     {
+        eventContextMessageContext.Items.TryGetValue("telemetryClient", out var telemetryClientOut);
+        var telemetryClient = (TelemetryClient)telemetryClientOut;
         telemetryClient.TrackException(eventContextException, new Dictionary<string, string>()
         {
             {"topic" , eventContextMessageContext.ConsumerContext.Topic},
@@ -50,4 +51,4 @@ public static Task OnConsumeCompleted(IMessageContext eventContextMessageContext
             theTimer.Elapsed, "200", true);
         return Task.CompletedTask;
     }
-}
+}