feat: Add slow and frozen frames to spans

Add total, slow and frozen frame numbers to span data. Fixes GH-3448
getsentry · Nov 27, 2023 · 5609fd3 · 5609fd3
1 parent d952dbd
commit 5609fd3
Show file tree

Hide file tree

Showing 10 changed files with 206 additions and 21 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,5 +1,11 @@
 # Changelog
 
+## Unreleased
+
+## Features
+
+- Add slow and frozen frames to spans (#3450)
+
 ## 8.16.1
 
 ### Features

diff --git a/Samples/iOS-Swift/iOS-Swift/AppDelegate.swift b/Samples/iOS-Swift/iOS-Swift/AppDelegate.swift
@@ -66,7 +66,7 @@ class AppDelegate: UIResponder, UIApplicationDelegate {
                 scope.setTag(value: "swift", key: "language")
 
                 let user = User(userId: "1")
-                user.email = "tony@example.com"
+                user.email = "philipp@example.com"
                 scope.setUser(user)
 
                 if let path = Bundle.main.path(forResource: "Tongariro", ofType: "jpg") {

diff --git a/Sources/Sentry/SentryBuildAppStartSpans.m b/Sources/Sentry/SentryBuildAppStartSpans.m
@@ -22,7 +22,7 @@
                                             origin:SentryTraceOriginAutoAppStart
                                            sampled:tracer.sampled];
 
-    return [[SentrySpan alloc] initWithTracer:tracer context:context];
+    return [[SentrySpan alloc] initWithTracer:tracer context:context framesTracker:nil];
 }
 
 NSArray<SentrySpan *> *

diff --git a/Sources/Sentry/SentryFramesTracker.m b/Sources/Sentry/SentryFramesTracker.m
@@ -242,4 +242,14 @@ - (void)dealloc
 
 @end
 
+BOOL
+sentryShouldAddSlowFrozenFramesData(
+    NSInteger totalFrames, NSInteger slowFrames, NSInteger frozenFrames)
+{
+    BOOL allBiggerThanZero = totalFrames >= 0 && slowFrames >= 0 && frozenFrames >= 0;
+    BOOL oneBiggerThanZero = totalFrames > 0 || slowFrames > 0 || frozenFrames > 0;
+
+    return allBiggerThanZero && oneBiggerThanZero;
+}
+
 #endif // SENTRY_HAS_UIKIT
diff --git a/Sources/Sentry/SentrySpan.m b/Sources/Sentry/SentrySpan.m
@@ -19,6 +19,11 @@
 #import "SentryTraceHeader.h"
 #import "SentryTracer.h"
 
+#if SENTRY_HAS_UIKIT
+#    import <SentryFramesTracker.h>
+#    import <SentryScreenFrames.h>
+#endif // SENTRY_HAS_UIKIT
+
 NS_ASSUME_NONNULL_BEGIN
 
 @interface
@@ -30,9 +35,18 @@ @implementation SentrySpan {
     NSMutableDictionary<NSString *, id> *_tags;
     NSObject *_stateLock;
     BOOL _isFinished;
+#if SENTRY_HAS_UIKIT
+    NSUInteger initTotalFrames;
+    NSUInteger initSlowFrames;
+    NSUInteger initFrozenFrames;
+    SentryFramesTracker *_framesTracker;
+#endif // SENTRY_HAS_UIKIT
 }
 
 - (instancetype)initWithContext:(SentrySpanContext *)context
+#if SENTRY_HAS_UIKIT
+                  framesTracker:(nullable SentryFramesTracker *)framesTracker;
+#endif // SENTRY_HAS_UIKIT
 {
     if (self = [super init]) {
         self.startTimestamp = [SentryDependencyContainer.sharedInstance.dateProvider date];
@@ -43,6 +57,17 @@ - (instancetype)initWithContext:(SentrySpanContext *)context
 
         if ([NSThread isMainThread]) {
             _data[SPAN_DATA_THREAD_NAME] = @"main";
+
+#if SENTRY_HAS_UIKIT
+            // Only track frames if running on main thread.
+            _framesTracker = framesTracker;
+            if (_framesTracker.isRunning) {
+                SentryScreenFrames *currentFrames = _framesTracker.currentFrames;
+                initTotalFrames = currentFrames.total;
+                initSlowFrames = currentFrames.slow;
+                initFrozenFrames = currentFrames.frozen;
+            }
+#endif // SENTRY_HAS_UIKIT
         } else {
             _data[SPAN_DATA_THREAD_NAME] = [SentryDependencyContainer.sharedInstance.threadInspector
                 getThreadName:currentThread];
@@ -64,9 +89,17 @@ - (instancetype)initWithContext:(SentrySpanContext *)context
     return self;
 }
 
-- (instancetype)initWithTracer:(SentryTracer *)tracer context:(SentrySpanContext *)context
+- (instancetype)initWithTracer:(SentryTracer *)tracer
+                       context:(SentrySpanContext *)context
+#if SENTRY_HAS_UIKIT
+                 framesTracker:(nullable SentryFramesTracker *)framesTracker
+{
+    if (self = [self initWithContext:context framesTracker:framesTracker]) {
+#else
 {
     if (self = [self initWithContext:context]) {
+#endif // SENTRY_HAS_UIKIT
+
         _tracer = tracer;
     }
     return self;
@@ -171,6 +204,27 @@ - (void)finishWithStatus:(SentrySpanStatus)status
         SENTRY_LOG_DEBUG(@"Setting span timestamp: %@ at system time %llu", self.timestamp,
             (unsigned long long)SentryDependencyContainer.sharedInstance.dateProvider.systemTime);
     }
+
+#if SENTRY_HAS_UIKIT
+    if (_framesTracker.isRunning) {
+
+        SentryScreenFrames *currentFrames = _framesTracker.currentFrames;
+        NSInteger totalFrames = currentFrames.total - initTotalFrames;
+        NSInteger slowFrames = currentFrames.slow - initSlowFrames;
+        NSInteger frozenFrames = currentFrames.frozen - initFrozenFrames;
+
+        if (sentryShouldAddSlowFrozenFramesData(totalFrames, slowFrames, frozenFrames)) {
+            [self setDataValue:@(totalFrames) forKey:@"frames.total"];
+            [self setDataValue:@(slowFrames) forKey:@"frames.slow"];
+            [self setDataValue:@(frozenFrames) forKey:@"frames.frozen"];
+
+            SENTRY_LOG_DEBUG(@"Frames for span \"%@\" Total:%ld Slow:%ld Frozen:%ld",
+                self.operation, (long)totalFrames, (long)slowFrames, (long)frozenFrames);
+        }
+    }
+
+#endif // SENTRY_HAS_UIKIT
+
     if (self.tracer == nil) {
         SENTRY_LOG_DEBUG(
             @"No tracer associated with span with id %@", self.spanId.sentrySpanIdString);

diff --git a/Sources/Sentry/SentryTracer.m b/Sources/Sentry/SentryTracer.m
@@ -123,7 +123,11 @@ - (instancetype)initWithTransactionContext:(SentryTransactionContext *)transacti
                                        hub:(nullable SentryHub *)hub
                              configuration:(SentryTracerConfiguration *)configuration;
 {
-    if (!(self = [super initWithContext:transactionContext])) {
+    if (!(self = [super initWithContext:transactionContext
+#if SENTRY_HAS_UIKIT
+                          framesTracker:SentryDependencyContainer.sharedInstance.framesTracker
+#endif // SENTRY_HAS_UIKIT
+    ])) {
         return nil;
     }
 
@@ -351,7 +355,13 @@ - (void)cancelDeadlineTimer
                                    spanDescription:description
                                            sampled:self.sampled];
 
-    SentrySpan *child = [[SentrySpan alloc] initWithTracer:self context:context];
+    SentrySpan *child =
+        [[SentrySpan alloc] initWithTracer:self
+                                   context:context
+#if SENTRY_HAS_UIKIT
+                             framesTracker:SentryDependencyContainer.sharedInstance.framesTracker
+#endif // SENTRY_HAS_UIKIT
+    ];
     child.startTimestamp = [SentryDependencyContainer.sharedInstance.dateProvider date];
     SENTRY_LOG_DEBUG(@"Started child span %@ under %@", child.spanId.sentrySpanIdString,
         parentId.sentrySpanIdString);
@@ -741,10 +751,7 @@ - (void)addMeasurements:(SentryTransaction *)transaction
         NSInteger slowFrames = currentFrames.slow - initSlowFrames;
         NSInteger frozenFrames = currentFrames.frozen - initFrozenFrames;
 
-        BOOL allBiggerThanZero = totalFrames >= 0 && slowFrames >= 0 && frozenFrames >= 0;
-        BOOL oneBiggerThanZero = totalFrames > 0 || slowFrames > 0 || frozenFrames > 0;
-
-        if (allBiggerThanZero && oneBiggerThanZero) {
+        if (sentryShouldAddSlowFrozenFramesData(totalFrames, slowFrames, frozenFrames)) {
             [self setMeasurement:@"frames_total" value:@(totalFrames)];
             [self setMeasurement:@"frames_slow" value:@(slowFrames)];
             [self setMeasurement:@"frames_frozen" value:@(frozenFrames)];

diff --git a/Sources/Sentry/include/SentryFramesTracker.h b/Sources/Sentry/include/SentryFramesTracker.h
@@ -40,6 +40,9 @@ NS_ASSUME_NONNULL_BEGIN
 
 @end
 
+BOOL sentryShouldAddSlowFrozenFramesData(
+    NSInteger totalFrames, NSInteger slowFrames, NSInteger frozenFrames);
+
 NS_ASSUME_NONNULL_END
 
 #endif // SENTRY_HAS_UIKIT
diff --git a/Sources/Sentry/include/SentrySpan.h b/Sources/Sentry/include/SentrySpan.h
@@ -6,6 +6,10 @@ NS_ASSUME_NONNULL_BEGIN
 @class SentryTracer, SentryId, SentrySpanId, SentryFrame, SentrySpanContext;
 @protocol SentrySerializable;
 
+#if SENTRY_HAS_UIKIT
+@class SentryFramesTracker;
+#endif // SENTRY_HAS_UIKIT
+
 @interface SentrySpan : NSObject <SentrySpan, SentrySerializable>
 SENTRY_NO_INIT
 
@@ -85,13 +89,22 @@ SENTRY_NO_INIT
  * @param transaction The @c SentryTracer managing the transaction this span is associated with.
  * @param context This span context information.
  */
-- (instancetype)initWithTracer:(SentryTracer *)transaction context:(SentrySpanContext *)context;
+- (instancetype)initWithTracer:(SentryTracer *)transaction
+                       context:(SentrySpanContext *)context
+#if SENTRY_HAS_UIKIT
+                 framesTracker:(nullable SentryFramesTracker *)framesTracker;
+#endif // SENTRY_HAS_UIKIT
+;
 
 /**
  * Init a @c SentrySpan with given context.
  * @param context This span context information.
  */
-- (instancetype)initWithContext:(SentrySpanContext *)context;
+- (instancetype)initWithContext:(SentrySpanContext *)context
+#if SENTRY_HAS_UIKIT
+                  framesTracker:(nullable SentryFramesTracker *)framesTracker;
+#endif // SENTRY_HAS_UIKIT
+;
 
 - (void)setExtraValue:(nullable id)value forKey:(NSString *)key DEPRECATED_ATTRIBUTE;
 @end

diff --git a/Tests/SentryTests/SentryClientTests.swift b/Tests/SentryTests/SentryClientTests.swift
@@ -348,7 +348,7 @@ class SentryClientTest: XCTestCase {
     func testCaptureTransactionWithoutScreen() {
         SentryDependencyContainer.sharedInstance().application = TestSentryUIApplication()
 
-        let event = Transaction(trace: SentryTracer(context: SpanContext(operation: "test")), children: [])
+        let event = Transaction(trace: SentryTracer(context: SpanContext(operation: "test"), framesTracker: nil), children: [])
         fixture.getSut().capture(event: event, scope: fixture.scope)
 
         try? assertLastSentEventWithAttachment { event in