feat: measure cpu energy usage estimates

getsentry · Aug 11, 2023 · d7ebd02 · d7ebd02
1 parent 2543034
commit d7ebd02
Show file tree

Hide file tree

Showing 4 changed files with 57 additions and 0 deletions.
diff --git a/Sources/Sentry/SentryMetricProfiler.mm b/Sources/Sentry/SentryMetricProfiler.mm
@@ -29,6 +29,7 @@ @implementation SentryMetricReading
 
 NSString *const kSentryMetricProfilerSerializationKeyMemoryFootprint = @"memory_footprint";
 NSString *const kSentryMetricProfilerSerializationKeyCPUUsageFormat = @"cpu_usage_%d";
+NSString *const kSentryMetricProfilerSerializationKeyCPUEnergyUsage = @"cpu_energy_usage";
 
 NSString *const kSentryMetricProfilerSerializationUnitBytes = @"byte";
 NSString *const kSentryMetricProfilerSerializationUnitPercentage = @"percent";
@@ -82,6 +83,9 @@ @implementation SentryMetricProfiler {
     NSMutableDictionary<NSNumber *, NSMutableArray<SentryMetricReading *> *> *_cpuUsage;
 
     NSMutableArray<SentryMetricReading *> *_memoryFootprint;
+
+    NSNumber *previousEnergyReading;
+    NSMutableArray<SentryMetricReading *> *_cpuEnergyUsage;
 }
 
 - (instancetype)init
@@ -98,6 +102,7 @@ - (instancetype)init
         }
 
         _memoryFootprint = [NSMutableArray<SentryMetricReading *> array];
+        _cpuEnergyUsage = [NSMutableArray<SentryMetricReading *> array];
     }
     return self;
 }
@@ -123,8 +128,10 @@ - (void)stop
                                                       and:(uint64_t)endSystemTime;
 {
     NSArray<SentryMetricReading *> *memoryFootprint;
+    NSArray<SentryMetricReading *> *cpuEnergyUsage;
     NSDictionary<NSNumber *, NSArray<SentryMetricReading *> *> *cpuUsage;
     @synchronized(self) {
+        cpuEnergyUsage = [NSArray<SentryMetricReading *> arrayWithArray:_cpuEnergyUsage];
         memoryFootprint = [NSArray<SentryMetricReading *> arrayWithArray:_memoryFootprint];
         cpuUsage = [NSDictionary<NSNumber *, NSArray<SentryMetricReading *> *>
             dictionaryWithDictionary:_cpuUsage];
@@ -136,6 +143,11 @@ - (void)stop
             = serializeValuesWithNormalizedTime(memoryFootprint,
                 kSentryMetricProfilerSerializationUnitBytes, startSystemTime, endSystemTime);
     }
+    if (cpuEnergyUsage.count > 0) {
+        dict[kSentryMetricProfilerSerializationKeyCPUEnergyUsage]
+            = serializeValuesWithNormalizedTime(cpuEnergyUsage,
+                kSentryMetricProfilerSerializationUnitBytes, startSystemTime, endSystemTime);
+    }
 
     [cpuUsage enumerateKeysAndObjectsUsingBlock:^(NSNumber *_Nonnull core,
         NSArray<SentryMetricReading *> *_Nonnull readings, BOOL *_Nonnull stop) {
@@ -167,6 +179,7 @@ - (void)registerSampler
               eventHandler:^{
                   [weakSelf recordCPUPercentagePerCore];
                   [weakSelf recordMemoryFootprint];
+                  [weakSelf recordEnergyUsageEstimate];
               }];
 }
 
@@ -205,6 +218,29 @@ - (void)recordCPUPercentagePerCore
     }
 }
 
+- (void)recordEnergyUsageEstimate
+{
+    NSError *error;
+    const auto reading =
+        [SentryDependencyContainer.sharedInstance.systemWrapper cpuEnergyUsageWithError:&error];
+    if (error) {
+        SENTRY_LOG_ERROR(@"Failed to read CPU energy usage: %@", error);
+        return;
+    }
+
+    if (previousEnergyReading == nil) {
+        previousEnergyReading = reading;
+        return;
+    }
+
+    const auto value = reading.unsignedIntegerValue - previousEnergyReading.unsignedIntegerValue;
+    previousEnergyReading = reading;
+
+    @synchronized(self) {
+        [_cpuEnergyUsage addObject:[self metricReadingForValue:@(value)]];
+    }
+}
+
 - (SentryMetricReading *)metricReadingForValue:(NSNumber *)value
 {
     const auto reading = [[SentryMetricReading alloc] init];

diff --git a/Sources/Sentry/SentrySystemWrapper.mm b/Sources/Sentry/SentrySystemWrapper.mm
@@ -59,4 +59,22 @@ - (SentryRAMBytes)memoryFootprintBytes:(NSError *__autoreleasing _Nullable *)err
     return result;
 }
 
+- (NSNumber *)cpuEnergyUsageWithError:(NSError **)error
+{
+    struct task_power_info_v2 powerInfo;
+
+    mach_msg_type_number_t size = TASK_POWER_INFO_V2_COUNT;
+
+    task_t task = mach_task_self();
+    kern_return_t kr = task_info(task, TASK_POWER_INFO_V2, (task_info_t)&powerInfo, &size);
+    if (kr != KERN_SUCCESS) {
+        if (error) {
+            *error = NSErrorFromSentryErrorWithKernelError(
+                kSentryErrorKernel, @"Error with task_info(…TASK_POWER_INFO_V2…).", kr);
+            ;
+        }
+    }
+    return @(powerInfo.cpu_energy.total_system + powerInfo.cpu_energy.total_user);
+}
+
 @end
diff --git a/Sources/Sentry/include/SentryMetricProfiler.h b/Sources/Sentry/include/SentryMetricProfiler.h
@@ -10,6 +10,7 @@ NS_ASSUME_NONNULL_BEGIN
 
 SENTRY_EXTERN NSString *const kSentryMetricProfilerSerializationKeyMemoryFootprint;
 SENTRY_EXTERN NSString *const kSentryMetricProfilerSerializationKeyCPUUsageFormat;
+SENTRY_EXTERN NSString *const kSentryMetricProfilerSerializationKeyCPUEnergyUsage;
 
 SENTRY_EXTERN NSString *const kSentryMetricProfilerSerializationUnitBytes;
 SENTRY_EXTERN NSString *const kSentryMetricProfilerSerializationUnitPercentage;

diff --git a/Sources/Sentry/include/SentrySystemWrapper.h b/Sources/Sentry/include/SentrySystemWrapper.h
@@ -26,6 +26,8 @@ typedef mach_vm_size_t SentryRAMBytes;
  */
 - (nullable NSArray<NSNumber *> *)cpuUsagePerCore:(NSError **)error;
 
+- (nullable NSNumber *)cpuEnergyUsageWithError:(NSError **)error;
+
 @end
 
 NS_ASSUME_NONNULL_END