feat: Implement TryGetW3CTraceHeader in Middlewares and Extensions

Author: hangy

src/Sentry.AspNetCore/Extensions/HttpContextExtensions.cs

@@ -63,6 +63,27 @@ internal static class HttpContextExtensions
             options?.LogError(ex, "Invalid Sentry trace header '{0}'.", value);
             return null;
         }
+    }    
+
+    public static W3CTraceHeader? TryGetW3CTraceHeader(this HttpContext context, SentryOptions? options)
+    {
+        var value = context.Request.Headers.GetValueOrDefault(W3CTraceHeader.HttpHeaderName);
+        if (string.IsNullOrWhiteSpace(value))
+        {
+            return null;
+        }
+
+        options?.LogDebug("Received Sentry trace header '{0}'.", value);
+
+        try
+        {
+            return W3CTraceHeader.Parse(value!);
+        }
+        catch (Exception ex)
+        {
+            options?.LogError(ex, "Invalid Sentry trace header '{0}'.", value);
+            return null;
+        }
     }
 
     public static BaggageHeader? TryGetBaggageHeader(this HttpContext context, SentryOptions? options)

src/Sentry.AspNetCore/SentryMiddleware.cs

@@ -106,6 +106,7 @@ public async Task InvokeAsync(HttpContext context, RequestDelegate next)
             }
 
             var traceHeader = context.TryGetSentryTraceHeader(_options);
+            traceHeader ??= context.TryGetW3CTraceHeader(_options)?.SentryTraceHeader;
             var baggageHeader = context.TryGetBaggageHeader(_options);
             var transactionContext = hub.ContinueTrace(traceHeader, baggageHeader);
 

src/Sentry.Azure.Functions.Worker/HttpRequestDataExtensions.cs

@@ -30,6 +30,31 @@ internal static class HttpRequestDataExtensions
         }
     }
 
+    public static W3CTraceHeader? TryGetW3CTraceHeader(this HttpRequestData context, IDiagnosticLogger? logger)
+    {
+        var traceHeaderValue = context.Headers.TryGetValues(W3CTraceHeader.HttpHeaderName, out var values)
+            ? values.FirstOrDefault()
+            : null;
+
+        if (traceHeaderValue is null)
+        {
+            logger?.LogDebug("Did not receive a Sentry trace header.");
+            return null;
+        }
+
+        logger?.LogDebug("Received Sentry trace header '{0}'.", traceHeaderValue);
+
+        try
+        {
+            return W3CTraceHeader.Parse(traceHeaderValue);
+        }
+        catch (Exception ex)
+        {
+            logger?.LogError(ex, "Invalid Sentry trace header '{0}'.", traceHeaderValue);
+            return null;
+        }
+    }
+
     public static BaggageHeader? TryGetBaggageHeader(this HttpRequestData context, IDiagnosticLogger? logger)
     {
         var baggageValue = context.Headers.TryGetValues(BaggageHeader.HttpHeaderName, out var value)

src/Sentry.Azure.Functions.Worker/SentryFunctionsWorkerMiddleware.cs

@@ -124,6 +124,7 @@ private async Task<TransactionContext> StartOrContinueTraceAsync(FunctionContext
         }
 
         var traceHeader = requestData.TryGetSentryTraceHeader(_logger);
+        traceHeader ??= requestData.TryGetW3CTraceHeader(_logger)?.SentryTraceHeader;
         var baggageHeader = requestData.TryGetBaggageHeader(_logger);
 
         return SentrySdk.ContinueTrace(traceHeader, baggageHeader, transactionName, Operation);