Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(32)

Side by Side Diff: base/task_scheduler/task_scheduler_impl.cc

Issue 2816653003: Do not take a MessageLoopForIO in the constructor of TaskTrackerPosix. (Closed)
Patch Set: rebase Created 3 years, 8 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
« no previous file with comments | « base/task_scheduler/task_scheduler_impl.h ('k') | base/task_scheduler/task_tracker_posix.h » ('j') | no next file with comments »
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
OLDNEW
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
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
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
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
OLDNEW
« no previous file with comments | « base/task_scheduler/task_scheduler_impl.h ('k') | base/task_scheduler/task_tracker_posix.h » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698