[GRC] Tab CPU Usage Observer

services/resource_coordinator/coordination_unit/coordination_unit_manager.cc

Index: services/resource_coordinator/coordination_unit/coordination_unit_manager.cc
diff --git a/services/resource_coordinator/coordination_unit/coordination_unit_manager.cc b/services/resource_coordinator/coordination_unit/coordination_unit_manager.cc
index c0e3c01076b4f6bd316f7f411eee32e2dd337eaa..0fe613e1dcdcf106d6d5cacab736a402bc82856c 100644
--- a/services/resource_coordinator/coordination_unit/coordination_unit_manager.cc
+++ b/services/resource_coordinator/coordination_unit/coordination_unit_manager.cc
@@ -10,7 +10,6 @@
 #include "base/bind_helpers.h"
 #include "base/macros.h"
 #include "components/ukm/public/mojo_ukm_recorder.h"
-#include "components/ukm/public/ukm_recorder.h"
 #include "services/resource_coordinator/coordination_unit/coordination_unit_graph_observer.h"
 #include "services/resource_coordinator/coordination_unit/coordination_unit_impl.h"
 #include "services/resource_coordinator/coordination_unit/coordination_unit_provider_impl.h"
@@ -37,6 +36,7 @@ void CoordinationUnitManager::OnStart(
 
 void CoordinationUnitManager::RegisterObserver(
     std::unique_ptr<CoordinationUnitGraphObserver> observer) {
+  observer->set_coordination_unit_manager(this);
   observers_.push_back(std::move(observer));
 }
 
@@ -56,10 +56,10 @@ void CoordinationUnitManager::OnCoordinationUnitWillBeDestroyed(
 }
 
 std::unique_ptr<ukm::UkmEntryBuilder>
-CoordinationUnitManager::CreateUkmEntryBuilder(const char* event_name) {
+CoordinationUnitManager::GetUkmEntryBuilder(ukm::SourceId source_id,
+                                            const char* event_name) {
   DCHECK(ukm_recorder_ != nullptr);
-  return ukm_recorder_->GetEntryBuilder(ukm::UkmRecorder::GetNewSourceID(),
-                                        event_name);
+  return ukm_recorder_->GetEntryBuilder(source_id, event_name);
 }
 
 }  // namespace resource_coordinator