From a61a2cc962fac363de1fee9bcdf1813d53db14a6 Mon Sep 17 00:00:00 2001
From: Becky Qiu <xuqiu@google.com>
Date: Tue, 7 Jun 2022 14:17:56 -0700
Subject: [PATCH] [Toast] Add query_length for launcher latency logging.

Bug: 235249335
Test: no test needed.
Change-Id: Ia16c241a2eb3cb50a17713216182192f7902be9e
---
 .../quickstep/logging/StatsLogCompatManager.java       | 10 +++++++++-
 src/com/android/launcher3/logging/StatsLogManager.java |  7 +++++++
 2 files changed, 16 insertions(+), 1 deletion(-)

diff --git a/quickstep/src/com/android/quickstep/logging/StatsLogCompatManager.java b/quickstep/src/com/android/quickstep/logging/StatsLogCompatManager.java
index 85ef6cb16e..3b9e2b2d8b 100644
--- a/quickstep/src/com/android/quickstep/logging/StatsLogCompatManager.java
+++ b/quickstep/src/com/android/quickstep/logging/StatsLogCompatManager.java
@@ -462,6 +462,7 @@ public class StatsLogCompatManager extends StatsLogManager {
         private LatencyType mType = LatencyType.UNKNOWN;
         private int mPackageId = 0;
         private long mLatencyInMillis;
+        private int mQueryLength = -1;
 
         StatsCompatLatencyLogger(Context context, ActivityContext activityContext) {
             mContext = context;
@@ -492,6 +493,12 @@ public class StatsLogCompatManager extends StatsLogManager {
             return this;
         }
 
+        @Override
+        public StatsLatencyLogger withQueryLength(int queryLength) {
+            this.mQueryLength = queryLength;
+            return this;
+        }
+
         @Override
         public void log(EventEnum event) {
             if (IS_VERBOSE) {
@@ -508,7 +515,8 @@ public class StatsLogCompatManager extends StatsLogManager {
                     mInstanceId.getId(), // instance_id
                     mPackageId, // package_id
                     mLatencyInMillis, // latency_in_millis
-                    mType.getId() //type
+                    mType.getId(), //type
+                    mQueryLength // query_length
             );
         }
     }
diff --git a/src/com/android/launcher3/logging/StatsLogManager.java b/src/com/android/launcher3/logging/StatsLogManager.java
index 5dcd48cbb3..7d24fe814c 100644
--- a/src/com/android/launcher3/logging/StatsLogManager.java
+++ b/src/com/android/launcher3/logging/StatsLogManager.java
@@ -771,6 +771,13 @@ public class StatsLogManager implements ResourceBasedOverride {
             return this;
         }
 
+        /**
+         * Sets query length of the event.
+         */
+        default StatsLatencyLogger withQueryLength(int queryLength) {
+            return this;
+        }
+
         /**
          * Sets packageId of log message.
          */