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" |
11 #include "base/memory/ptr_util.h" | 11 #include "base/memory/ptr_util.h" |
12 #include "base/task_scheduler/delayed_task_manager.h" | 12 #include "base/task_scheduler/delayed_task_manager.h" |
13 #include "base/task_scheduler/scheduler_single_thread_task_runner_manager.h" | 13 #include "base/task_scheduler/scheduler_single_thread_task_runner_manager.h" |
14 #include "base/task_scheduler/scheduler_worker_pool_params.h" | 14 #include "base/task_scheduler/scheduler_worker_pool_params.h" |
15 #include "base/task_scheduler/sequence_sort_key.h" | 15 #include "base/task_scheduler/sequence_sort_key.h" |
16 #include "base/task_scheduler/task.h" | 16 #include "base/task_scheduler/task.h" |
17 #include "base/task_scheduler/task_tracker.h" | 17 #include "base/task_scheduler/task_tracker.h" |
18 #include "build/build_config.h" | |
19 | |
20 #if defined(OS_POSIX) && !defined(OS_NACL_SFI) | |
21 #include "base/task_scheduler/task_tracker_posix.h" | |
22 #endif | |
23 | 18 |
24 namespace base { | 19 namespace base { |
25 namespace internal { | 20 namespace internal { |
26 | 21 |
27 namespace { | 22 namespace { |
28 | 23 |
29 enum EnvironmentType { | 24 enum EnvironmentType { |
30 BACKGROUND = 0, | 25 BACKGROUND = 0, |
31 BACKGROUND_BLOCKING, | 26 BACKGROUND_BLOCKING, |
32 FOREGROUND, | 27 FOREGROUND, |
(...skipping 94 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
127 return histograms; | 122 return histograms; |
128 } | 123 } |
129 | 124 |
130 int TaskSchedulerImpl::GetMaxConcurrentTasksWithTraitsDeprecated( | 125 int TaskSchedulerImpl::GetMaxConcurrentTasksWithTraitsDeprecated( |
131 const TaskTraits& traits) const { | 126 const TaskTraits& traits) const { |
132 return GetWorkerPoolForTraits(traits)->GetMaxConcurrentTasksDeprecated(); | 127 return GetWorkerPoolForTraits(traits)->GetMaxConcurrentTasksDeprecated(); |
133 } | 128 } |
134 | 129 |
135 void TaskSchedulerImpl::Shutdown() { | 130 void TaskSchedulerImpl::Shutdown() { |
136 // TODO(fdoray): Increase the priority of BACKGROUND tasks blocking shutdown. | 131 // TODO(fdoray): Increase the priority of BACKGROUND tasks blocking shutdown. |
137 DCHECK(task_tracker_); | 132 task_tracker_.Shutdown(); |
138 task_tracker_->Shutdown(); | |
139 } | 133 } |
140 | 134 |
141 void TaskSchedulerImpl::FlushForTesting() { | 135 void TaskSchedulerImpl::FlushForTesting() { |
142 DCHECK(task_tracker_); | 136 task_tracker_.Flush(); |
143 task_tracker_->Flush(); | |
144 } | 137 } |
145 | 138 |
146 void TaskSchedulerImpl::JoinForTesting() { | 139 void TaskSchedulerImpl::JoinForTesting() { |
147 #if DCHECK_IS_ON() | 140 #if DCHECK_IS_ON() |
148 DCHECK(!join_for_testing_returned_.IsSet()); | 141 DCHECK(!join_for_testing_returned_.IsSet()); |
149 #endif | 142 #endif |
150 single_thread_task_runner_manager_->JoinForTesting(); | 143 single_thread_task_runner_manager_->JoinForTesting(); |
151 for (const auto& worker_pool : worker_pools_) | 144 for (const auto& worker_pool : worker_pools_) |
152 worker_pool->DisallowWorkerDetachmentForTesting(); | 145 worker_pool->DisallowWorkerDetachmentForTesting(); |
153 for (const auto& worker_pool : worker_pools_) | 146 for (const auto& worker_pool : worker_pools_) |
(...skipping 15 matching lines...) Expand all Loading... |
169 Thread::Options service_thread_options; | 162 Thread::Options service_thread_options; |
170 service_thread_options.message_loop_type = | 163 service_thread_options.message_loop_type = |
171 #if defined(OS_POSIX) && !defined(OS_NACL_SFI) | 164 #if defined(OS_POSIX) && !defined(OS_NACL_SFI) |
172 MessageLoop::TYPE_IO; | 165 MessageLoop::TYPE_IO; |
173 #else | 166 #else |
174 MessageLoop::TYPE_DEFAULT; | 167 MessageLoop::TYPE_DEFAULT; |
175 #endif | 168 #endif |
176 service_thread_options.timer_slack = TIMER_SLACK_MAXIMUM; | 169 service_thread_options.timer_slack = TIMER_SLACK_MAXIMUM; |
177 CHECK(service_thread_.StartWithOptions(service_thread_options)); | 170 CHECK(service_thread_.StartWithOptions(service_thread_options)); |
178 | 171 |
179 // Instantiate TaskTracker. Needs to happen after starting the service thread | |
180 // to get its message_loop(). | |
181 task_tracker_ = | |
182 #if defined(OS_POSIX) && !defined(OS_NACL_SFI) | 172 #if defined(OS_POSIX) && !defined(OS_NACL_SFI) |
183 base::MakeUnique<TaskTrackerPosix>( | 173 // Needs to happen after starting the service thread to get its |
184 static_cast<MessageLoopForIO*>(service_thread_.message_loop())); | 174 // message_loop(). |
185 #else | 175 task_tracker_.set_watch_file_descriptor_message_loop( |
186 base::MakeUnique<TaskTracker>(); | 176 static_cast<MessageLoopForIO*>(service_thread_.message_loop())); |
187 #endif | 177 #endif |
188 | 178 |
189 // Instantiate DelayedTaskManager. Needs to happen after starting the service | 179 // Needs to happen after starting the service thread to get its task_runner(). |
190 // thread to get its task_runner(). | |
191 delayed_task_manager_ = | 180 delayed_task_manager_ = |
192 base::MakeUnique<DelayedTaskManager>(service_thread_.task_runner()); | 181 base::MakeUnique<DelayedTaskManager>(service_thread_.task_runner()); |
193 | 182 |
194 single_thread_task_runner_manager_ = | 183 single_thread_task_runner_manager_ = |
195 MakeUnique<SchedulerSingleThreadTaskRunnerManager>( | 184 MakeUnique<SchedulerSingleThreadTaskRunnerManager>( |
196 task_tracker_.get(), delayed_task_manager_.get()); | 185 &task_tracker_, delayed_task_manager_.get()); |
197 single_thread_task_runner_manager_->Start(); | 186 single_thread_task_runner_manager_->Start(); |
198 | 187 |
199 // Callback invoked by workers to re-enqueue a sequence in the appropriate | 188 // Callback invoked by workers to re-enqueue a sequence in the appropriate |
200 // PriorityQueue. | 189 // PriorityQueue. |
201 const SchedulerWorkerPoolImpl::ReEnqueueSequenceCallback | 190 const SchedulerWorkerPoolImpl::ReEnqueueSequenceCallback |
202 re_enqueue_sequence_callback = | 191 re_enqueue_sequence_callback = |
203 Bind(&TaskSchedulerImpl::ReEnqueueSequenceCallback, Unretained(this)); | 192 Bind(&TaskSchedulerImpl::ReEnqueueSequenceCallback, Unretained(this)); |
204 | 193 |
205 // Order must match the EnvironmentType enum. | 194 // Order must match the EnvironmentType enum. |
206 const SchedulerWorkerPoolParams* worker_pool_params[] = { | 195 const SchedulerWorkerPoolParams* worker_pool_params[] = { |
(...skipping 13 matching lines...) Expand all Loading... |
220 | 209 |
221 // Start worker pools. | 210 // Start worker pools. |
222 for (int environment_type = 0; environment_type < ENVIRONMENT_COUNT; | 211 for (int environment_type = 0; environment_type < ENVIRONMENT_COUNT; |
223 ++environment_type) { | 212 ++environment_type) { |
224 // Passing pointers to objects owned by |this| to the constructor of | 213 // Passing pointers to objects owned by |this| to the constructor of |
225 // SchedulerWorkerPoolImpl is safe because a TaskSchedulerImpl can't be | 214 // SchedulerWorkerPoolImpl is safe because a TaskSchedulerImpl can't be |
226 // deleted before all its worker pools have been joined. | 215 // deleted before all its worker pools have been joined. |
227 worker_pools_[environment_type] = MakeUnique<SchedulerWorkerPoolImpl>( | 216 worker_pools_[environment_type] = MakeUnique<SchedulerWorkerPoolImpl>( |
228 name_ + kEnvironmentParams[environment_type].name_suffix, | 217 name_ + kEnvironmentParams[environment_type].name_suffix, |
229 kEnvironmentParams[environment_type].priority_hint, | 218 kEnvironmentParams[environment_type].priority_hint, |
230 re_enqueue_sequence_callback, task_tracker_.get(), | 219 re_enqueue_sequence_callback, &task_tracker_, |
231 delayed_task_manager_.get()); | 220 delayed_task_manager_.get()); |
232 worker_pools_[environment_type]->Start( | 221 worker_pools_[environment_type]->Start( |
233 *worker_pool_params[environment_type]); | 222 *worker_pool_params[environment_type]); |
234 } | 223 } |
235 } | 224 } |
236 | 225 |
237 SchedulerWorkerPoolImpl* TaskSchedulerImpl::GetWorkerPoolForTraits( | 226 SchedulerWorkerPoolImpl* TaskSchedulerImpl::GetWorkerPoolForTraits( |
238 const TaskTraits& traits) const { | 227 const TaskTraits& traits) const { |
239 return worker_pools_[GetEnvironmentIndexForTraits(traits)].get(); | 228 return worker_pools_[GetEnvironmentIndexForTraits(traits)].get(); |
240 } | 229 } |
241 | 230 |
242 void TaskSchedulerImpl::ReEnqueueSequenceCallback( | 231 void TaskSchedulerImpl::ReEnqueueSequenceCallback( |
243 scoped_refptr<Sequence> sequence) { | 232 scoped_refptr<Sequence> sequence) { |
244 DCHECK(sequence); | 233 DCHECK(sequence); |
245 | 234 |
246 const SequenceSortKey sort_key = sequence->GetSortKey(); | 235 const SequenceSortKey sort_key = sequence->GetSortKey(); |
247 | 236 |
248 // The next task in |sequence| should run in a worker pool suited for its | 237 // The next task in |sequence| should run in a worker pool suited for its |
249 // traits, except for the priority which is adjusted to the highest priority | 238 // traits, except for the priority which is adjusted to the highest priority |
250 // in |sequence|. | 239 // in |sequence|. |
251 const TaskTraits traits = | 240 const TaskTraits traits = |
252 sequence->PeekTaskTraits().WithPriority(sort_key.priority()); | 241 sequence->PeekTaskTraits().WithPriority(sort_key.priority()); |
253 | 242 |
254 GetWorkerPoolForTraits(traits)->ReEnqueueSequence(std::move(sequence), | 243 GetWorkerPoolForTraits(traits)->ReEnqueueSequence(std::move(sequence), |
255 sort_key); | 244 sort_key); |
256 } | 245 } |
257 | 246 |
258 } // namespace internal | 247 } // namespace internal |
259 } // namespace base | 248 } // namespace base |
OLD | NEW |