Index: services/resource_coordinator/coordination_unit/coordination_unit_manager.h |
diff --git a/services/resource_coordinator/coordination_unit/coordination_unit_manager.h b/services/resource_coordinator/coordination_unit/coordination_unit_manager.h |
index d5588704e075e40f40b27611bffe2c8dd5118e78..2f3bcba58c52126ac536cb1cf52b88bceec23379 100644 |
--- a/services/resource_coordinator/coordination_unit/coordination_unit_manager.h |
+++ b/services/resource_coordinator/coordination_unit/coordination_unit_manager.h |
@@ -11,6 +11,11 @@ |
#include "base/macros.h" |
#include "services/service_manager/public/cpp/binder_registry.h" |
+namespace ukm { |
+class MojoUkmRecorder; |
+class UkmEntryBuilder; |
+} // namespace ukm |
+ |
namespace service_manager { |
class ServiceContextRefFactory; |
} // service_manager |
@@ -29,6 +34,10 @@ class CoordinationUnitManager { |
CoordinationUnitManager(); |
~CoordinationUnitManager(); |
+ void set_ukm_recorder(ukm::MojoUkmRecorder* ukm_recorder) { |
+ ukm_recorder_ = ukm_recorder; |
+ } |
+ |
void OnStart(service_manager::BinderRegistry* registry, |
service_manager::ServiceContextRefFactory* service_ref_factory); |
void RegisterObserver( |
@@ -37,6 +46,9 @@ class CoordinationUnitManager { |
void OnCoordinationUnitWillBeDestroyed( |
CoordinationUnitImpl* coordination_unit); |
+ std::unique_ptr<ukm::UkmEntryBuilder> CreateUkmEntryBuilder( |
+ const char* event_name); |
+ |
std::vector<std::unique_ptr<CoordinationUnitGraphObserver>>& |
observers_for_testing() { |
return observers_; |
@@ -44,6 +56,7 @@ class CoordinationUnitManager { |
private: |
std::vector<std::unique_ptr<CoordinationUnitGraphObserver>> observers_; |
+ ukm::MojoUkmRecorder* ukm_recorder_ = nullptr; |
static void Create( |
service_manager::ServiceContextRefFactory* service_ref_factory); |