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

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

Issue 2809163003: Always use TaskScheduler::InitParams to initialize a TaskScheduler. (Closed)
Patch Set: CR-gab-robliao-17-18 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.cc ('k') | base/task_scheduler/task_scheduler_impl.cc » ('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 #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 std::unique_ptr<SchedulerWorkerPoolImpl> worker_pools_[4];
robliao 2017/04/12 21:09:05 Nit: Document how we got 4.
fdoray 2017/04/19 12:21:21 Done.
105 std::vector<std::unique_ptr<SchedulerWorkerPoolImpl>> worker_pools_;
106 92
107 #if DCHECK_IS_ON() 93 #if DCHECK_IS_ON()
108 // Set once JoinForTesting() has returned. 94 // Set once JoinForTesting() has returned.
109 AtomicFlag join_for_testing_returned_; 95 AtomicFlag join_for_testing_returned_;
110 #endif 96 #endif
111 97
112 DISALLOW_COPY_AND_ASSIGN(TaskSchedulerImpl); 98 DISALLOW_COPY_AND_ASSIGN(TaskSchedulerImpl);
113 }; 99 };
114 100
115 } // namespace internal 101 } // namespace internal
116 } // namespace base 102 } // namespace base
117 103
118 #endif // BASE_TASK_SCHEDULER_TASK_SCHEDULER_IMPL_H_ 104 #endif // BASE_TASK_SCHEDULER_TASK_SCHEDULER_IMPL_H_
OLDNEW
« no previous file with comments | « base/task_scheduler/task_scheduler.cc ('k') | base/task_scheduler/task_scheduler_impl.cc » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698