Benchmark Events by ID

Signal/src/ViewControllers/ConversationView/ConversationViewController.m

@@ -3834,6 +3834,7 @@ typedef enum : NSUInteger {
 
 - (void)sendButtonPressed
 {
+    [BenchManager startEventWithTitle:@"Send message" eventId:@"message-send"];
     [self tryToSendTextMessage:self.inputToolbar.messageText updateKeyboardState:YES];
 }
 
@@ -4539,6 +4540,7 @@ typedef enum : NSUInteger {
                              if (scrollToBottom) {
                                  [self scrollToBottomAnimated:NO];
                              }
+                             [BenchManager completeEventWithEventId:@"message-send"];
                          }];
         }
     } @catch (NSException *exception) {

SignalMessaging/utils/Bench.swift

@@ -28,3 +28,69 @@ public func Bench(title: String, block: () -> Void) {
         finish()
     }
 }
+
+/// When it's not convenient to retain the event completion handler, e.g. when the measured event
+/// crosses multiple classes, you can use the BenchEvent tools
+///
+///     // in one class
+///     let message = getMessage()
+///     BenchEventStart(title: "message sending", eventId: message.id)
+///
+/// ...
+///
+///    // in another class
+///    BenchEventComplete(title: "message sending", eventId: message.id)
+///
+/// Or in objc
+///
+///    [BenchManager startEventWithTitle:"message sending" eventId:message.id]
+///    ...
+///    [BenchManager startEventWithTitle:"message sending" eventId:message.id]
+public func BenchEventStart(title: String, eventId: BenchmarkEventId) {
+    BenchAsync(title: title) { finish in
+        runningEvents[eventId] = Event(title: title, eventId: eventId, completion: finish)
+    }
+}
+
+public func BenchEventComplete(eventId: BenchmarkEventId) {
+    guard let event = runningEvents.removeValue(forKey: eventId) else {
+        Logger.debug("no active event with id: \(eventId)")
+        return
+    }
+
+    event.completion()
+}
+
+public typealias BenchmarkEventId = String
+
+private struct Event {
+    let title: String
+    let eventId: BenchmarkEventId
+    let completion: () -> Void
+}
+
+private var runningEvents: [BenchmarkEventId: Event] = [:]
+
+@objc
+public class BenchManager: NSObject {
+
+    @objc
+    public class func startEvent(title: String, eventId: BenchmarkEventId) {
+        BenchEventStart(title: title, eventId: eventId)
+    }
+
+    @objc
+    public class func completeEvent(eventId: BenchmarkEventId) {
+        BenchEventComplete(eventId: eventId)
+    }
+
+    @objc
+    public class func benchAsync(title: String, block: (@escaping () -> Void) -> Void) {
+        BenchAsync(title: title, block: block)
+    }
+
+    @objc
+    public class func bench(title: String, block: () -> Void) {
+        Bench(title: title, block: block)
+    }
+}