@sentry/replay 10.60.0 → 10.62.0

package/build/npm/cjs/index.js

@@ -5869,6 +5869,26 @@ function maybeApplyCallback(event, callback) {
   return event;
 }
 
+const MAX_TRACE_IDS = 100;
+function addTraceIdToContext(replay, traceId) {
+  const replayContext = replay.getContext();
+  if (replayContext.traceIds.size < MAX_TRACE_IDS) {
+    replayContext.traceIds.add(traceId);
+  }
+}
+
+function handleAfterSegmentSpanEnd(replay) {
+  return (segmentSpan) => {
+    if (!replay.isEnabled() || !core.spanIsSampled(segmentSpan)) {
+      return;
+    }
+    const traceId = segmentSpan.spanContext().traceId;
+    if (traceId) {
+      addTraceIdToContext(replay, traceId);
+    }
+  };
+}
+
 function isErrorEvent(event) {
   return !event.type;
 }
@@ -5899,9 +5919,9 @@ function handleAfterSendEvent(replay) {
   };
 }
 function handleTransactionEvent(replay, event) {
-  const replayContext = replay.getContext();
-  if (event.contexts?.trace?.trace_id && replayContext.traceIds.size < 100) {
-    replayContext.traceIds.add(event.contexts.trace.trace_id);
+  const traceId = event.contexts?.trace?.trace_id;
+  if (traceId) {
+    addTraceIdToContext(replay, traceId);
   }
 }
 function handleErrorEvent(replay, event) {
@@ -6761,6 +6781,7 @@ function addGlobalListeners(replay) {
         }
       }
     });
+    client.on("afterSegmentSpanEnd", handleAfterSegmentSpanEnd(replay));
     client.on("spanStart", (span) => {
       replay.lastActiveSpan = span;
     });