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_scheduler_impl.h" | 5 #include "base/task_scheduler/task_scheduler_impl.h" |
6 | 6 |
7 #include <utility> | 7 #include <utility> |
8 | 8 |
9 #include "base/bind.h" | 9 #include "base/bind.h" |
10 #include "base/bind_helpers.h" | 10 #include "base/bind_helpers.h" |
(...skipping 13 matching lines...) Expand all Loading... |
24 const WorkerPoolIndexForTraitsCallback& | 24 const WorkerPoolIndexForTraitsCallback& |
25 worker_pool_index_for_traits_callback) { | 25 worker_pool_index_for_traits_callback) { |
26 std::unique_ptr<TaskSchedulerImpl> scheduler( | 26 std::unique_ptr<TaskSchedulerImpl> scheduler( |
27 new TaskSchedulerImpl(worker_pool_index_for_traits_callback)); | 27 new TaskSchedulerImpl(worker_pool_index_for_traits_callback)); |
28 scheduler->Initialize(worker_pool_params_vector); | 28 scheduler->Initialize(worker_pool_params_vector); |
29 return scheduler; | 29 return scheduler; |
30 } | 30 } |
31 | 31 |
32 TaskSchedulerImpl::~TaskSchedulerImpl() { | 32 TaskSchedulerImpl::~TaskSchedulerImpl() { |
33 #if DCHECK_IS_ON() | 33 #if DCHECK_IS_ON() |
34 DCHECK(join_for_testing_returned_.IsSignaled()); | 34 DCHECK(join_for_testing_returned_.IsSet()); |
35 #endif | 35 #endif |
36 } | 36 } |
37 | 37 |
38 void TaskSchedulerImpl::PostTaskWithTraits( | 38 void TaskSchedulerImpl::PostTaskWithTraits( |
39 const tracked_objects::Location& from_here, | 39 const tracked_objects::Location& from_here, |
40 const TaskTraits& traits, | 40 const TaskTraits& traits, |
41 const Closure& task) { | 41 const Closure& task) { |
42 // Post |task| as part of a one-off single-task Sequence. | 42 // Post |task| as part of a one-off single-task Sequence. |
43 GetWorkerPoolForTraits(traits)->PostTaskWithSequence( | 43 GetWorkerPoolForTraits(traits)->PostTaskWithSequence( |
44 WrapUnique(new Task(from_here, task, traits, TimeDelta())), | 44 WrapUnique(new Task(from_here, task, traits, TimeDelta())), |
45 make_scoped_refptr(new Sequence), nullptr); | 45 make_scoped_refptr(new Sequence), nullptr); |
46 } | 46 } |
47 | 47 |
48 scoped_refptr<TaskRunner> TaskSchedulerImpl::CreateTaskRunnerWithTraits( | 48 scoped_refptr<TaskRunner> TaskSchedulerImpl::CreateTaskRunnerWithTraits( |
49 const TaskTraits& traits, | 49 const TaskTraits& traits, |
50 ExecutionMode execution_mode) { | 50 ExecutionMode execution_mode) { |
51 return GetWorkerPoolForTraits(traits)->CreateTaskRunnerWithTraits( | 51 return GetWorkerPoolForTraits(traits)->CreateTaskRunnerWithTraits( |
52 traits, execution_mode); | 52 traits, execution_mode); |
53 } | 53 } |
54 | 54 |
55 void TaskSchedulerImpl::Shutdown() { | 55 void TaskSchedulerImpl::Shutdown() { |
56 // TODO(fdoray): Increase the priority of BACKGROUND tasks blocking shutdown. | 56 // TODO(fdoray): Increase the priority of BACKGROUND tasks blocking shutdown. |
57 task_tracker_.Shutdown(); | 57 task_tracker_.Shutdown(); |
58 } | 58 } |
59 | 59 |
60 void TaskSchedulerImpl::JoinForTesting() { | 60 void TaskSchedulerImpl::JoinForTesting() { |
61 #if DCHECK_IS_ON() | 61 #if DCHECK_IS_ON() |
62 DCHECK(!join_for_testing_returned_.IsSignaled()); | 62 DCHECK(!join_for_testing_returned_.IsSet()); |
63 #endif | 63 #endif |
64 for (const auto& worker_pool : worker_pools_) | 64 for (const auto& worker_pool : worker_pools_) |
65 worker_pool->JoinForTesting(); | 65 worker_pool->JoinForTesting(); |
66 service_thread_->JoinForTesting(); | 66 service_thread_->JoinForTesting(); |
67 #if DCHECK_IS_ON() | 67 #if DCHECK_IS_ON() |
68 join_for_testing_returned_.Signal(); | 68 join_for_testing_returned_.Set(); |
69 #endif | 69 #endif |
70 } | 70 } |
71 | 71 |
72 TaskSchedulerImpl::TaskSchedulerImpl(const WorkerPoolIndexForTraitsCallback& | 72 TaskSchedulerImpl::TaskSchedulerImpl(const WorkerPoolIndexForTraitsCallback& |
73 worker_pool_index_for_traits_callback) | 73 worker_pool_index_for_traits_callback) |
74 : delayed_task_manager_( | 74 : delayed_task_manager_( |
75 Bind(&TaskSchedulerImpl::OnDelayedRunTimeUpdated, Unretained(this))), | 75 Bind(&TaskSchedulerImpl::OnDelayedRunTimeUpdated, Unretained(this))), |
76 worker_pool_index_for_traits_callback_( | 76 worker_pool_index_for_traits_callback_( |
77 worker_pool_index_for_traits_callback) | 77 worker_pool_index_for_traits_callback) |
78 #if DCHECK_IS_ON() | |
79 , | |
80 join_for_testing_returned_(WaitableEvent::ResetPolicy::MANUAL, | |
81 WaitableEvent::InitialState::NOT_SIGNALED) | |
82 #endif | |
83 { | 78 { |
84 DCHECK(!worker_pool_index_for_traits_callback_.is_null()); | 79 DCHECK(!worker_pool_index_for_traits_callback_.is_null()); |
85 } | 80 } |
86 | 81 |
87 void TaskSchedulerImpl::Initialize( | 82 void TaskSchedulerImpl::Initialize( |
88 const std::vector<SchedulerWorkerPoolParams>& worker_pool_params_vector) { | 83 const std::vector<SchedulerWorkerPoolParams>& worker_pool_params_vector) { |
89 DCHECK(!worker_pool_params_vector.empty()); | 84 DCHECK(!worker_pool_params_vector.empty()); |
90 | 85 |
91 const SchedulerWorkerPoolImpl::ReEnqueueSequenceCallback | 86 const SchedulerWorkerPoolImpl::ReEnqueueSequenceCallback |
92 re_enqueue_sequence_callback = | 87 re_enqueue_sequence_callback = |
(...skipping 36 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
129 GetWorkerPoolForTraits(traits)->ReEnqueueSequence(std::move(sequence), | 124 GetWorkerPoolForTraits(traits)->ReEnqueueSequence(std::move(sequence), |
130 sort_key); | 125 sort_key); |
131 } | 126 } |
132 | 127 |
133 void TaskSchedulerImpl::OnDelayedRunTimeUpdated() { | 128 void TaskSchedulerImpl::OnDelayedRunTimeUpdated() { |
134 service_thread_->WakeUp(); | 129 service_thread_->WakeUp(); |
135 } | 130 } |
136 | 131 |
137 } // namespace internal | 132 } // namespace internal |
138 } // namespace base | 133 } // namespace base |
OLD | NEW |