Index: components/scheduler/child/scheduler_helper.cc |
diff --git a/components/scheduler/child/scheduler_helper.cc b/components/scheduler/child/scheduler_helper.cc |
index f39da7927192fe67c633abb3c192ebae476b2133..8c0fb1b82e745c3e4c987bbeb1d881a7ac88d2a7 100644 |
--- a/components/scheduler/child/scheduler_helper.cc |
+++ b/components/scheduler/child/scheduler_helper.cc |
@@ -24,18 +24,17 @@ SchedulerHelper::SchedulerHelper( |
disabled_by_default_verbose_tracing_category)), |
control_task_runner_(NewTaskQueue( |
TaskQueue::Spec("control_tq") |
- .SetWakeupPolicy( |
- TaskQueue::WakeupPolicy::DONT_WAKE_OTHER_QUEUES) |
+ .SetWakeupPolicy(TaskQueue::WakeupPolicy::DONT_WAKE_OTHER_QUEUES) |
.SetShouldNotifyObservers(false))), |
control_after_wakeup_task_runner_(NewTaskQueue( |
TaskQueue::Spec("control_after_wakeup_tq") |
.SetPumpPolicy(TaskQueue::PumpPolicy::AFTER_WAKEUP) |
- .SetWakeupPolicy( |
- TaskQueue::WakeupPolicy::DONT_WAKE_OTHER_QUEUES) |
+ .SetWakeupPolicy(TaskQueue::WakeupPolicy::DONT_WAKE_OTHER_QUEUES) |
.SetShouldNotifyObservers(false))), |
default_task_runner_(NewTaskQueue(TaskQueue::Spec("default_tq") |
.SetShouldMonitorQuiescence(true))), |
time_source_(new base::DefaultTickClock), |
+ queue_observer_(nullptr), |
tracing_category_(tracing_category), |
disabled_by_default_tracing_category_( |
disabled_by_default_tracing_category) { |
@@ -124,4 +123,17 @@ void SchedulerHelper::RemoveTaskObserver( |
task_queue_manager_->RemoveTaskObserver(task_observer); |
} |
+void SchedulerHelper::SetQueueObserver(Observer* observer) { |
+ CheckOnValidThread(); |
+ queue_observer_ = observer; |
+ DCHECK(task_queue_manager_); |
+ task_queue_manager_->SetQueueObserver(this); |
+} |
+ |
+void SchedulerHelper::OnUnregisterTaskQueue( |
+ const scoped_refptr<internal::TaskQueueImpl>& queue) { |
+ if (queue_observer_) |
+ queue_observer_->OnUnregisterTaskQueue(queue); |
+} |
+ |
} // namespace scheduler |