OLD | NEW |
1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "components/domain_reliability/dispatcher.h" | 5 #include "components/domain_reliability/dispatcher.h" |
6 | 6 |
7 #include "base/bind.h" | 7 #include "base/bind.h" |
8 #include "base/message_loop/message_loop.h" | 8 #include "base/message_loop/message_loop.h" |
9 #include "base/stl_util.h" | 9 #include "base/stl_util.h" |
10 #include "base/timer/timer.h" | 10 #include "base/timer/timer.h" |
11 #include "components/domain_reliability/util.h" | 11 #include "components/domain_reliability/util.h" |
12 | 12 |
13 namespace domain_reliability { | 13 namespace domain_reliability { |
14 | 14 |
| 15 struct DomainReliabilityDispatcher::Task { |
| 16 Task(const base::Closure& closure, |
| 17 scoped_ptr<MockableTime::Timer> timer, |
| 18 base::TimeDelta min_delay, |
| 19 base::TimeDelta max_delay); |
| 20 ~Task(); |
| 21 |
| 22 base::Closure closure; |
| 23 scoped_ptr<MockableTime::Timer> timer; |
| 24 base::TimeDelta min_delay; |
| 25 base::TimeDelta max_delay; |
| 26 bool eligible; |
| 27 }; |
| 28 |
| 29 DomainReliabilityDispatcher::Task::Task(const base::Closure& closure, |
| 30 scoped_ptr<MockableTime::Timer> timer, |
| 31 base::TimeDelta min_delay, |
| 32 base::TimeDelta max_delay) |
| 33 : closure(closure), |
| 34 timer(timer.Pass()), |
| 35 min_delay(min_delay), |
| 36 max_delay(max_delay), |
| 37 eligible(false) {} |
| 38 |
| 39 DomainReliabilityDispatcher::Task::~Task() {} |
| 40 |
15 DomainReliabilityDispatcher::DomainReliabilityDispatcher(MockableTime* time) | 41 DomainReliabilityDispatcher::DomainReliabilityDispatcher(MockableTime* time) |
16 : time_(time) {} | 42 : time_(time) {} |
17 | 43 |
18 DomainReliabilityDispatcher::~DomainReliabilityDispatcher() { | 44 DomainReliabilityDispatcher::~DomainReliabilityDispatcher() { |
19 // TODO(ttuttle): STLElementDeleter? | 45 // TODO(ttuttle): STLElementDeleter? |
20 STLDeleteElements(&tasks_); | 46 STLDeleteElements(&tasks_); |
21 } | 47 } |
22 | 48 |
23 void DomainReliabilityDispatcher::ScheduleTask( | 49 void DomainReliabilityDispatcher::ScheduleTask( |
24 const base::Closure& closure, | 50 const base::Closure& closure, |
(...skipping 24 matching lines...) Expand all Loading... |
49 for (std::set<Task*>::const_iterator it = tasks.begin(); | 75 for (std::set<Task*>::const_iterator it = tasks.begin(); |
50 it != tasks.end(); | 76 it != tasks.end(); |
51 ++it) { | 77 ++it) { |
52 Task* task = *it; | 78 Task* task = *it; |
53 DCHECK(task); | 79 DCHECK(task); |
54 DCHECK(task->eligible); | 80 DCHECK(task->eligible); |
55 RunAndDeleteTask(task); | 81 RunAndDeleteTask(task); |
56 } | 82 } |
57 } | 83 } |
58 | 84 |
59 DomainReliabilityDispatcher::Task::Task(const base::Closure& closure, | |
60 scoped_ptr<MockableTime::Timer> timer, | |
61 base::TimeDelta min_delay, | |
62 base::TimeDelta max_delay) | |
63 : closure(closure), | |
64 timer(timer.Pass()), | |
65 min_delay(min_delay), | |
66 max_delay(max_delay), | |
67 eligible(false) {} | |
68 | |
69 DomainReliabilityDispatcher::Task::~Task() {} | |
70 | |
71 void DomainReliabilityDispatcher::MakeTaskWaiting(Task* task) { | 85 void DomainReliabilityDispatcher::MakeTaskWaiting(Task* task) { |
72 DCHECK(task); | 86 DCHECK(task); |
73 DCHECK(!task->eligible); | 87 DCHECK(!task->eligible); |
74 DCHECK(!task->timer->IsRunning()); | 88 DCHECK(!task->timer->IsRunning()); |
75 task->timer->Start( | 89 task->timer->Start(FROM_HERE, |
76 FROM_HERE, | 90 task->min_delay, |
77 task->min_delay, | 91 base::Bind(&DomainReliabilityDispatcher::MakeTaskEligible, |
78 base::Bind( | 92 base::Unretained(this), |
79 &DomainReliabilityDispatcher::MakeTaskEligible, | 93 task)); |
80 base::Unretained(this), | |
81 task)); | |
82 } | 94 } |
83 | 95 |
84 void | 96 void |
85 DomainReliabilityDispatcher::MakeTaskEligible(Task* task) { | 97 DomainReliabilityDispatcher::MakeTaskEligible(Task* task) { |
86 DCHECK(task); | 98 DCHECK(task); |
87 DCHECK(!task->eligible); | 99 DCHECK(!task->eligible); |
88 task->eligible = true; | 100 task->eligible = true; |
89 eligible_tasks_.insert(task); | 101 eligible_tasks_.insert(task); |
90 task->timer->Start( | 102 task->timer->Start(FROM_HERE, |
91 FROM_HERE, | 103 task->max_delay - task->min_delay, |
92 task->max_delay - task->min_delay, | 104 base::Bind(&DomainReliabilityDispatcher::RunAndDeleteTask, |
93 base::Bind( | 105 base::Unretained(this), |
94 &DomainReliabilityDispatcher::RunAndDeleteTask, | 106 task)); |
95 base::Unretained(this), | |
96 task)); | |
97 } | 107 } |
98 | 108 |
99 void DomainReliabilityDispatcher::RunAndDeleteTask(Task* task) { | 109 void DomainReliabilityDispatcher::RunAndDeleteTask(Task* task) { |
100 DCHECK(task); | 110 DCHECK(task); |
101 DCHECK(!task->closure.is_null()); | 111 DCHECK(!task->closure.is_null()); |
102 task->closure.Run(); | 112 task->closure.Run(); |
103 if (task->eligible) | 113 if (task->eligible) |
104 eligible_tasks_.erase(task); | 114 eligible_tasks_.erase(task); |
105 tasks_.erase(task); | 115 tasks_.erase(task); |
106 delete task; | 116 delete task; |
107 } | 117 } |
108 | 118 |
109 } // namespace domain_reliability | 119 } // namespace domain_reliability |
OLD | NEW |