Index: third_party/WebKit/Source/platform/scheduler/renderer/renderer_scheduler_impl.cc |
diff --git a/third_party/WebKit/Source/platform/scheduler/renderer/renderer_scheduler_impl.cc b/third_party/WebKit/Source/platform/scheduler/renderer/renderer_scheduler_impl.cc |
index 4ae63b510f0ea3f368b47e6f97ddafc908e63ece..388a2633e1745c15982866f9677ed3847d134c7a 100644 |
--- a/third_party/WebKit/Source/platform/scheduler/renderer/renderer_scheduler_impl.cc |
+++ b/third_party/WebKit/Source/platform/scheduler/renderer/renderer_scheduler_impl.cc |
@@ -224,7 +224,8 @@ RendererSchedulerImpl::MainThreadOnly::MainThreadOnly( |
in_idle_period_for_testing(false), |
use_virtual_time(false), |
is_audio_playing(false), |
- rail_mode_observer(nullptr) { |
+ rail_mode_observer(nullptr), |
+ wake_up_budget_pool(nullptr) { |
foreground_main_thread_load_tracker.Resume(now); |
} |
@@ -338,6 +339,7 @@ scoped_refptr<TaskQueue> RendererSchedulerImpl::NewLoadingTaskQueue( |
} |
loading_task_queue->AddTaskObserver( |
&GetMainThreadOnly().loading_task_cost_estimator); |
+ AddQueueToWakeUpBudgetPool(loading_task_queue.get()); |
return loading_task_queue; |
} |
@@ -364,6 +366,7 @@ scoped_refptr<TaskQueue> RendererSchedulerImpl::NewTimerTaskQueue( |
} |
timer_task_queue->AddTaskObserver( |
&GetMainThreadOnly().timer_task_cost_estimator); |
+ AddQueueToWakeUpBudgetPool(timer_task_queue.get()); |
return timer_task_queue; |
} |
@@ -1357,6 +1360,10 @@ IdleTimeEstimator* RendererSchedulerImpl::GetIdleTimeEstimatorForTesting() { |
return &GetMainThreadOnly().idle_time_estimator; |
} |
+WakeUpBudgetPool* RendererSchedulerImpl::GetWakeUpBudgetPoolForTesting() { |
+ return GetMainThreadOnly().wake_up_budget_pool; |
+} |
+ |
void RendererSchedulerImpl::SuspendTimerQueue() { |
GetMainThreadOnly().timer_queue_suspend_count++; |
ForceUpdatePolicy(); |
@@ -1886,6 +1893,18 @@ bool RendererSchedulerImpl::ShouldDisableThrottlingBecauseOfAudio( |
now; |
} |
+void RendererSchedulerImpl::AddQueueToWakeUpBudgetPool(TaskQueue* queue) { |
+ if (!GetMainThreadOnly().wake_up_budget_pool) { |
+ GetMainThreadOnly().wake_up_budget_pool = |
+ task_queue_throttler()->CreateWakeUpBudgetPool("renderer-wake-up-pool"); |
+ GetMainThreadOnly().wake_up_budget_pool->SetWakeUpRate(1); |
+ GetMainThreadOnly().wake_up_budget_pool->SetWakeUpDuration( |
+ base::TimeDelta()); |
+ } |
+ GetMainThreadOnly().wake_up_budget_pool->AddQueue(tick_clock()->NowTicks(), |
+ queue); |
+} |
+ |
TimeDomain* RendererSchedulerImpl::GetActiveTimeDomain() { |
if (GetMainThreadOnly().use_virtual_time) { |
return GetVirtualTimeDomain(); |