Added aspnet-traceidentifier for AspNet for NLog.Web

NLog · Dec 2, 2018 · 8c1c6ce · 8c1c6ce
1 parent e40731f
commit 8c1c6ce
Show file tree

Hide file tree

Showing 2 changed files with 35 additions and 1 deletion.
diff --git a/NLog.Web.AspNetCore/LayoutRenderers/AspNetTraceIdentifierLayoutRenderer.cs b/NLog.Web.AspNetCore/LayoutRenderers/AspNetTraceIdentifierLayoutRenderer.cs
@@ -17,7 +17,38 @@ public class AspNetTraceIdentifierLayoutRenderer : AspNetLayoutRendererBase
         protected override void DoAppend(StringBuilder builder, LogEventInfo logEvent)
         {
             var httpContext = HttpContextAccessor.HttpContext;
-            builder.Append(httpContext.TraceIdentifier);
+            builder.Append(LookupTraceIdentifier(httpContext));
         }
+
+#if ASP_NET_CORE
+        private string LookupTraceIdentifier(Microsoft.AspNetCore.Http.HttpContext httpContext)
+        {
+            return httpContext.TraceIdentifier;
+        }
+#else
+        /// <summary>
+        /// Requires IIS ETW feature enabled. https://docs.microsoft.com/en-us/iis/configuration/system.webServer/httpTracing/
+        ///
+        /// See also http://blog.tatham.oddie.com.au/2012/02/07/code-request-correlation-in-asp-net/
+        /// </summary>
+        private string LookupTraceIdentifier(System.Web.HttpContextBase httpContext)
+        {
+            var provider = httpContext as IServiceProvider;
+            if (provider != null)
+            {
+                var workerRequest = (System.Web.HttpWorkerRequest)provider.GetService(typeof(System.Web.HttpWorkerRequest));
+                if (workerRequest != null)
+                {
+                    Guid requestIdGuid = workerRequest.RequestTraceIdentifier;
+                    if (requestIdGuid != Guid.Empty)
+                    {
+                        return requestIdGuid.ToString();
+                    }
+                }
+            }
+
+            return null;
+        }
+#endif
     }
 }
diff --git a/NLog.Web/NLog.Web.csproj b/NLog.Web/NLog.Web.csproj
@@ -75,6 +75,9 @@
     <Compile Include="..\NLog.Web.AspNetCore\LayoutRenderers\AspNetRequestIpLayoutRenderer.cs" />
     <Compile Include="..\NLog.Web.AspNetCore\LayoutRenderers\AspNetSessionIdLayoutRenderer.cs" />
     <Compile Include="..\NLog.Web.AspNetCore\LayoutRenderers\AspNetSessionValueLayoutRenderer.cs" />
+    <Compile Include="..\NLog.Web.AspNetCore\LayoutRenderers\AspNetTraceIdentifierLayoutRenderer.cs">
+      <Link>AspNetTraceIdentifierLayoutRenderer.cs</Link>
+    </Compile>
     <Compile Include="..\NLog.Web.AspNetCore\LayoutRenderers\AspNetUserAuthTypeLayoutRenderer.cs" />
     <Compile Include="..\NLog.Web.AspNetCore\LayoutRenderers\AspNetUserIdentityLayoutRenderer.cs" />
     <Compile Include="..\NLog.Web.AspNetCore\LayoutRenderers\AspNetWebRootPathLayoutRenderer.cs" />