fix issue (#84)

EcovadisCode · Nov 28, 2024 · a252fa0 · a252fa0
1 parent 267fec5
commit a252fa0
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 7 deletions.
diff --git a/docs/CHANGELOG.md b/docs/CHANGELOG.md
@@ -3,6 +3,9 @@ All notable changes to this project will be documented in this file.
 
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
+## 5.1.4
+- Changed
+  - Fix diagnostic id null reference
 
 ## 5.1.3
 - Changed

diff --git a/src/Ev.ServiceBus.Abstractions/Extensions/DiagnosticIdExtension.cs b/src/Ev.ServiceBus.Abstractions/Extensions/DiagnosticIdExtension.cs
@@ -5,23 +5,28 @@ namespace Ev.ServiceBus.Abstractions.Extensions;
 public static class DiagnosticIdExtension
 {
     private const string DiagnosticIdKey = "Diagnostic-Id";
+
     public static string? GetDiagnosticId(this ServiceBusReceivedMessage message)
     {
-        return message.ApplicationProperties.ContainsKey(DiagnosticIdKey)
-            ? message.ApplicationProperties[DiagnosticIdKey].ToString()
-            : null;
+        if (message.ApplicationProperties.ContainsKey(DiagnosticIdKey) && message.ApplicationProperties[DiagnosticIdKey] != null)
+        {
+            return message.ApplicationProperties[DiagnosticIdKey].ToString();
+        }
+        return null;
     }
 
     public static string? GetDiagnosticId(this ServiceBusMessage message)
     {
-        return message.ApplicationProperties.ContainsKey(DiagnosticIdKey)
-            ? message.ApplicationProperties[DiagnosticIdKey].ToString()
-            : null;
+        if (message.ApplicationProperties.ContainsKey(DiagnosticIdKey) && message.ApplicationProperties[DiagnosticIdKey] != null)
+        {
+            return message.ApplicationProperties[DiagnosticIdKey].ToString();
+        }
+        return null;
     }
 
     public static void SetDiagnosticIdIfIsNot(this ServiceBusMessage message, string diagnosticId)
     {
-        if(message.ApplicationProperties.ContainsKey(DiagnosticIdKey))
+        if (message.ApplicationProperties.ContainsKey(DiagnosticIdKey) && message.ApplicationProperties[DiagnosticIdKey] != null)
             return;
         message.ApplicationProperties.Add(DiagnosticIdKey, diagnosticId);
     }