Fixing an exception when the header already exists (#29)

Fixing an exception when the header already exists
ekmsystems · Nov 8, 2019 · 900281c · 900281c
2 parents d69e401 + fc30f7c
commit 900281c
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/src/Serilog.Enrichers.CorrelationId/Enrichers/CorrelationIdHeaderEnricher.cs b/src/Serilog.Enrichers.CorrelationId/Enrichers/CorrelationIdHeaderEnricher.cs
@@ -52,8 +52,15 @@ private string GetCorrelationId()
             var correlationId = string.IsNullOrEmpty(header)
                                     ? Guid.NewGuid().ToString()
                                     : header;
-
-            _contextAccessor.HttpContext.Response.Headers.Add(_headerKey, correlationId);
+
+#if NETFULL
+            if(!_contextAccessor.HttpContext.Response.Headers.AllKeys.Contains(_headerKey))
+#else
+            if (!_contextAccessor.HttpContext.Response.Headers.ContainsKey(_headerKey))
+#endif
+            {
+                _contextAccessor.HttpContext.Response.Headers.Add(_headerKey, correlationId);
+            }
 
             return correlationId;
         }