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

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

Issue 2749303002: [reference - do not submit] Always create four pools in TaskSchedulerImpl. (Closed)
Patch Set: rebase Created 3 years, 9 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"
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_
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