Chromium Code Reviews| Index: components/scheduler/child/virtual_time_task_runner_delegate.cc |
| diff --git a/components/scheduler/child/virtual_time_task_runner_delegate.cc b/components/scheduler/child/virtual_time_task_runner_delegate.cc |
| new file mode 100644 |
| index 0000000000000000000000000000000000000000..c4644b36830bd7eb3af192c290986b277e34055c |
| --- /dev/null |
| +++ b/components/scheduler/child/virtual_time_task_runner_delegate.cc |
| @@ -0,0 +1,97 @@ |
| +// Copyright 2015 The Chromium Authors. All rights reserved. |
| +// Use of this source code is governed by a BSD-style license that can be |
| +// found in the LICENSE file. |
| + |
| +#include "components/scheduler/child/virtual_time_task_runner_delegate.h" |
| + |
| +namespace scheduler { |
| + |
| +// static |
| +scoped_refptr<VirtualTimeTaskRunnerDelegate> |
| +VirtualTimeTaskRunnerDelegate::Create(base::MessageLoop* message_loop, |
| + base::TimeTicks initial_now) { |
| + return make_scoped_refptr( |
| + new VirtualTimeTaskRunnerDelegate(message_loop, initial_now)); |
| +} |
| + |
| +VirtualTimeTaskRunnerDelegate::VirtualTimeTaskRunnerDelegate( |
| + base::MessageLoop* message_loop, |
| + base::TimeTicks initial_now) |
| + : message_loop_(message_loop), |
| + message_loop_task_runner_(message_loop->task_runner()), |
| + now_(initial_now), |
| + fast_forward_timers_(true) {} |
| + |
| +VirtualTimeTaskRunnerDelegate::~VirtualTimeTaskRunnerDelegate() { |
| + RestoreDefaultTaskRunner(); |
| +} |
| + |
| +void VirtualTimeTaskRunnerDelegate::SetDefaultTaskRunner( |
| + scoped_refptr<base::SingleThreadTaskRunner> task_runner) { |
| + message_loop_->SetTaskRunner(task_runner); |
| +} |
| + |
| +void VirtualTimeTaskRunnerDelegate::RestoreDefaultTaskRunner() { |
| + if (base::MessageLoop::current() == message_loop_) |
| + message_loop_->SetTaskRunner(message_loop_task_runner_); |
| +} |
| + |
| +bool VirtualTimeTaskRunnerDelegate::PostDelayedTask( |
| + const tracked_objects::Location& from_here, |
| + const base::Closure& task, |
| + base::TimeDelta delay) { |
| + if (delay > base::TimeDelta()) { |
| + base::TimeTicks fire_time = now_ + delay; |
| + delayed_wakeup_multimap_.insert(std::make_pair(fire_time, task)); |
| + } |
| + return message_loop_task_runner_->PostDelayedTask(from_here, task, delay); |
| +} |
| + |
| +bool VirtualTimeTaskRunnerDelegate::PostNonNestableDelayedTask( |
| + const tracked_objects::Location& from_here, |
| + const base::Closure& task, |
| + base::TimeDelta delay) { |
| + return message_loop_task_runner_->PostNonNestableDelayedTask(from_here, task, |
| + delay); |
| +} |
| + |
| +bool VirtualTimeTaskRunnerDelegate::RunsTasksOnCurrentThread() const { |
| + return message_loop_task_runner_->RunsTasksOnCurrentThread(); |
| +} |
| + |
| +bool VirtualTimeTaskRunnerDelegate::IsNested() const { |
| + return message_loop_->IsNested(); |
| +} |
| + |
| +base::TimeTicks VirtualTimeTaskRunnerDelegate::NowTicks() { |
| + return now_; |
| +} |
| + |
| +void VirtualTimeTaskRunnerDelegate::AdvancedTimeTo(base::TimeTicks now) { |
| + now_ = now; |
|
Sami
2015/10/28 20:00:35
DCHECK_GT(now, now_);
alex clarke (OOO till 29th)
2015/10/29 18:23:13
Done.
|
| + |
| + while (!delayed_wakeup_multimap_.empty()) { |
| + DelayedWakeupMultimap::iterator next_wakeup = |
| + delayed_wakeup_multimap_.begin(); |
| + if (next_wakeup->first > now) |
| + break; |
| + message_loop_task_runner_->PostDelayedTask(FROM_HERE, next_wakeup->second, |
| + base::TimeDelta()); |
| + delayed_wakeup_multimap_.erase(next_wakeup); |
| + } |
| +} |
| + |
| +void VirtualTimeTaskRunnerDelegate::OnNoMoreWork() { |
| + if (delayed_wakeup_multimap_.empty()) |
| + return; |
| + |
| + if (fast_forward_timers_) |
| + AdvancedTimeTo(delayed_wakeup_multimap_.begin()->first); |
| +} |
| + |
| +void VirtualTimeTaskRunnerDelegate::SetFastForwardTimers( |
| + bool fast_forward_timers) { |
| + fast_forward_timers_ = fast_forward_timers; |
| +} |
| + |
| +} // namespace scheduler |