From 50c49447ccdd6538fa2a72958c9c6c30f948fba8 Mon Sep 17 00:00:00 2001
From: Becky Qiu <xuqiu@google.com>
Date: Fri, 4 Mar 2022 13:57:17 -0800
Subject: [PATCH] [OneSearch] Add packageId as an int for latency logging.

Enum defined in NexusLauncher.

Bug: 210790418
Test: manual, see log result in the NL cl.
Change-Id: I03a2a9831fc4cf34f3160f098739a7afd2900ec2
---
 .../android/quickstep/logging/StatsLogCompatManager.java | 9 ++++++++-
 src/com/android/launcher3/logging/StatsLogManager.java   | 6 ++++++
 2 files changed, 14 insertions(+), 1 deletion(-)

diff --git a/quickstep/src/com/android/quickstep/logging/StatsLogCompatManager.java b/quickstep/src/com/android/quickstep/logging/StatsLogCompatManager.java
index 44c0c4dfaa..d85515aef0 100644
--- a/quickstep/src/com/android/quickstep/logging/StatsLogCompatManager.java
+++ b/quickstep/src/com/android/quickstep/logging/StatsLogCompatManager.java
@@ -442,6 +442,7 @@ public class StatsLogCompatManager extends StatsLogManager {
         private final Optional<ActivityContext> mActivityContext;
         private InstanceId mInstanceId = DEFAULT_INSTANCE_ID;
         private LatencyType mType = LatencyType.UNKNOWN;
+        private int mPackageId = 0;
         private long mLatencyInMillis;
 
         StatsCompatLatencyLogger(Context context, ActivityContext activityContext) {
@@ -461,6 +462,12 @@ public class StatsLogCompatManager extends StatsLogManager {
             return this;
         }
 
+        @Override
+        public StatsLatencyLogger withPackageId(int packageId) {
+            this.mPackageId = packageId;
+            return this;
+        }
+
         @Override
         public StatsLatencyLogger withLatency(long latencyInMillis) {
             this.mLatencyInMillis = latencyInMillis;
@@ -482,7 +489,7 @@ public class StatsLogCompatManager extends StatsLogManager {
             SysUiStatsLog.write(SysUiStatsLog.LAUNCHER_LATENCY,
                     event.getId(), // event_id
                     mInstanceId.getId(), // instance_id
-                    0, // package_id
+                    mPackageId, // package_id
                     mLatencyInMillis, // latency_in_millis
                     mType.getId() //type
             );
diff --git a/src/com/android/launcher3/logging/StatsLogManager.java b/src/com/android/launcher3/logging/StatsLogManager.java
index 9604766783..0c1ba8b781 100644
--- a/src/com/android/launcher3/logging/StatsLogManager.java
+++ b/src/com/android/launcher3/logging/StatsLogManager.java
@@ -705,6 +705,12 @@ public class StatsLogManager implements ResourceBasedOverride {
             return this;
         }
 
+        /**
+         * Sets packageId of log message.
+         */
+        default StatsLatencyLogger withPackageId(int packageId) {
+            return this;
+        }
 
         /**
          * Builds the final message and logs it as {@link EventEnum}.