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 #ifndef BASE_TASK_SCHEDULER_TASK_SCHEDULER_IMPL_H_ | 5 #ifndef BASE_TASK_SCHEDULER_TASK_SCHEDULER_IMPL_H_ |
6 #define BASE_TASK_SCHEDULER_TASK_SCHEDULER_IMPL_H_ | 6 #define BASE_TASK_SCHEDULER_TASK_SCHEDULER_IMPL_H_ |
7 | 7 |
8 #include <memory> | 8 #include <memory> |
9 #include <vector> | 9 #include <vector> |
10 | 10 |
11 #include "base/base_export.h" | 11 #include "base/base_export.h" |
12 #include "base/callback.h" | |
13 #include "base/logging.h" | 12 #include "base/logging.h" |
14 #include "base/macros.h" | 13 #include "base/macros.h" |
15 #include "base/memory/ref_counted.h" | 14 #include "base/memory/ref_counted.h" |
16 #include "base/synchronization/atomic_flag.h" | 15 #include "base/synchronization/atomic_flag.h" |
17 #include "base/task_scheduler/scheduler_worker_pool_impl.h" | 16 #include "base/task_scheduler/scheduler_worker_pool_impl.h" |
18 #include "base/task_scheduler/sequence.h" | 17 #include "base/task_scheduler/sequence.h" |
19 #include "base/task_scheduler/task_scheduler.h" | 18 #include "base/task_scheduler/task_scheduler.h" |
20 #include "base/task_scheduler/task_traits.h" | 19 #include "base/task_scheduler/task_traits.h" |
21 #include "base/threading/thread.h" | 20 #include "base/threading/thread.h" |
22 | 21 |
23 namespace base { | 22 namespace base { |
24 | 23 |
25 class HistogramBase; | 24 class HistogramBase; |
26 class SchedulerWorkerPoolParams; | |
27 | 25 |
28 namespace internal { | 26 namespace internal { |
29 | 27 |
30 class DelayedTaskManager; | 28 class DelayedTaskManager; |
31 class SchedulerSingleThreadTaskRunnerManager; | 29 class SchedulerSingleThreadTaskRunnerManager; |
32 class TaskTracker; | 30 class TaskTracker; |
33 | 31 |
34 // Default TaskScheduler implementation. This class is thread-safe. | 32 // Default TaskScheduler implementation. This class is thread-safe. |
35 class BASE_EXPORT TaskSchedulerImpl : public TaskScheduler { | 33 class BASE_EXPORT TaskSchedulerImpl : public TaskScheduler { |
36 public: | 34 public: |
37 // Creates and returns an initialized TaskSchedulerImpl. CHECKs on failure. | 35 // Creates and returns an initialized TaskSchedulerImpl. CHECKs on failure. |
38 // |worker_pool_params_vector| describes the worker pools to create. | 36 // |name| is used to label threads and histograms. It should identify the |
39 // |worker_pool_index_for_traits_callback| returns the index in |worker_pools| | 37 // component that creates the TaskScheduler. |init_params| is used to |
40 // of the worker pool in which a task with given traits should run. | 38 // initialize the worker pools. |
41 static std::unique_ptr<TaskSchedulerImpl> Create( | 39 static std::unique_ptr<TaskSchedulerImpl> Create( |
42 const std::vector<SchedulerWorkerPoolParams>& worker_pool_params_vector, | 40 const std::string& name, |
43 const WorkerPoolIndexForTraitsCallback& | 41 const TaskSchedulerInitParams& init_params); |
44 worker_pool_index_for_traits_callback); | |
45 | 42 |
46 ~TaskSchedulerImpl() override; | 43 ~TaskSchedulerImpl() override; |
47 | 44 |
48 // TaskScheduler: | 45 // TaskScheduler: |
49 void PostDelayedTaskWithTraits(const tracked_objects::Location& from_here, | 46 void PostDelayedTaskWithTraits(const tracked_objects::Location& from_here, |
50 const TaskTraits& traits, | 47 const TaskTraits& traits, |
51 const Closure& task, | 48 const Closure& task, |
52 TimeDelta delay) override; | 49 TimeDelta delay) override; |
53 scoped_refptr<TaskRunner> CreateTaskRunnerWithTraits( | 50 scoped_refptr<TaskRunner> CreateTaskRunnerWithTraits( |
54 const TaskTraits& traits) override; | 51 const TaskTraits& traits) override; |
55 scoped_refptr<SequencedTaskRunner> CreateSequencedTaskRunnerWithTraits( | 52 scoped_refptr<SequencedTaskRunner> CreateSequencedTaskRunnerWithTraits( |
56 const TaskTraits& traits) override; | 53 const TaskTraits& traits) override; |
57 scoped_refptr<SingleThreadTaskRunner> CreateSingleThreadTaskRunnerWithTraits( | 54 scoped_refptr<SingleThreadTaskRunner> CreateSingleThreadTaskRunnerWithTraits( |
58 const TaskTraits& traits) override; | 55 const TaskTraits& traits) override; |
59 std::vector<const HistogramBase*> GetHistograms() const override; | 56 std::vector<const HistogramBase*> GetHistograms() const override; |
60 int GetMaxConcurrentTasksWithTraitsDeprecated( | 57 int GetMaxConcurrentTasksWithTraitsDeprecated( |
61 const TaskTraits& traits) const override; | 58 const TaskTraits& traits) const override; |
62 void Shutdown() override; | 59 void Shutdown() override; |
63 void FlushForTesting() override; | 60 void FlushForTesting() override; |
64 void JoinForTesting() override; | 61 void JoinForTesting() override; |
65 | 62 |
66 private: | 63 private: |
67 explicit TaskSchedulerImpl(const WorkerPoolIndexForTraitsCallback& | 64 TaskSchedulerImpl(const std::string& name); |
68 worker_pool_index_for_traits_callback); | |
69 | 65 |
70 void Initialize( | 66 void Initialize(const TaskSchedulerInitParams& init_params); |
71 const std::vector<SchedulerWorkerPoolParams>& worker_pool_params_vector); | |
72 | 67 |
73 // Returns the worker pool that runs Tasks with |traits|. | 68 // Returns the worker pool that runs Tasks with |traits|. |
74 SchedulerWorkerPoolImpl* GetWorkerPoolForTraits( | 69 SchedulerWorkerPoolImpl* GetWorkerPoolForTraits( |
75 const TaskTraits& traits) const; | 70 const TaskTraits& traits) const; |
76 | 71 |
77 // Callback invoked when a non-single-thread |sequence| isn't empty after a | 72 // Callback invoked when a non-single-thread |sequence| isn't empty after a |
78 // worker pops a Task from it. | 73 // worker pops a Task from it. |
79 void ReEnqueueSequenceCallback(scoped_refptr<Sequence> sequence); | 74 void ReEnqueueSequenceCallback(scoped_refptr<Sequence> sequence); |
80 | 75 |
| 76 const std::string name_; |
81 Thread service_thread_; | 77 Thread service_thread_; |
82 std::unique_ptr<TaskTracker> task_tracker_; | 78 std::unique_ptr<TaskTracker> task_tracker_; |
83 std::unique_ptr<DelayedTaskManager> delayed_task_manager_; | 79 std::unique_ptr<DelayedTaskManager> delayed_task_manager_; |
84 std::unique_ptr<SchedulerSingleThreadTaskRunnerManager> | 80 std::unique_ptr<SchedulerSingleThreadTaskRunnerManager> |
85 single_thread_task_runner_manager_; | 81 single_thread_task_runner_manager_; |
86 const WorkerPoolIndexForTraitsCallback worker_pool_index_for_traits_callback_; | 82 std::unique_ptr<SchedulerWorkerPoolImpl> worker_pools_[4]; |
87 std::vector<std::unique_ptr<SchedulerWorkerPoolImpl>> worker_pools_; | |
88 | 83 |
89 #if DCHECK_IS_ON() | 84 #if DCHECK_IS_ON() |
90 // Set once JoinForTesting() has returned. | 85 // Set once JoinForTesting() has returned. |
91 AtomicFlag join_for_testing_returned_; | 86 AtomicFlag join_for_testing_returned_; |
92 #endif | 87 #endif |
93 | 88 |
94 DISALLOW_COPY_AND_ASSIGN(TaskSchedulerImpl); | 89 DISALLOW_COPY_AND_ASSIGN(TaskSchedulerImpl); |
95 }; | 90 }; |
96 | 91 |
97 } // namespace internal | 92 } // namespace internal |
98 } // namespace base | 93 } // namespace base |
99 | 94 |
100 #endif // BASE_TASK_SCHEDULER_TASK_SCHEDULER_IMPL_H_ | 95 #endif // BASE_TASK_SCHEDULER_TASK_SCHEDULER_IMPL_H_ |
OLD | NEW |