[GRC] Tab CPU Usage Observer

services/resource_coordinator/coordination_unit/tab_cpu_usage_observer.h

Index: services/resource_coordinator/coordination_unit/tab_cpu_usage_observer.h
diff --git a/services/resource_coordinator/coordination_unit/tab_cpu_usage_observer.h b/services/resource_coordinator/coordination_unit/tab_cpu_usage_observer.h
new file mode 100644
index 0000000000000000000000000000000000000000..bde0585b85cab213337489979c83d789c2216c7c
--- /dev/null
+++ b/services/resource_coordinator/coordination_unit/tab_cpu_usage_observer.h
@@ -0,0 +1,121 @@
+// Copyright 2017 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 SERVICES_RESOURCE_COORDINATOR_COORDINATION_UNIT_TAB_CPU_USAGE_OBSERVER_H_
+#define SERVICES_RESOURCE_COORDINATOR_COORDINATION_UNIT_TAB_CPU_USAGE_OBSERVER_H_
+
+#include <stdint.h>
+
+#include <memory>
+#include <string>
+#include <unordered_map>
+
+#include "base/macros.h"
+#include "components/ukm/public/ukm_entry_builder.h"
+#include "components/ukm/public/ukm_recorder.h"
+#include "services/resource_coordinator/coordination_unit/coordination_unit_graph_observer.h"
+#include "services/resource_coordinator/public/interfaces/coordination_unit.mojom.h"
+
+namespace resource_coordinator {
+
+class CoordinationUnitImpl;
+class TabCPUUsageObserver;
+
+class TabObserver {
+ public:
+  TabObserver();
+  ~TabObserver();
+
+  static const size_t kCPUUsageMeasurementTimeInSeconds;
+  static const size_t kMaxCPUUsageMeasurementTicks;
+
+  bool IsObserving();
+  void AttributeCPUUsage(double cpu_usage);
+  void SetTabCPUUsageObserver(TabCPUUsageObserver* tab_cpu_usage_observer) {
+    tab_cpu_usage_observer_ = tab_cpu_usage_observer;
+  }
+  void RestartObserving();
+  void UkmSourceIdChanged(ukm::SourceId source_id);
+
+ private:
+  size_t cpu_measurement_ticks_;
+  TabCPUUsageObserver* tab_cpu_usage_observer_;
+  ukm::SourceId source_id_;
+  std::unique_ptr<ukm::UkmEntryBuilder> ukm_entry_builder_;
+
+  DISALLOW_COPY_AND_ASSIGN(TabObserver);
+};
+
+class ProcessObserver {
+ public:
+  ProcessObserver();
+  ~ProcessObserver();
+
+  void CPUUsageUpdated(double cpu_usage);
+  void RecalculateAttributableTabs(
+      const CoordinationUnitImpl* process_coordination_unit,
+      const CoordinationUnitImpl* frame_coordination_unit);
+  void SetTabCPUUsageObserver(TabCPUUsageObserver* tab_cpu_usage_observer) {
+    tab_cpu_usage_observer_ = tab_cpu_usage_observer;
+  }
+
+ private:
+  std::unordered_map<CoordinationUnitID, TabObserver*> attributable_tabs_;
+  TabCPUUsageObserver* tab_cpu_usage_observer_;
+
+  DISALLOW_COPY_AND_ASSIGN(ProcessObserver);
+};
+
+class TabCPUUsageObserver : public CoordinationUnitGraphObserver {
+ public:
+  TabCPUUsageObserver();
+  ~TabCPUUsageObserver() override;
+
+  std::unordered_map<CoordinationUnitID, ProcessObserver>& process_observers() {
+    return process_observers_;
+  }
+  std::unordered_map<CoordinationUnitID, TabObserver>& tab_observers() {
+    return tab_observers_;
+  }
+
+  // Overriden from CoordinationUnitGraphobserver
+  bool ShouldObserve(const CoordinationUnitImpl* coordination_unit) override;
+  void OnCoordinationUnitCreated(
+      const CoordinationUnitImpl* coordination_unit) override;
+  void OnChildAdded(
+      const CoordinationUnitImpl* coordination_unit,
+      const CoordinationUnitImpl* child_coordination_unit) override;
+  void OnPropertyChanged(const CoordinationUnitImpl* coordination_unit,
+                         mojom::PropertyType property) override;
+  void OnChildRemoved(
+      const CoordinationUnitImpl* coordination_unit,
+      const CoordinationUnitImpl* child_coordination_unit) override;
+  void OnCoordinationUnitWillBeDestroyed(
+      const CoordinationUnitImpl* coordination_unit) override;
+
+  void TabUkmSourceIdChanged(const CoordinationUnitImpl* tab_coordination_unit);
+  void ProcessCPUUsageUpdated(
+      const CoordinationUnitImpl* process_coordination_unit);
+  void MaybeRecalculateAttributableTabsForProcess(
+      const CoordinationUnitImpl* process_coordination_unit,
+      const CoordinationUnitImpl* tab_coordination_unit);
+
+  void ProcessCoordinationUnitCreated(
+      const CoordinationUnitImpl* process_coordination_unit);
+  void TabCoordinationUnitCreated(
+      const CoordinationUnitImpl* tab_coordination_unit);
+  void RemoveProcessObserver(
+      const CoordinationUnitImpl* process_coordination_unit);
+  void RemoveTabObserver(const CoordinationUnitImpl* tab_coordination_unit);
+
+ private:
+  std::unordered_map<CoordinationUnitID, ProcessObserver> process_observers_;
+  std::unordered_map<CoordinationUnitID, TabObserver> tab_observers_;
+
+  DISALLOW_COPY_AND_ASSIGN(TabCPUUsageObserver);
+};
+
+}  // namespace resource_coordinator
+
+#endif  // SERVICES_RESOURCE_COORDINATOR_COORDINATION_UNIT_TAB_CPU_USAGE_OBSERVER_H_