From 4d6fa05c83e2b1c79123ba837de726ffe38a1986 Mon Sep 17 00:00:00 2001
From: Holly Sun <jiuyu@google.com>
Date: Wed, 26 Oct 2022 21:32:21 -0700
Subject: [PATCH] Log sub event type to latency logger.

Bug: 255355348
Test: build
Change-Id: Idac85b5e31e2abdd98f6fd1bed4ec687d8bb5cd5
---
 .../quickstep/logging/StatsLogCompatManager.java       | 10 +++++++++-
 src/com/android/launcher3/logging/StatsLogManager.java |  8 +++++++-
 2 files changed, 16 insertions(+), 2 deletions(-)

diff --git a/quickstep/src/com/android/quickstep/logging/StatsLogCompatManager.java b/quickstep/src/com/android/quickstep/logging/StatsLogCompatManager.java
index 791f93b63e..dac5a31006 100644
--- a/quickstep/src/com/android/quickstep/logging/StatsLogCompatManager.java
+++ b/quickstep/src/com/android/quickstep/logging/StatsLogCompatManager.java
@@ -473,6 +473,7 @@ public class StatsLogCompatManager extends StatsLogManager {
         private int mPackageId = 0;
         private long mLatencyInMillis;
         private int mQueryLength = -1;
+        private int mSubEventType = 0;
 
         StatsCompatLatencyLogger(Context context, ActivityContext activityContext) {
             mContext = context;
@@ -509,6 +510,12 @@ public class StatsLogCompatManager extends StatsLogManager {
             return this;
         }
 
+        @Override
+        public StatsLatencyLogger withSubEventType(int type) {
+            this.mSubEventType = type;
+            return this;
+        }
+
         @Override
         public void log(EventEnum event) {
             if (IS_VERBOSE) {
@@ -526,7 +533,8 @@ public class StatsLogCompatManager extends StatsLogManager {
                     mPackageId, // package_id
                     mLatencyInMillis, // latency_in_millis
                     mType.getId(), //type
-                    mQueryLength // query_length
+                    mQueryLength, // query_length
+                    mSubEventType // sub_event_type
             );
         }
     }
diff --git a/src/com/android/launcher3/logging/StatsLogManager.java b/src/com/android/launcher3/logging/StatsLogManager.java
index 05f53fd209..bd01d51370 100644
--- a/src/com/android/launcher3/logging/StatsLogManager.java
+++ b/src/com/android/launcher3/logging/StatsLogManager.java
@@ -781,7 +781,6 @@ public class StatsLogManager implements ResourceBasedOverride {
             public int getId() {
                 return mId;
             }
-
         }
 
         /**
@@ -813,6 +812,13 @@ public class StatsLogManager implements ResourceBasedOverride {
             return this;
         }
 
+        /**
+         * Sets sub event type.
+         */
+        default StatsLatencyLogger withSubEventType(int type) {
+            return this;
+        }
+
         /**
          * Sets packageId of log message.
          */