| OLD | NEW |
| 1 // Copyright 2015 The Chromium Authors. All rights reserved. | 1 // Copyright 2015 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 "components/scheduler/child/scheduler_helper.h" | 5 #include "components/scheduler/child/scheduler_helper.h" |
| 6 | 6 |
| 7 #include "base/time/default_tick_clock.h" | 7 #include "base/time/default_tick_clock.h" |
| 8 #include "base/trace_event/trace_event.h" | 8 #include "base/trace_event/trace_event.h" |
| 9 #include "base/trace_event/trace_event_argument.h" | 9 #include "base/trace_event/trace_event_argument.h" |
| 10 #include "components/scheduler/base/task_queue_impl.h" | 10 #include "components/scheduler/base/task_queue_impl.h" |
| 11 #include "components/scheduler/child/scheduler_task_runner_delegate.h" | 11 #include "components/scheduler/child/scheduler_tqm_delegate.h" |
| 12 | 12 |
| 13 namespace scheduler { | 13 namespace scheduler { |
| 14 | 14 |
| 15 SchedulerHelper::SchedulerHelper( | 15 SchedulerHelper::SchedulerHelper( |
| 16 scoped_refptr<SchedulerTaskRunnerDelegate> main_task_runner, | 16 scoped_refptr<SchedulerTqmDelegate> task_queue_manager_delegate, |
| 17 const char* tracing_category, | 17 const char* tracing_category, |
| 18 const char* disabled_by_default_tracing_category, | 18 const char* disabled_by_default_tracing_category, |
| 19 const char* disabled_by_default_verbose_tracing_category) | 19 const char* disabled_by_default_verbose_tracing_category) |
| 20 : main_task_runner_(main_task_runner), | 20 : task_queue_manager_delegate_(task_queue_manager_delegate), |
| 21 task_queue_manager_( | 21 task_queue_manager_( |
| 22 new TaskQueueManager(main_task_runner, | 22 new TaskQueueManager(task_queue_manager_delegate, |
| 23 tracing_category, | 23 tracing_category, |
| 24 disabled_by_default_tracing_category, | 24 disabled_by_default_tracing_category, |
| 25 disabled_by_default_verbose_tracing_category)), | 25 disabled_by_default_verbose_tracing_category)), |
| 26 control_task_runner_(NewTaskQueue( | 26 control_task_runner_(NewTaskQueue( |
| 27 TaskQueue::Spec("control_tq") | 27 TaskQueue::Spec("control_tq") |
| 28 .SetWakeupPolicy(TaskQueue::WakeupPolicy::DONT_WAKE_OTHER_QUEUES) | 28 .SetWakeupPolicy(TaskQueue::WakeupPolicy::DONT_WAKE_OTHER_QUEUES) |
| 29 .SetShouldNotifyObservers(false))), | 29 .SetShouldNotifyObservers(false))), |
| 30 control_after_wakeup_task_runner_(NewTaskQueue( | 30 control_after_wakeup_task_runner_(NewTaskQueue( |
| 31 TaskQueue::Spec("control_after_wakeup_tq") | 31 TaskQueue::Spec("control_after_wakeup_tq") |
| 32 .SetPumpPolicy(TaskQueue::PumpPolicy::AFTER_WAKEUP) | 32 .SetPumpPolicy(TaskQueue::PumpPolicy::AFTER_WAKEUP) |
| 33 .SetWakeupPolicy(TaskQueue::WakeupPolicy::DONT_WAKE_OTHER_QUEUES) | 33 .SetWakeupPolicy(TaskQueue::WakeupPolicy::DONT_WAKE_OTHER_QUEUES) |
| 34 .SetShouldNotifyObservers(false))), | 34 .SetShouldNotifyObservers(false))), |
| 35 default_task_runner_(NewTaskQueue(TaskQueue::Spec("default_tq") | 35 default_task_runner_(NewTaskQueue(TaskQueue::Spec("default_tq") |
| 36 .SetShouldMonitorQuiescence(true))), | 36 .SetShouldMonitorQuiescence(true))), |
| 37 time_source_(new base::DefaultTickClock), | |
| 38 observer_(nullptr), | 37 observer_(nullptr), |
| 39 tracing_category_(tracing_category), | 38 tracing_category_(tracing_category), |
| 40 disabled_by_default_tracing_category_( | 39 disabled_by_default_tracing_category_( |
| 41 disabled_by_default_tracing_category) { | 40 disabled_by_default_tracing_category) { |
| 42 control_task_runner_->SetQueuePriority(TaskQueue::CONTROL_PRIORITY); | 41 control_task_runner_->SetQueuePriority(TaskQueue::CONTROL_PRIORITY); |
| 43 control_after_wakeup_task_runner_->SetQueuePriority( | 42 control_after_wakeup_task_runner_->SetQueuePriority( |
| 44 TaskQueue::CONTROL_PRIORITY); | 43 TaskQueue::CONTROL_PRIORITY); |
| 45 | 44 |
| 46 task_queue_manager_->SetWorkBatchSize(4); | 45 task_queue_manager_->SetWorkBatchSize(4); |
| 47 | 46 |
| 48 main_task_runner_->SetDefaultTaskRunner(default_task_runner_.get()); | 47 task_queue_manager_delegate_->SetDefaultTaskRunner( |
| 48 default_task_runner_.get()); |
| 49 } | 49 } |
| 50 | 50 |
| 51 SchedulerHelper::~SchedulerHelper() { | 51 SchedulerHelper::~SchedulerHelper() { |
| 52 Shutdown(); | 52 Shutdown(); |
| 53 } | 53 } |
| 54 | 54 |
| 55 void SchedulerHelper::Shutdown() { | 55 void SchedulerHelper::Shutdown() { |
| 56 CheckOnValidThread(); | 56 CheckOnValidThread(); |
| 57 if (task_queue_manager_) | 57 if (task_queue_manager_) |
| 58 task_queue_manager_->SetObserver(nullptr); | 58 task_queue_manager_->SetObserver(nullptr); |
| 59 task_queue_manager_.reset(); | 59 task_queue_manager_.reset(); |
| 60 main_task_runner_->RestoreDefaultTaskRunner(); | 60 task_queue_manager_delegate_->RestoreDefaultTaskRunner(); |
| 61 } | 61 } |
| 62 | 62 |
| 63 scoped_refptr<TaskQueue> SchedulerHelper::NewTaskQueue( | 63 scoped_refptr<TaskQueue> SchedulerHelper::NewTaskQueue( |
| 64 const TaskQueue::Spec& spec) { | 64 const TaskQueue::Spec& spec) { |
| 65 DCHECK(task_queue_manager_.get()); | 65 DCHECK(task_queue_manager_.get()); |
| 66 return task_queue_manager_->NewTaskQueue(spec); | 66 return task_queue_manager_->NewTaskQueue(spec); |
| 67 } | 67 } |
| 68 | 68 |
| 69 scoped_refptr<TaskQueue> SchedulerHelper::DefaultTaskRunner() { | 69 scoped_refptr<TaskQueue> SchedulerHelper::DefaultTaskRunner() { |
| 70 CheckOnValidThread(); | 70 CheckOnValidThread(); |
| 71 return default_task_runner_; | 71 return default_task_runner_; |
| 72 } | 72 } |
| 73 | 73 |
| 74 scoped_refptr<TaskQueue> SchedulerHelper::ControlTaskRunner() { | 74 scoped_refptr<TaskQueue> SchedulerHelper::ControlTaskRunner() { |
| 75 return control_task_runner_; | 75 return control_task_runner_; |
| 76 } | 76 } |
| 77 | 77 |
| 78 scoped_refptr<TaskQueue> SchedulerHelper::ControlAfterWakeUpTaskRunner() { | 78 scoped_refptr<TaskQueue> SchedulerHelper::ControlAfterWakeUpTaskRunner() { |
| 79 return control_after_wakeup_task_runner_; | 79 return control_after_wakeup_task_runner_; |
| 80 } | 80 } |
| 81 | 81 |
| 82 void SchedulerHelper::SetTimeSourceForTesting( | |
| 83 scoped_ptr<base::TickClock> time_source) { | |
| 84 CheckOnValidThread(); | |
| 85 time_source_ = time_source.Pass(); | |
| 86 } | |
| 87 | |
| 88 void SchedulerHelper::SetWorkBatchSizeForTesting(size_t work_batch_size) { | 82 void SchedulerHelper::SetWorkBatchSizeForTesting(size_t work_batch_size) { |
| 89 CheckOnValidThread(); | 83 CheckOnValidThread(); |
| 90 DCHECK(task_queue_manager_.get()); | 84 DCHECK(task_queue_manager_.get()); |
| 91 task_queue_manager_->SetWorkBatchSize(work_batch_size); | 85 task_queue_manager_->SetWorkBatchSize(work_batch_size); |
| 92 } | 86 } |
| 93 | 87 |
| 94 TaskQueueManager* SchedulerHelper::GetTaskQueueManagerForTesting() { | 88 TaskQueueManager* SchedulerHelper::GetTaskQueueManagerForTesting() { |
| 95 CheckOnValidThread(); | 89 CheckOnValidThread(); |
| 96 return task_queue_manager_.get(); | 90 return task_queue_manager_.get(); |
| 97 } | 91 } |
| 98 | 92 |
| 99 base::TimeTicks SchedulerHelper::Now() const { | 93 base::TickClock* SchedulerHelper::tick_clock() const { |
| 100 return time_source_->NowTicks(); | 94 return task_queue_manager_->tick_clock(); |
| 101 } | 95 } |
| 102 | 96 |
| 103 base::TimeTicks SchedulerHelper::NextPendingDelayedTaskRunTime() const { | 97 base::TimeTicks SchedulerHelper::NextPendingDelayedTaskRunTime() const { |
| 104 CheckOnValidThread(); | 98 CheckOnValidThread(); |
| 105 DCHECK(task_queue_manager_.get()); | 99 DCHECK(task_queue_manager_.get()); |
| 106 return task_queue_manager_->NextPendingDelayedTaskRunTime(); | 100 return task_queue_manager_->NextPendingDelayedTaskRunTime(); |
| 107 } | 101 } |
| 108 | 102 |
| 109 bool SchedulerHelper::GetAndClearSystemIsQuiescentBit() { | 103 bool SchedulerHelper::GetAndClearSystemIsQuiescentBit() { |
| 110 CheckOnValidThread(); | 104 CheckOnValidThread(); |
| (...skipping 22 matching lines...) Expand all Loading... |
| 133 task_queue_manager_->SetObserver(this); | 127 task_queue_manager_->SetObserver(this); |
| 134 } | 128 } |
| 135 | 129 |
| 136 void SchedulerHelper::OnUnregisterTaskQueue( | 130 void SchedulerHelper::OnUnregisterTaskQueue( |
| 137 const scoped_refptr<internal::TaskQueueImpl>& queue) { | 131 const scoped_refptr<internal::TaskQueueImpl>& queue) { |
| 138 if (observer_) | 132 if (observer_) |
| 139 observer_->OnUnregisterTaskQueue(queue); | 133 observer_->OnUnregisterTaskQueue(queue); |
| 140 } | 134 } |
| 141 | 135 |
| 142 } // namespace scheduler | 136 } // namespace scheduler |
| OLD | NEW |