OLD | NEW |
1 // Copyright 2017 The Chromium Authors. All rights reserved. | 1 // Copyright 2017 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/scheduler_single_thread_task_runner_manager.h" | 5 #include "base/task_scheduler/scheduler_single_thread_task_runner_manager.h" |
6 | 6 |
7 #include <algorithm> | 7 #include <algorithm> |
8 #include <memory> | 8 #include <memory> |
9 #include <string> | 9 #include <string> |
10 | 10 |
(...skipping 189 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
200 } | 200 } |
201 | 201 |
202 SchedulerSingleThreadTaskRunnerManager* const outer_; | 202 SchedulerSingleThreadTaskRunnerManager* const outer_; |
203 const TaskTraits traits_; | 203 const TaskTraits traits_; |
204 SchedulerWorker* const worker_; | 204 SchedulerWorker* const worker_; |
205 | 205 |
206 DISALLOW_COPY_AND_ASSIGN(SchedulerSingleThreadTaskRunner); | 206 DISALLOW_COPY_AND_ASSIGN(SchedulerSingleThreadTaskRunner); |
207 }; | 207 }; |
208 | 208 |
209 SchedulerSingleThreadTaskRunnerManager::SchedulerSingleThreadTaskRunnerManager( | 209 SchedulerSingleThreadTaskRunnerManager::SchedulerSingleThreadTaskRunnerManager( |
210 const std::vector<SchedulerWorkerPoolParams>& worker_pool_params_vector, | |
211 const TaskScheduler::WorkerPoolIndexForTraitsCallback& | |
212 worker_pool_index_for_traits_callback, | |
213 TaskTracker* task_tracker, | 210 TaskTracker* task_tracker, |
214 DelayedTaskManager* delayed_task_manager) | 211 DelayedTaskManager* delayed_task_manager) |
215 : worker_pool_params_vector_(worker_pool_params_vector), | 212 : task_tracker_(task_tracker), delayed_task_manager_(delayed_task_manager) { |
216 worker_pool_index_for_traits_callback_( | |
217 worker_pool_index_for_traits_callback), | |
218 task_tracker_(task_tracker), | |
219 delayed_task_manager_(delayed_task_manager) { | |
220 DCHECK_GT(worker_pool_params_vector_.size(), 0U); | |
221 DCHECK(worker_pool_index_for_traits_callback_); | |
222 DCHECK(task_tracker_); | 213 DCHECK(task_tracker_); |
223 DCHECK(delayed_task_manager_); | 214 DCHECK(delayed_task_manager_); |
224 } | 215 } |
225 | 216 |
226 SchedulerSingleThreadTaskRunnerManager:: | 217 SchedulerSingleThreadTaskRunnerManager:: |
227 ~SchedulerSingleThreadTaskRunnerManager() { | 218 ~SchedulerSingleThreadTaskRunnerManager() { |
228 #if DCHECK_IS_ON() | 219 #if DCHECK_IS_ON() |
229 size_t workers_unregistered_during_join = | 220 size_t workers_unregistered_during_join = |
230 subtle::NoBarrier_Load(&workers_unregistered_during_join_); | 221 subtle::NoBarrier_Load(&workers_unregistered_during_join_); |
231 DCHECK_EQ(workers_unregistered_during_join, workers_.size()) | 222 DCHECK_EQ(workers_unregistered_during_join, workers_.size()) |
232 << "There cannot be outstanding SingleThreadTaskRunners upon destruction " | 223 << "There cannot be outstanding SingleThreadTaskRunners upon destruction " |
233 "of SchedulerSingleThreadTaskRunnerManager or the Task Scheduler"; | 224 "of SchedulerSingleThreadTaskRunnerManager or the Task Scheduler"; |
234 #endif | 225 #endif |
235 } | 226 } |
236 | 227 |
237 scoped_refptr<SingleThreadTaskRunner> | 228 scoped_refptr<SingleThreadTaskRunner> |
238 SchedulerSingleThreadTaskRunnerManager::CreateSingleThreadTaskRunnerWithTraits( | 229 SchedulerSingleThreadTaskRunnerManager::CreateSingleThreadTaskRunnerWithTraits( |
| 230 const std::string& name, |
| 231 ThreadPriority priority_hint, |
239 const TaskTraits& traits) { | 232 const TaskTraits& traits) { |
240 size_t index = worker_pool_index_for_traits_callback_.Run(traits); | |
241 DCHECK_LT(index, worker_pool_params_vector_.size()); | |
242 return new SchedulerSingleThreadTaskRunner( | 233 return new SchedulerSingleThreadTaskRunner( |
243 this, traits, | 234 this, traits, CreateAndRegisterSchedulerWorker(name, priority_hint)); |
244 CreateAndRegisterSchedulerWorker(worker_pool_params_vector_[index])); | |
245 } | 235 } |
246 | 236 |
247 void SchedulerSingleThreadTaskRunnerManager::JoinForTesting() { | 237 void SchedulerSingleThreadTaskRunnerManager::JoinForTesting() { |
248 decltype(workers_) local_workers; | 238 decltype(workers_) local_workers; |
249 { | 239 { |
250 AutoSchedulerLock auto_lock(workers_lock_); | 240 AutoSchedulerLock auto_lock(workers_lock_); |
251 local_workers = std::move(workers_); | 241 local_workers = std::move(workers_); |
252 } | 242 } |
253 | 243 |
254 for (const auto& worker : local_workers) | 244 for (const auto& worker : local_workers) |
255 worker->JoinForTesting(); | 245 worker->JoinForTesting(); |
256 | 246 |
257 { | 247 { |
258 AutoSchedulerLock auto_lock(workers_lock_); | 248 AutoSchedulerLock auto_lock(workers_lock_); |
259 DCHECK(workers_.empty()) | 249 DCHECK(workers_.empty()) |
260 << "New worker(s) unexpectedly registered during join."; | 250 << "New worker(s) unexpectedly registered during join."; |
261 workers_ = std::move(local_workers); | 251 workers_ = std::move(local_workers); |
262 } | 252 } |
263 } | 253 } |
264 | 254 |
265 SchedulerWorker* | 255 SchedulerWorker* |
266 SchedulerSingleThreadTaskRunnerManager::CreateAndRegisterSchedulerWorker( | 256 SchedulerSingleThreadTaskRunnerManager::CreateAndRegisterSchedulerWorker( |
267 const SchedulerWorkerPoolParams& params) { | 257 const std::string& name, |
| 258 ThreadPriority priority_hint) { |
268 AutoSchedulerLock auto_lock(workers_lock_); | 259 AutoSchedulerLock auto_lock(workers_lock_); |
269 int id = next_worker_id_++; | 260 int id = next_worker_id_++; |
270 auto delegate = MakeUnique<SchedulerWorkerDelegate>(base::StringPrintf( | 261 auto delegate = MakeUnique<SchedulerWorkerDelegate>( |
271 "TaskSchedulerSingleThreadWorker%d%s", id, params.name().c_str())); | 262 StringPrintf("TaskScheduler%sSingleThread%d", name.c_str(), id)); |
272 workers_.emplace_back(SchedulerWorker::Create( | 263 workers_.emplace_back( |
273 params.priority_hint(), std::move(delegate), task_tracker_, | 264 SchedulerWorker::Create(priority_hint, std::move(delegate), task_tracker_, |
274 SchedulerWorker::InitialState::DETACHED)); | 265 SchedulerWorker::InitialState::DETACHED)); |
275 return workers_.back().get(); | 266 return workers_.back().get(); |
276 } | 267 } |
277 | 268 |
278 void SchedulerSingleThreadTaskRunnerManager::UnregisterSchedulerWorker( | 269 void SchedulerSingleThreadTaskRunnerManager::UnregisterSchedulerWorker( |
279 SchedulerWorker* worker) { | 270 SchedulerWorker* worker) { |
280 // Cleanup uses a SchedulerLock, so call Cleanup() after releasing | 271 // Cleanup uses a SchedulerLock, so call Cleanup() after releasing |
281 // |workers_lock_|. | 272 // |workers_lock_|. |
282 scoped_refptr<SchedulerWorker> worker_to_destroy; | 273 scoped_refptr<SchedulerWorker> worker_to_destroy; |
283 { | 274 { |
284 AutoSchedulerLock auto_lock(workers_lock_); | 275 AutoSchedulerLock auto_lock(workers_lock_); |
(...skipping 14 matching lines...) Expand all Loading... |
299 }); | 290 }); |
300 DCHECK(worker_iter != workers_.end()); | 291 DCHECK(worker_iter != workers_.end()); |
301 worker_to_destroy = std::move(*worker_iter); | 292 worker_to_destroy = std::move(*worker_iter); |
302 workers_.erase(worker_iter); | 293 workers_.erase(worker_iter); |
303 } | 294 } |
304 worker_to_destroy->Cleanup(); | 295 worker_to_destroy->Cleanup(); |
305 } | 296 } |
306 | 297 |
307 } // namespace internal | 298 } // namespace internal |
308 } // namespace base | 299 } // namespace base |
OLD | NEW |