From ab9ecff4d4cd995e647c54e3dbbf0d77e82d1c9d Mon Sep 17 00:00:00 2001
From: Greyson Parrelli <greyson@signal.org>
Date: Wed, 1 May 2024 11:14:28 -0400
Subject: [PATCH] Improve timing of query methods.

---
 .../org/thoughtcrime/securesms/database/SQLiteDatabase.java   | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/app/src/main/java/org/thoughtcrime/securesms/database/SQLiteDatabase.java b/app/src/main/java/org/thoughtcrime/securesms/database/SQLiteDatabase.java
index 9759ac5735..ede32a3919 100644
--- a/app/src/main/java/org/thoughtcrime/securesms/database/SQLiteDatabase.java
+++ b/app/src/main/java/org/thoughtcrime/securesms/database/SQLiteDatabase.java
@@ -110,6 +110,10 @@ public class SQLiteDatabase implements SupportSQLiteDatabase {
 
     tracer.start(methodName, params);
     E result = returnable.run();
+    if (result instanceof Cursor) {
+      // Triggers filling the window (which is about to be done anyway), but lets us capture that time inside the trace
+      ((Cursor) result).getCount();
+    }
     tracer.end(methodName);
 
     if (locked) {