Make ResourceReporter a client of memory coordinator

chrome/browser/chromeos/resource_reporter/resource_reporter.h

 // Copyright 2015 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #ifndef CHROME_BROWSER_CHROMEOS_RESOURCE_REPORTER_RESOURCE_REPORTER_H_
 #define CHROME_BROWSER_CHROMEOS_RESOURCE_REPORTER_RESOURCE_REPORTER_H_
 
 #include <stddef.h>
 #include <stdint.h>
 
 #include <map>
 #include <string>
 #include <vector>
 
 #include "base/gtest_prod_util.h"
 #include "base/macros.h"
+#include "base/memory/memory_coordinator_client.h"
 #include "base/memory/memory_pressure_listener.h"
 #include "base/memory/singleton.h"
 #include "base/time/time.h"
 #include "chrome/browser/task_manager/task_manager_observer.h"
 #include "components/metrics/metrics_service.h"
 #include "components/rappor/sample.h"
 
 namespace chromeos {
 
 // A system that tracks the top |kTopConsumersCount| CPU and memory consumer
 // Chrome tasks and reports a weighted random sample of them via Rappor whenever
 // memory pressure is critical. The reporting is limited to once per
 // |kMinimumTimeBetweenReportsInMS|.
-class ResourceReporter : public task_manager::TaskManagerObserver {
+class ResourceReporter : public task_manager::TaskManagerObserver,
+                         public base::MemoryCoordinatorClient {
  public:
   // A collection of the data of a task manager's task that the ResourceReporter
   // is interested in.
   struct TaskRecord {
     explicit TaskRecord(task_manager::TaskId task_id);
 
     TaskRecord(task_manager::TaskId task_id,
                const std::string& task_name,
                double cpu_percent,
                int64_t memory_bytes,
@@ skipping 100 matching lines @@
       const TaskRecord& task_record);
 
   // Gets the CPU/memory usage ranges given the |cpu| / |memory_in_bytes|
   // values.
   static CpuUsageRange GetCpuUsageRange(double cpu);
   static MemoryUsageRange GetMemoryUsageRange(int64_t memory_in_bytes);
 
   // Gets the bucket in which the current system's number of CPU cores fall.
   static CpuCoresNumberRange GetCurrentSystemCpuCoresRange();
 
+  // base::MemoryCoordinatorClient implementation:
+  void OnMemoryStateChange(base::MemoryState state) override;
+
   // Perform a weighted random sampling to select a task by its CPU or memory
   // usage weights so that we can record samples for them via Rappor.
   // They return nullptr if no TaskRecord has been selected.
   const TaskRecord* SampleTaskByCpu() const;
   const TaskRecord* SampleTaskByMemory() const;
 
   // The callback function that will be invoked on memory pressure events.
   using MemoryPressureLevel = base::MemoryPressureListener::MemoryPressureLevel;
   void OnMemoryPressure(MemoryPressureLevel memory_pressure_level);
 
+  // Records the current CPU and memory usage with UMA and Rappor.
+  void RecordCurrentState();
+
   // We'll use this to watch for memory pressure events so that we can trigger
   // Rappor sampling at at the critical memory pressure level.
   std::unique_ptr<base::MemoryPressureListener> memory_pressure_listener_;
 
   // Contains the collected data about the currently running tasks from the most
   // recent task manager refresh.
   std::map<task_manager::TaskId, std::unique_ptr<TaskRecord>> task_records_;
 
   // Contains the top |kTopConsumerCount| CPU consumer tasks sorted in a
   // descending order by their CPU usage.
@@ skipping 28 matching lines @@
 
   // True after the first task manager OnTasksRefreshed() event is received.
   bool have_seen_first_task_manager_refresh_ = false;
 
   DISALLOW_COPY_AND_ASSIGN(ResourceReporter);
 };
 
 }  // namespace chromeos
 
 #endif  // CHROME_BROWSER_CHROMEOS_RESOURCE_REPORTER_RESOURCE_REPORTER_H_