feat: Associate replays with errors and traces on Android

CHANGELOG.md

@@ -1,5 +1,11 @@
 # Changelog
 
+## Unreleased
+
+### Features
+
+- Associate replays with errors and traces on Android ([#4133](https://github.com/getsentry/sentry-dotnet/pull/4133))
+
 ## 5.7.0-beta.0
 
 ### Features

src/Sentry.AspNet/HttpContextExtensions.cs

@@ -1,4 +1,5 @@
 using Sentry.Extensibility;
+using Sentry.Internal;
 using Sentry.Protocol;
 
 namespace Sentry.AspNet;
@@ -125,8 +126,9 @@ public static ITransactionTracer StartSentryTransaction(this HttpContext httpCon
             ["__HttpContext"] = httpContext,
         };
 
-        // Set the Dynamic Sampling Context from the baggage header, if it exists.
-        var dynamicSamplingContext = baggageHeader?.CreateDynamicSamplingContext();
+        // Set the Dynamic Sampling Context from the baggage header, if it exists
+        // Note: We don't record Session Replays in ASP.NET
+        var dynamicSamplingContext = baggageHeader?.CreateDynamicSamplingContext(ReplaySession.DisabledInstance);
 
         if (traceHeader is not null && baggageHeader is null)
         {

src/Sentry.AspNetCore/SentryTracingMiddleware.cs

@@ -2,6 +2,7 @@
 using Microsoft.Extensions.Options;
 using Sentry.AspNetCore.Extensions;
 using Sentry.Extensibility;
+using Sentry.Internal;
 using Sentry.Internal.OpenTelemetry;
 
 namespace Sentry.AspNetCore;
@@ -65,7 +66,8 @@ public SentryTracingMiddleware(
             var baggageHeader = context.Items.TryGetValue(SentryMiddleware.BaggageHeaderItemKey, out var baggageHeaderObject)
                 ? baggageHeaderObject as BaggageHeader : null;
 
-            var dynamicSamplingContext = baggageHeader?.CreateDynamicSamplingContext();
+            // Note: We don't record Session Replays in ASP.NET core
+            var dynamicSamplingContext = baggageHeader?.CreateDynamicSamplingContext(ReplaySession.DisabledInstance);
 
             if (traceHeader is not null && baggageHeader is null)
             {

src/Sentry.OpenTelemetry/SentrySpanProcessor.cs

@@ -13,6 +13,7 @@ public class SentrySpanProcessor : BaseProcessor<Activity>
 {
     private readonly IHub _hub;
     internal readonly IEnumerable<IOpenTelemetryEnricher> _enrichers;
+    private readonly IReplaySession _replaySession;
     internal const string OpenTelemetryOrigin = "auto.otel";
 
     // ReSharper disable once MemberCanBePrivate.Global - Used by tests
@@ -38,7 +39,7 @@ public SentrySpanProcessor(IHub hub) : this(hub, null)
     {
     }
 
-    internal SentrySpanProcessor(IHub hub, IEnumerable<IOpenTelemetryEnricher>? enrichers)
+    internal SentrySpanProcessor(IHub hub, IEnumerable<IOpenTelemetryEnricher>? enrichers, IReplaySession? replaySession = null)
     {
         _hub = hub;
         _realHub = new Lazy<Hub?>(() =>
@@ -57,7 +58,8 @@ internal SentrySpanProcessor(IHub hub, IEnumerable<IOpenTelemetryEnricher>? enri
                 "You should use the TracerProviderBuilderExtensions to configure Sentry with OpenTelemetry");
         }
 
-        _enrichers = enrichers ?? Enumerable.Empty<IOpenTelemetryEnricher>();
+        _enrichers = enrichers ?? [];
+        _replaySession = replaySession ?? ReplaySession.Instance;
         _options = hub.GetSentryOptions();
 
         if (_options is null)
@@ -158,7 +160,7 @@ private void CreateRootSpan(Activity data)
         };
 
         var baggageHeader = data.Baggage.AsBaggageHeader();
-        var dynamicSamplingContext = baggageHeader.CreateDynamicSamplingContext();
+        var dynamicSamplingContext = baggageHeader.CreateDynamicSamplingContext(_replaySession);
         var transaction = (TransactionTracer)_hub.StartTransaction(
             transactionContext, new Dictionary<string, object?>(), dynamicSamplingContext
         );

src/Sentry/DynamicSamplingContext.cs

@@ -21,14 +21,16 @@ internal class DynamicSamplingContext
     public static readonly DynamicSamplingContext Empty = new(new Dictionary<string, string>().AsReadOnly());
 
     private DynamicSamplingContext(
+        IReplaySession replaySession,
         SentryId traceId,
         string publicKey,
         bool? sampled,
         double? sampleRate = null,
         double? sampleRand = null,
         string? release = null,
         string? environment = null,
-        string? transactionName = null)
+        string? transactionName = null
+        )
     {
         // Validate and set required values
         if (traceId == SentryId.Empty)
@@ -51,7 +53,7 @@ private DynamicSamplingContext(
             throw new ArgumentOutOfRangeException(nameof(sampleRand), "Arg invalid if < 0.0 or >= 1.0");
         }
 
-        var items = new Dictionary<string, string>(capacity: 8)
+        var items = new Dictionary<string, string>(capacity: 9)
         {
             ["trace_id"] = traceId.ToString(),
             ["public_key"] = publicKey,
@@ -88,12 +90,29 @@ private DynamicSamplingContext(
             items.Add("transaction", transactionName);
         }
 
+        if (replaySession.ActiveReplayId is { } replayId && replayId != SentryId.Empty)
+        {
+            items.Add("replay_id", replayId.ToString());
+        }
+
         Items = items;
     }
 
     public BaggageHeader ToBaggageHeader() => BaggageHeader.Create(Items, useSentryPrefix: true);
 
-    public static DynamicSamplingContext? CreateFromBaggageHeader(BaggageHeader baggage)
+    public DynamicSamplingContext WithReplayId(IReplaySession replaySession)
+    {
+        if (replaySession.ActiveReplayId is not { } replayId || replayId == SentryId.Empty)
+        {
+            return this;
+        }
+
+        var items = Items.ToDictionary(kvp => kvp.Key, kvp => kvp.Value);
+        items["replay_id"] = replayId.ToString();
+        return new DynamicSamplingContext(items);
+    }
+
+    public static DynamicSamplingContext? CreateFromBaggageHeader(BaggageHeader baggage, IReplaySession replaySession)
     {
         var items = baggage.GetSentryMembers();
 
@@ -144,10 +163,18 @@ private DynamicSamplingContext(
             }
             items.Add("sample_rand", rand.ToString("N4", CultureInfo.InvariantCulture));
         }
+
+        if (replaySession.ActiveReplayId is { } replayId)
+        {
+            // Overwrite any existing value - the DSC is simply used as a transport mechanism so that SDKs can
+            // communicate the replayId to Sentry Relay (SDKs don't need to propagate the replayId to each other).
+            items["replay_id"] = replayId.ToString();
+        }
+
         return new DynamicSamplingContext(items);
     }
 
-    public static DynamicSamplingContext CreateFromTransaction(TransactionTracer transaction, SentryOptions options)
+    public static DynamicSamplingContext CreateFromTransaction(TransactionTracer transaction, SentryOptions options, IReplaySession replaySession)
     {
         // These should already be set on the transaction.
         var publicKey = options.ParsedDsn.PublicKey;
@@ -162,6 +189,7 @@ public static DynamicSamplingContext CreateFromTransaction(TransactionTracer tra
         var environment = options.SettingLocator.GetEnvironment();
 
         return new DynamicSamplingContext(
+            replaySession,
             traceId,
             publicKey,
             sampled,
@@ -172,14 +200,15 @@ public static DynamicSamplingContext CreateFromTransaction(TransactionTracer tra
             transactionName);
     }
 
-    public static DynamicSamplingContext CreateFromPropagationContext(SentryPropagationContext propagationContext, SentryOptions options)
+    public static DynamicSamplingContext CreateFromPropagationContext(SentryPropagationContext propagationContext, SentryOptions options, IReplaySession replaySession)
     {
         var traceId = propagationContext.TraceId;
         var publicKey = options.ParsedDsn.PublicKey;
         var release = options.SettingLocator.GetRelease();
         var environment = options.SettingLocator.GetEnvironment();
 
         return new DynamicSamplingContext(
+            replaySession,
             traceId,
             publicKey,
             null,
@@ -190,12 +219,12 @@ public static DynamicSamplingContext CreateFromPropagationContext(SentryPropagat
 
 internal static class DynamicSamplingContextExtensions
 {
-    public static DynamicSamplingContext? CreateDynamicSamplingContext(this BaggageHeader baggage)
-        => DynamicSamplingContext.CreateFromBaggageHeader(baggage);
+    public static DynamicSamplingContext? CreateDynamicSamplingContext(this BaggageHeader baggage, IReplaySession replaySession)
+        => DynamicSamplingContext.CreateFromBaggageHeader(baggage, replaySession);
 
-    public static DynamicSamplingContext CreateDynamicSamplingContext(this TransactionTracer transaction, SentryOptions options)
-        => DynamicSamplingContext.CreateFromTransaction(transaction, options);
+    public static DynamicSamplingContext CreateDynamicSamplingContext(this TransactionTracer transaction, SentryOptions options, IReplaySession replaySession)
+        => DynamicSamplingContext.CreateFromTransaction(transaction, options, replaySession);
 
-    public static DynamicSamplingContext CreateDynamicSamplingContext(this SentryPropagationContext propagationContext, SentryOptions options)
-        => DynamicSamplingContext.CreateFromPropagationContext(propagationContext, options);
+    public static DynamicSamplingContext CreateDynamicSamplingContext(this SentryPropagationContext propagationContext, SentryOptions options, IReplaySession replaySession)
+        => DynamicSamplingContext.CreateFromPropagationContext(propagationContext, options, replaySession);
 }

src/Sentry/Internal/Hub.cs

@@ -13,6 +13,7 @@ internal class Hub : IHub, IDisposable
     private readonly ISessionManager _sessionManager;
     private readonly SentryOptions _options;
     private readonly RandomValuesFactory _randomValuesFactory;
+    private readonly IReplaySession _replaySession;
 
 #if MEMORY_DUMP_SUPPORTED
     private readonly MemoryMonitor? _memoryMonitor;
@@ -39,7 +40,8 @@ internal Hub(
         ISessionManager? sessionManager = null,
         ISystemClock? clock = null,
         IInternalScopeManager? scopeManager = null,
-        RandomValuesFactory? randomValuesFactory = null)
+        RandomValuesFactory? randomValuesFactory = null,
+        IReplaySession? replaySession = null)
     {
         if (string.IsNullOrWhiteSpace(options.Dsn))
         {
@@ -55,7 +57,7 @@ internal Hub(
         _sessionManager = sessionManager ?? new GlobalSessionManager(options);
         _clock = clock ?? SystemClock.Clock;
         client ??= new SentryClient(options, randomValuesFactory: _randomValuesFactory, sessionManager: _sessionManager);
-
+        _replaySession = replaySession ?? ReplaySession.Instance;
         ScopeManager = scopeManager ?? new SentryScopeManager(options, client);
 
         if (!options.IsGlobalModeEnabled)
@@ -178,10 +180,10 @@ _options.TransactionProfilerFactory is { } profilerFactory &&
             }
         }
 
-        // Use the provided DSC, or create one based on this transaction.
+        // Use the provided DSC (adding the active replayId if necessary), or create one based on this transaction.
         // DSC creation must be done AFTER the sampling decision has been made.
-        transaction.DynamicSamplingContext =
-            dynamicSamplingContext ?? transaction.CreateDynamicSamplingContext(_options);
+        transaction.DynamicSamplingContext = dynamicSamplingContext?.WithReplayId(_replaySession)
+            ?? transaction.CreateDynamicSamplingContext(_options, _replaySession);
 
         // A sampled out transaction still appears fully functional to the user
         // but will be dropped by the client and won't reach Sentry's servers.
@@ -224,7 +226,7 @@ public BaggageHeader GetBaggage()
         }
 
         var propagationContext = CurrentScope.PropagationContext;
-        return propagationContext.GetOrCreateDynamicSamplingContext(_options).ToBaggageHeader();
+        return propagationContext.GetOrCreateDynamicSamplingContext(_options, _replaySession).ToBaggageHeader();
     }
 
     public TransactionContext ContinueTrace(
@@ -254,7 +256,7 @@ public TransactionContext ContinueTrace(
         string? name = null,
         string? operation = null)
     {
-        var propagationContext = SentryPropagationContext.CreateFromHeaders(_options.DiagnosticLogger, traceHeader, baggageHeader);
+        var propagationContext = SentryPropagationContext.CreateFromHeaders(_options.DiagnosticLogger, traceHeader, baggageHeader, _replaySession);
         ConfigureScope(scope => scope.SetPropagationContext(propagationContext));
 
         return new TransactionContext(
@@ -382,7 +384,7 @@ private void ApplyTraceContextToEvent(SentryEvent evt, SentryPropagationContext
         evt.Contexts.Trace.TraceId = propagationContext.TraceId;
         evt.Contexts.Trace.SpanId = propagationContext.SpanId;
         evt.Contexts.Trace.ParentSpanId = propagationContext.ParentSpanId;
-        evt.DynamicSamplingContext = propagationContext.GetOrCreateDynamicSamplingContext(_options);
+        evt.DynamicSamplingContext = propagationContext.GetOrCreateDynamicSamplingContext(_options, _replaySession);
     }
 
     public bool CaptureEnvelope(Envelope envelope) => CurrentClient.CaptureEnvelope(envelope);
@@ -471,12 +473,9 @@ private SentryId CaptureEvent(SentryEvent evt, SentryHint? hint, Scope scope)
         {
             // We get the span linked to the event or fall back to the current span
             var span = GetLinkedSpan(evt) ?? scope.Span;
-            if (span is not null)
+            if (span is not null && span.IsSampled is not false)
             {
-                if (span.IsSampled is not false)
-                {
-                    ApplyTraceContextToEvent(evt, span);
-                }
+                ApplyTraceContextToEvent(evt, span);
             }
             else
             {

src/Sentry/Internal/ReplaySession.cs

@@ -0,0 +1,40 @@
+#if __ANDROID__
+using Sentry.Android.Extensions;
+#endif
+
+namespace Sentry.Internal;
+
+internal interface IReplaySession
+{
+    public SentryId? ActiveReplayId { get; }
+}
+
+internal class ReplaySession : IReplaySession
+{
+    public static readonly IReplaySession Instance = new ReplaySession();
+
+    internal static readonly IReplaySession DisabledInstance = new DisabledReplaySession();
+
+    private ReplaySession()
+    {
+    }
+
+    public SentryId? ActiveReplayId
+    {
+        get
+        {
+#if __ANDROID__
+            // Check to see if a Replay ID is available
+            var replayId = JavaSdk.ScopesAdapter.Instance?.Options?.ReplayController?.ReplayId?.ToSentryId();
+            return (replayId is { } id && id != SentryId.Empty) ? id : null;
+#else
+            return null;
+#endif
+        }
+    }
+
+    private class DisabledReplaySession : IReplaySession
+    {
+        public SentryId? ActiveReplayId => null;
+    }
+}