OLD | NEW |
1 // Copyright 2016 The Chromium Authors. All rights reserved. | 1 // Copyright 2016 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 "base/task_scheduler/task_tracker.h" | 5 #include "base/task_scheduler/task_tracker.h" |
6 | 6 |
7 #include <limits> | 7 #include <limits> |
8 | 8 |
9 #include "base/atomicops.h" | 9 #include "base/atomicops.h" |
10 #include "base/callback.h" | 10 #include "base/callback.h" |
(...skipping 106 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
117 TaskTracker::TaskTracker() : state_(new State) {} | 117 TaskTracker::TaskTracker() : state_(new State) {} |
118 TaskTracker::~TaskTracker() = default; | 118 TaskTracker::~TaskTracker() = default; |
119 | 119 |
120 void TaskTracker::Shutdown() { | 120 void TaskTracker::Shutdown() { |
121 { | 121 { |
122 AutoSchedulerLock auto_lock(shutdown_lock_); | 122 AutoSchedulerLock auto_lock(shutdown_lock_); |
123 | 123 |
124 // This method can only be called once. | 124 // This method can only be called once. |
125 DCHECK(!shutdown_event_); | 125 DCHECK(!shutdown_event_); |
126 DCHECK(!num_block_shutdown_tasks_posted_during_shutdown_); | 126 DCHECK(!num_block_shutdown_tasks_posted_during_shutdown_); |
| 127 DCHECK(!state_->HasShutdownStarted()); |
127 | 128 |
128 shutdown_event_.reset( | 129 shutdown_event_.reset( |
129 new WaitableEvent(WaitableEvent::ResetPolicy::MANUAL, | 130 new WaitableEvent(WaitableEvent::ResetPolicy::MANUAL, |
130 WaitableEvent::InitialState::NOT_SIGNALED)); | 131 WaitableEvent::InitialState::NOT_SIGNALED)); |
131 | 132 |
132 const bool tasks_are_blocking_shutdown = state_->StartShutdown(); | 133 const bool tasks_are_blocking_shutdown = state_->StartShutdown(); |
133 | 134 |
134 // From now, if a thread causes the number of tasks blocking shutdown to | 135 // From now, if a thread causes the number of tasks blocking shutdown to |
135 // become zero, it will call OnBlockingShutdownTasksComplete(). | 136 // become zero, it will call OnBlockingShutdownTasksComplete(). |
136 | 137 |
(...skipping 70 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
207 | 208 |
208 TRACE_TASK_EXECUTION(kRunFunctionName, *task); | 209 TRACE_TASK_EXECUTION(kRunFunctionName, *task); |
209 | 210 |
210 debug::TaskAnnotator task_annotator; | 211 debug::TaskAnnotator task_annotator; |
211 task_annotator.RunTask(kQueueFunctionName, *task); | 212 task_annotator.RunTask(kQueueFunctionName, *task); |
212 } | 213 } |
213 | 214 |
214 AfterRunTask(shutdown_behavior); | 215 AfterRunTask(shutdown_behavior); |
215 } | 216 } |
216 | 217 |
| 218 bool TaskTracker::HasShutdownStarted() const { |
| 219 return state_->HasShutdownStarted(); |
| 220 } |
| 221 |
217 bool TaskTracker::IsShutdownComplete() const { | 222 bool TaskTracker::IsShutdownComplete() const { |
218 AutoSchedulerLock auto_lock(shutdown_lock_); | 223 AutoSchedulerLock auto_lock(shutdown_lock_); |
219 return shutdown_event_ && shutdown_event_->IsSignaled(); | 224 return shutdown_event_ && shutdown_event_->IsSignaled(); |
220 } | 225 } |
221 | 226 |
222 bool TaskTracker::IsShuttingDownForTesting() const { | 227 void TaskTracker::SetHasShutdownStartedForTesting() { |
223 AutoSchedulerLock auto_lock(shutdown_lock_); | 228 state_->StartShutdown(); |
224 return shutdown_event_ && !shutdown_event_->IsSignaled(); | |
225 } | 229 } |
226 | 230 |
227 bool TaskTracker::BeforePostTask(TaskShutdownBehavior shutdown_behavior) { | 231 bool TaskTracker::BeforePostTask(TaskShutdownBehavior shutdown_behavior) { |
228 if (shutdown_behavior == TaskShutdownBehavior::BLOCK_SHUTDOWN) { | 232 if (shutdown_behavior == TaskShutdownBehavior::BLOCK_SHUTDOWN) { |
229 // BLOCK_SHUTDOWN tasks block shutdown between the moment they are posted | 233 // BLOCK_SHUTDOWN tasks block shutdown between the moment they are posted |
230 // and the moment they complete their execution. | 234 // and the moment they complete their execution. |
231 const bool shutdown_started = state_->IncrementNumTasksBlockingShutdown(); | 235 const bool shutdown_started = state_->IncrementNumTasksBlockingShutdown(); |
232 | 236 |
233 if (shutdown_started) { | 237 if (shutdown_started) { |
234 AutoSchedulerLock auto_lock(shutdown_lock_); | 238 AutoSchedulerLock auto_lock(shutdown_lock_); |
(...skipping 83 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
318 | 322 |
319 // This method can only be called after shutdown has started. | 323 // This method can only be called after shutdown has started. |
320 DCHECK(state_->HasShutdownStarted()); | 324 DCHECK(state_->HasShutdownStarted()); |
321 DCHECK(shutdown_event_); | 325 DCHECK(shutdown_event_); |
322 | 326 |
323 shutdown_event_->Signal(); | 327 shutdown_event_->Signal(); |
324 } | 328 } |
325 | 329 |
326 } // namespace internal | 330 } // namespace internal |
327 } // namespace base | 331 } // namespace base |
OLD | NEW |