| 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 12 matching lines...) Expand all Loading... |
| 23 const WorkerPoolIndexForTraitsCallback& | 23 const WorkerPoolIndexForTraitsCallback& |
| 24 worker_pool_index_for_traits_callback) { | 24 worker_pool_index_for_traits_callback) { |
| 25 std::unique_ptr<TaskSchedulerImpl> scheduler( | 25 std::unique_ptr<TaskSchedulerImpl> scheduler( |
| 26 new TaskSchedulerImpl(worker_pool_index_for_traits_callback)); | 26 new TaskSchedulerImpl(worker_pool_index_for_traits_callback)); |
| 27 scheduler->Initialize(worker_pools); | 27 scheduler->Initialize(worker_pools); |
| 28 return scheduler; | 28 return scheduler; |
| 29 } | 29 } |
| 30 | 30 |
| 31 TaskSchedulerImpl::~TaskSchedulerImpl() { | 31 TaskSchedulerImpl::~TaskSchedulerImpl() { |
| 32 #if DCHECK_IS_ON() | 32 #if DCHECK_IS_ON() |
| 33 DCHECK(join_for_testing_returned_.IsSignaled()); | 33 DCHECK(join_for_testing_returned_.IsSet()); |
| 34 #endif | 34 #endif |
| 35 } | 35 } |
| 36 | 36 |
| 37 void TaskSchedulerImpl::PostTaskWithTraits( | 37 void TaskSchedulerImpl::PostTaskWithTraits( |
| 38 const tracked_objects::Location& from_here, | 38 const tracked_objects::Location& from_here, |
| 39 const TaskTraits& traits, | 39 const TaskTraits& traits, |
| 40 const Closure& task) { | 40 const Closure& task) { |
| 41 // Post |task| as part of a one-off single-task Sequence. | 41 // Post |task| as part of a one-off single-task Sequence. |
| 42 GetWorkerPoolForTraits(traits)->PostTaskWithSequence( | 42 GetWorkerPoolForTraits(traits)->PostTaskWithSequence( |
| 43 WrapUnique(new Task(from_here, task, traits, TimeDelta())), | 43 WrapUnique(new Task(from_here, task, traits, TimeDelta())), |
| 44 make_scoped_refptr(new Sequence), nullptr); | 44 make_scoped_refptr(new Sequence), nullptr); |
| 45 } | 45 } |
| 46 | 46 |
| 47 scoped_refptr<TaskRunner> TaskSchedulerImpl::CreateTaskRunnerWithTraits( | 47 scoped_refptr<TaskRunner> TaskSchedulerImpl::CreateTaskRunnerWithTraits( |
| 48 const TaskTraits& traits, | 48 const TaskTraits& traits, |
| 49 ExecutionMode execution_mode) { | 49 ExecutionMode execution_mode) { |
| 50 return GetWorkerPoolForTraits(traits)->CreateTaskRunnerWithTraits( | 50 return GetWorkerPoolForTraits(traits)->CreateTaskRunnerWithTraits( |
| 51 traits, execution_mode); | 51 traits, execution_mode); |
| 52 } | 52 } |
| 53 | 53 |
| 54 void TaskSchedulerImpl::Shutdown() { | 54 void TaskSchedulerImpl::Shutdown() { |
| 55 // TODO(fdoray): Increase the priority of BACKGROUND tasks blocking shutdown. | 55 // TODO(fdoray): Increase the priority of BACKGROUND tasks blocking shutdown. |
| 56 task_tracker_.Shutdown(); | 56 task_tracker_.Shutdown(); |
| 57 } | 57 } |
| 58 | 58 |
| 59 void TaskSchedulerImpl::JoinForTesting() { | 59 void TaskSchedulerImpl::JoinForTesting() { |
| 60 #if DCHECK_IS_ON() | 60 #if DCHECK_IS_ON() |
| 61 DCHECK(!join_for_testing_returned_.IsSignaled()); | 61 DCHECK(!join_for_testing_returned_.IsSet()); |
| 62 #endif | 62 #endif |
| 63 for (const auto& worker_pool : worker_pools_) | 63 for (const auto& worker_pool : worker_pools_) |
| 64 worker_pool->JoinForTesting(); | 64 worker_pool->JoinForTesting(); |
| 65 service_thread_->JoinForTesting(); | 65 service_thread_->JoinForTesting(); |
| 66 #if DCHECK_IS_ON() | 66 #if DCHECK_IS_ON() |
| 67 join_for_testing_returned_.Signal(); | 67 join_for_testing_returned_.Set(); |
| 68 #endif | 68 #endif |
| 69 } | 69 } |
| 70 | 70 |
| 71 TaskSchedulerImpl::TaskSchedulerImpl(const WorkerPoolIndexForTraitsCallback& | 71 TaskSchedulerImpl::TaskSchedulerImpl(const WorkerPoolIndexForTraitsCallback& |
| 72 worker_pool_index_for_traits_callback) | 72 worker_pool_index_for_traits_callback) |
| 73 : delayed_task_manager_( | 73 : delayed_task_manager_( |
| 74 Bind(&TaskSchedulerImpl::OnDelayedRunTimeUpdated, Unretained(this))), | 74 Bind(&TaskSchedulerImpl::OnDelayedRunTimeUpdated, Unretained(this))), |
| 75 worker_pool_index_for_traits_callback_( | 75 worker_pool_index_for_traits_callback_( |
| 76 worker_pool_index_for_traits_callback) | 76 worker_pool_index_for_traits_callback) |
| 77 #if DCHECK_IS_ON() | |
| 78 , | |
| 79 join_for_testing_returned_(WaitableEvent::ResetPolicy::MANUAL, | |
| 80 WaitableEvent::InitialState::NOT_SIGNALED) | |
| 81 #endif | |
| 82 { | 77 { |
| 83 DCHECK(!worker_pool_index_for_traits_callback_.is_null()); | 78 DCHECK(!worker_pool_index_for_traits_callback_.is_null()); |
| 84 } | 79 } |
| 85 | 80 |
| 86 void TaskSchedulerImpl::Initialize( | 81 void TaskSchedulerImpl::Initialize( |
| 87 const std::vector<WorkerPoolCreationArgs>& worker_pools) { | 82 const std::vector<WorkerPoolCreationArgs>& worker_pools) { |
| 88 DCHECK(!worker_pools.empty()); | 83 DCHECK(!worker_pools.empty()); |
| 89 | 84 |
| 90 const SchedulerWorkerPoolImpl::ReEnqueueSequenceCallback | 85 const SchedulerWorkerPoolImpl::ReEnqueueSequenceCallback |
| 91 re_enqueue_sequence_callback = | 86 re_enqueue_sequence_callback = |
| (...skipping 37 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 |