OLD | NEW |
---|---|
(Empty) | |
1 // Copyright 2015 The Chromium Authors. All rights reserved. | |
2 // Use of this source code is governed by a BSD-style license that can be | |
3 // found in the LICENSE file. | |
4 | |
5 #include "components/scheduler/child/virtual_time_task_runner_delegate.h" | |
6 | |
7 namespace scheduler { | |
8 | |
9 // static | |
10 scoped_refptr<VirtualTimeTaskRunnerDelegate> | |
11 VirtualTimeTaskRunnerDelegate::Create(base::MessageLoop* message_loop, | |
12 base::TimeTicks initial_now) { | |
13 return make_scoped_refptr( | |
14 new VirtualTimeTaskRunnerDelegate(message_loop, initial_now)); | |
15 } | |
16 | |
17 VirtualTimeTaskRunnerDelegate::VirtualTimeTaskRunnerDelegate( | |
18 base::MessageLoop* message_loop, | |
19 base::TimeTicks initial_now) | |
20 : message_loop_(message_loop), | |
21 message_loop_task_runner_(message_loop->task_runner()), | |
22 now_(initial_now), | |
23 fast_forward_timers_(true) {} | |
24 | |
25 VirtualTimeTaskRunnerDelegate::~VirtualTimeTaskRunnerDelegate() { | |
26 RestoreDefaultTaskRunner(); | |
27 } | |
28 | |
29 void VirtualTimeTaskRunnerDelegate::SetDefaultTaskRunner( | |
30 scoped_refptr<base::SingleThreadTaskRunner> task_runner) { | |
31 message_loop_->SetTaskRunner(task_runner); | |
32 } | |
33 | |
34 void VirtualTimeTaskRunnerDelegate::RestoreDefaultTaskRunner() { | |
35 if (base::MessageLoop::current() == message_loop_) | |
36 message_loop_->SetTaskRunner(message_loop_task_runner_); | |
37 } | |
38 | |
39 bool VirtualTimeTaskRunnerDelegate::PostDelayedTask( | |
40 const tracked_objects::Location& from_here, | |
41 const base::Closure& task, | |
42 base::TimeDelta delay) { | |
43 if (delay > base::TimeDelta()) { | |
44 base::TimeTicks fire_time = now_ + delay; | |
45 delayed_wakeup_multimap_.insert(std::make_pair(fire_time, task)); | |
46 } | |
47 return message_loop_task_runner_->PostDelayedTask(from_here, task, delay); | |
48 } | |
49 | |
50 bool VirtualTimeTaskRunnerDelegate::PostNonNestableDelayedTask( | |
51 const tracked_objects::Location& from_here, | |
52 const base::Closure& task, | |
53 base::TimeDelta delay) { | |
54 return message_loop_task_runner_->PostNonNestableDelayedTask(from_here, task, | |
55 delay); | |
56 } | |
57 | |
58 bool VirtualTimeTaskRunnerDelegate::RunsTasksOnCurrentThread() const { | |
59 return message_loop_task_runner_->RunsTasksOnCurrentThread(); | |
60 } | |
61 | |
62 bool VirtualTimeTaskRunnerDelegate::IsNested() const { | |
63 return message_loop_->IsNested(); | |
64 } | |
65 | |
66 base::TimeTicks VirtualTimeTaskRunnerDelegate::NowTicks() { | |
67 return now_; | |
68 } | |
69 | |
70 void VirtualTimeTaskRunnerDelegate::AdvancedTimeTo(base::TimeTicks now) { | |
71 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.
| |
72 | |
73 while (!delayed_wakeup_multimap_.empty()) { | |
74 DelayedWakeupMultimap::iterator next_wakeup = | |
75 delayed_wakeup_multimap_.begin(); | |
76 if (next_wakeup->first > now) | |
77 break; | |
78 message_loop_task_runner_->PostDelayedTask(FROM_HERE, next_wakeup->second, | |
79 base::TimeDelta()); | |
80 delayed_wakeup_multimap_.erase(next_wakeup); | |
81 } | |
82 } | |
83 | |
84 void VirtualTimeTaskRunnerDelegate::OnNoMoreWork() { | |
85 if (delayed_wakeup_multimap_.empty()) | |
86 return; | |
87 | |
88 if (fast_forward_timers_) | |
89 AdvancedTimeTo(delayed_wakeup_multimap_.begin()->first); | |
90 } | |
91 | |
92 void VirtualTimeTaskRunnerDelegate::SetFastForwardTimers( | |
93 bool fast_forward_timers) { | |
94 fast_forward_timers_ = fast_forward_timers; | |
95 } | |
96 | |
97 } // namespace scheduler | |
OLD | NEW |