From a497253688fa04606bdb66f595dff4047a7c6dae Mon Sep 17 00:00:00 2001
From: Denny cy Lee <dennycylee@google.com>
Date: Fri, 17 Apr 2020 11:56:14 +0800
Subject: [PATCH] Register Display Wake Reason Puller

Bug: 150519505
Test: adb shell cmd stats pull-source 10081
Change-Id: Ia6ee7b296b589b8c15763023db683a64129377c7
Merged-In: Ia6ee7b296b589b8c15763023db683a64129377c7
Signed-off-by: Denny cy Lee <dennycylee@google.com>
(cherry picked from commit e9dcb27f695ba0926eaac96a42ab8d0bbcd98c20)
---
 .../stats/pull/StatsPullAtomService.java       | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/services/core/java/com/android/server/stats/pull/StatsPullAtomService.java b/services/core/java/com/android/server/stats/pull/StatsPullAtomService.java
index 3b4c4235d8a45..4beab78d455ce 100644
--- a/services/core/java/com/android/server/stats/pull/StatsPullAtomService.java
+++ b/services/core/java/com/android/server/stats/pull/StatsPullAtomService.java
@@ -423,6 +423,8 @@ public class StatsPullAtomService extends SystemService {
                         return pullAttributedAppOps(atomTag, data);
                     case FrameworkStatsLog.SETTING_SNAPSHOT:
                         return pullSettingsStats(atomTag, data);
+                    case FrameworkStatsLog.DISPLAY_WAKE_REASON:
+                        return pullDisplayWakeStats(atomTag, data);
                     default:
                         throw new UnsupportedOperationException("Unknown tagId=" + atomTag);
                 }
@@ -588,6 +590,7 @@ public class StatsPullAtomService extends SystemService {
         registerBatteryVoltage();
         registerBatteryCycleCount();
         registerSettingsStats();
+        registerDisplayWakeStats();
     }
 
     /**
@@ -3295,6 +3298,21 @@ public class StatsPullAtomService extends SystemService {
         return StatsManager.PULL_SUCCESS;
     }
 
+    private void registerDisplayWakeStats() {
+        int tagId = FrameworkStatsLog.DISPLAY_WAKE_REASON;
+        mStatsManager.setPullAtomCallback(
+                tagId,
+                null, // use default PullAtomMetadata values
+                BackgroundThread.getExecutor(),
+                mStatsCallbackImpl
+        );
+    }
+
+    int pullDisplayWakeStats(int atomTag, List<StatsEvent> pulledData) {
+        //TODO: Denny, implement read/write DisplayWakeStats, b/154172964
+        return 0;
+    }
+
     // Thermal event received from vendor thermal management subsystem
     private static final class ThermalEventListener extends IThermalEventListener.Stub {
         @Override