scheduler: Implement task observers

content/renderer/scheduler/renderer_scheduler_impl.cc

Index: content/renderer/scheduler/renderer_scheduler_impl.cc
diff --git a/content/renderer/scheduler/renderer_scheduler_impl.cc b/content/renderer/scheduler/renderer_scheduler_impl.cc
index 7a91a2d7cb117e4ba7a3584e885489d2b70fdfa6..1536745bf0cd531febab4c3e03226a677c00a6d7 100644
--- a/content/renderer/scheduler/renderer_scheduler_impl.cc
+++ b/content/renderer/scheduler/renderer_scheduler_impl.cc
@@ -336,6 +336,11 @@ void RendererSchedulerImpl::SetTimeSourceForTesting(
   task_queue_manager_->SetTimeSourceForTesting(time_source);
 }
 
+void RendererSchedulerImpl::SetWorkBatchSizeForTesting(size_t work_batch_size) {
+  DCHECK(main_thread_checker_.CalledOnValidThread());
+  task_queue_manager_->SetWorkBatchSize(work_batch_size);
+}
+
 base::TimeTicks RendererSchedulerImpl::Now() const {
   return UNLIKELY(time_source_) ? time_source_->Now() : base::TimeTicks::Now();
 }
@@ -464,4 +469,18 @@ RendererSchedulerImpl::ComputeNewInputStreamState(
   return INPUT_ACTIVE;
 }
 
+void RendererSchedulerImpl::AddTaskObserver(
+    base::MessageLoop::TaskObserver* task_observer) {
+  DCHECK(main_thread_checker_.CalledOnValidThread());
+  if (task_queue_manager_)
+    task_queue_manager_->AddTaskObserver(task_observer);
+}
+
+void RendererSchedulerImpl::RemoveTaskObserver(
+    base::MessageLoop::TaskObserver* task_observer) {
+  DCHECK(main_thread_checker_.CalledOnValidThread());
+  if (task_queue_manager_)
+    task_queue_manager_->RemoveTaskObserver(task_observer);
+}
+
 }  // namespace content