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