chore: add logs for throubleshooting

Farfetch · Jun 11, 2024 · 129b58d · 129b58d
1 parent 2e2c6e4
commit 129b58d
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/src/KafkaFlow/Producers/MessageProducer.cs b/src/KafkaFlow/Producers/MessageProducer.cs
@@ -188,9 +188,19 @@ public void Dispose()
         _producer?.Dispose();
     }
 
-    private static void FillContextWithResultMetadata(IMessageContext context, DeliveryResult<byte[], byte[]> result)
+    private void FillContextWithResultMetadata(IMessageContext context, DeliveryResult<byte[], byte[]> result)
     {
-        var concreteProducerContext = (ProducerContext)context.ProducerContext;
+        var concreteProducerContext = context.ProducerContext as ProducerContext;
+
+        if (concreteProducerContext is null)
+        {
+            _logHandler.Warning("Producer context is null on FillContextWithResultMetadata", new
+            {
+                DeliveryResult = result,
+            });
+
+            return;
+        }
 
         concreteProducerContext.Offset = result.Offset;
         concreteProducerContext.Partition = result.Partition;