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

Side by Side Diff: components/scheduler/base/task_queue_manager.h

Issue 2118903002: scheduler: Move the Blink scheduler into Blink (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: Rebased Created 4 years, 4 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
OLDNEW
(Empty)
1 // Copyright 2014 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 #ifndef CONTENT_RENDERER_SCHEDULER_BASE_TASK_QUEUE_MANAGER_H_
6 #define CONTENT_RENDERER_SCHEDULER_BASE_TASK_QUEUE_MANAGER_H_
7
8 #include <map>
9
10 #include "base/atomic_sequence_num.h"
11 #include "base/debug/task_annotator.h"
12 #include "base/macros.h"
13 #include "base/memory/weak_ptr.h"
14 #include "base/message_loop/message_loop.h"
15 #include "base/pending_task.h"
16 #include "base/synchronization/lock.h"
17 #include "base/threading/thread_checker.h"
18 #include "components/scheduler/base/enqueue_order.h"
19 #include "components/scheduler/base/task_queue_impl.h"
20 #include "components/scheduler/base/task_queue_selector.h"
21 #include "components/scheduler/scheduler_export.h"
22
23 namespace base {
24 class TickClock;
25
26 namespace trace_event {
27 class ConvertableToTraceFormat;
28 class TracedValue;
29 } // namespace trace_event
30 } // namespace base
31
32 namespace scheduler {
33 namespace internal {
34 class TaskQueueImpl;
35 } // namespace internal
36
37 class LazyNow;
38 class RealTimeDomain;
39 class TimeDomain;
40 class TaskQueueManagerDelegate;
41 class TaskTimeTracker;
42
43 // The task queue manager provides N task queues and a selector interface for
44 // choosing which task queue to service next. Each task queue consists of two
45 // sub queues:
46 //
47 // 1. Incoming task queue. Tasks that are posted get immediately appended here.
48 // When a task is appended into an empty incoming queue, the task manager
49 // work function (DoWork) is scheduled to run on the main task runner.
50 //
51 // 2. Work queue. If a work queue is empty when DoWork() is entered, tasks from
52 // the incoming task queue (if any) are moved here. The work queues are
53 // registered with the selector as input to the scheduling decision.
54 //
55 class SCHEDULER_EXPORT TaskQueueManager
56 : public internal::TaskQueueSelector::Observer {
57 public:
58 // Create a task queue manager where |delegate| identifies the thread
59 // on which where the tasks are eventually run. Category strings must have
60 // application lifetime (statics or literals). They may not include " chars.
61 TaskQueueManager(scoped_refptr<TaskQueueManagerDelegate> delegate,
62 const char* tracing_category,
63 const char* disabled_by_default_tracing_category,
64 const char* disabled_by_default_verbose_tracing_category);
65 ~TaskQueueManager() override;
66
67 // Requests that a task to process work is posted on the main task runner.
68 // These tasks are de-duplicated in two buckets: main-thread and all other
69 // threads. This distinction is done to reduce the overehead from locks, we
70 // assume the main-thread path will be hot.
71 void MaybeScheduleImmediateWork(const tracked_objects::Location& from_here);
72
73 // Requests that a delayed task to process work is posted on the main task
74 // runner. These delayed tasks are de-duplicated. Must be called on the thread
75 // this class was created on.
76 void MaybeScheduleDelayedWork(const tracked_objects::Location& from_here,
77 base::TimeTicks now,
78 base::TimeDelta delay);
79
80 // Set the number of tasks executed in a single invocation of the task queue
81 // manager. Increasing the batch size can reduce the overhead of yielding
82 // back to the main message loop -- at the cost of potentially delaying other
83 // tasks posted to the main loop. The batch size is 1 by default.
84 void SetWorkBatchSize(int work_batch_size);
85
86 // When given a non-null TaskTimeTracker, the TaskQueueManager calls its
87 // ReportTaskTime method for every top level task. The task_time_tracker must
88 // outlive this object, or be removed via SetTaskTimeTracker(nullptr).
89 void SetTaskTimeTracker(TaskTimeTracker* task_time_tracker) {
90 task_time_tracker_ = task_time_tracker;
91 }
92
93 // These functions can only be called on the same thread that the task queue
94 // manager executes its tasks on.
95 void AddTaskObserver(base::MessageLoop::TaskObserver* task_observer);
96 void RemoveTaskObserver(base::MessageLoop::TaskObserver* task_observer);
97
98 // Returns true if any task from a monitored task queue was was run since the
99 // last call to GetAndClearSystemIsQuiescentBit.
100 bool GetAndClearSystemIsQuiescentBit();
101
102 // Creates a task queue with the given |spec|. Must be called on the thread
103 // this class was created on.
104 scoped_refptr<internal::TaskQueueImpl> NewTaskQueue(
105 const TaskQueue::Spec& spec);
106
107 class SCHEDULER_EXPORT Observer {
108 public:
109 virtual ~Observer() {}
110
111 // Called when |queue| is unregistered.
112 virtual void OnUnregisterTaskQueue(
113 const scoped_refptr<TaskQueue>& queue) = 0;
114
115 // Called when the manager tried to execute a task from a disabled
116 // queue. See TaskQueue::Spec::SetShouldReportWhenExecutionBlocked.
117 virtual void OnTriedToExecuteBlockedTask(const TaskQueue& queue,
118 const base::PendingTask& task) = 0;
119 };
120
121 // Called once to set the Observer. This function is called on the main
122 // thread. If |observer| is null, then no callbacks will occur.
123 // Note |observer| is expected to outlive the SchedulerHelper.
124 void SetObserver(Observer* observer);
125
126 // Returns the delegate used by the TaskQueueManager.
127 const scoped_refptr<TaskQueueManagerDelegate>& delegate() const;
128
129 // Time domains must be registered for the task queues to get updated.
130 void RegisterTimeDomain(TimeDomain* time_domain);
131 void UnregisterTimeDomain(TimeDomain* time_domain);
132
133 RealTimeDomain* real_time_domain() const { return real_time_domain_.get(); }
134
135 LazyNow CreateLazyNow() const;
136
137 // Returns the currently executing TaskQueue if any. Must be called on the
138 // thread this class was created on.
139 TaskQueue* currently_executing_task_queue() const {
140 DCHECK(main_thread_checker_.CalledOnValidThread());
141 return currently_executing_task_queue_;
142 }
143
144 private:
145 friend class LazyNow;
146 friend class internal::TaskQueueImpl;
147 friend class TaskQueueManagerTest;
148
149 class DeletionSentinel : public base::RefCounted<DeletionSentinel> {
150 private:
151 friend class base::RefCounted<DeletionSentinel>;
152 ~DeletionSentinel() {}
153 };
154
155 // Unregisters a TaskQueue previously created by |NewTaskQueue()|.
156 // NOTE we have to flush the queue from |newly_updatable_| which means as a
157 // side effect MoveNewlyUpdatableQueuesIntoUpdatableQueueSet is called by this
158 // function.
159 void UnregisterTaskQueue(scoped_refptr<internal::TaskQueueImpl> task_queue);
160
161 // TaskQueueSelector::Observer implementation:
162 void OnTaskQueueEnabled(internal::TaskQueueImpl* queue) override;
163 void OnTriedToSelectBlockedWorkQueue(
164 internal::WorkQueue* work_queue) override;
165
166 // Called by the task queue to register a new pending task.
167 void DidQueueTask(const internal::TaskQueueImpl::Task& pending_task);
168
169 // Use the selector to choose a pending task and run it.
170 void DoWork(base::TimeTicks run_time, bool from_main_thread);
171
172 // Delayed Tasks with run_times <= Now() are enqueued onto the work queue.
173 // Reloads any empty work queues which have automatic pumping enabled and
174 // which are eligible to be auto pumped based on the |previous_task| which was
175 // run and |should_trigger_wakeup|. Call with an empty |previous_task| if no
176 // task was just run.
177 void UpdateWorkQueues(bool should_trigger_wakeup,
178 const internal::TaskQueueImpl::Task* previous_task,
179 LazyNow lazy_now);
180
181 // Chooses the next work queue to service. Returns true if |out_queue|
182 // indicates the queue from which the next task should be run, false to
183 // avoid running any tasks.
184 bool SelectWorkQueueToService(internal::WorkQueue** out_work_queue);
185
186 // Runs a single nestable task from the |queue|. On exit, |out_task| will
187 // contain the task which was executed. Non-nestable task are reposted on the
188 // run loop. The queue must not be empty.
189 enum class ProcessTaskResult {
190 DEFERRED,
191 EXECUTED,
192 TASK_QUEUE_MANAGER_DELETED
193 };
194 ProcessTaskResult ProcessTaskFromWorkQueue(
195 internal::WorkQueue* work_queue,
196 internal::TaskQueueImpl::Task* out_previous_task);
197
198 bool RunsTasksOnCurrentThread() const;
199 bool PostNonNestableDelayedTask(const tracked_objects::Location& from_here,
200 const base::Closure& task,
201 base::TimeDelta delay);
202
203 internal::EnqueueOrder GetNextSequenceNumber();
204
205 // Calls MaybeAdvanceTime on all time domains and returns true if one of them
206 // was able to advance.
207 bool TryAdvanceTimeDomains();
208
209 void MaybeRecordTaskDelayHistograms(
210 const internal::TaskQueueImpl::Task& pending_task,
211 const internal::TaskQueueImpl* queue);
212
213 std::unique_ptr<base::trace_event::ConvertableToTraceFormat>
214 AsValueWithSelectorResult(bool should_run,
215 internal::WorkQueue* selected_work_queue) const;
216
217 std::set<TimeDomain*> time_domains_;
218 std::unique_ptr<RealTimeDomain> real_time_domain_;
219
220 std::set<scoped_refptr<internal::TaskQueueImpl>> queues_;
221
222 // We have to be careful when deleting a queue because some of the code uses
223 // raw pointers and doesn't expect the rug to be pulled out from underneath.
224 std::set<scoped_refptr<internal::TaskQueueImpl>> queues_to_delete_;
225
226 internal::EnqueueOrderGenerator enqueue_order_generator_;
227 base::debug::TaskAnnotator task_annotator_;
228
229 base::ThreadChecker main_thread_checker_;
230 scoped_refptr<TaskQueueManagerDelegate> delegate_;
231 internal::TaskQueueSelector selector_;
232
233 base::Closure from_main_thread_immediate_do_work_closure_;
234 base::Closure from_other_thread_immediate_do_work_closure_;
235
236 bool task_was_run_on_quiescence_monitored_queue_;
237
238 // To reduce locking overhead we track pending calls to DoWork separately for
239 // the main thread and other threads.
240 std::set<base::TimeTicks> main_thread_pending_wakeups_;
241
242 // Protects |other_thread_pending_wakeups_|.
243 mutable base::Lock other_thread_lock_;
244 std::set<base::TimeTicks> other_thread_pending_wakeups_;
245
246 int work_batch_size_;
247 size_t task_count_;
248
249 base::ObserverList<base::MessageLoop::TaskObserver> task_observers_;
250
251 TaskTimeTracker* task_time_tracker_; // NOT OWNED
252
253 const char* tracing_category_;
254 const char* disabled_by_default_tracing_category_;
255 const char* disabled_by_default_verbose_tracing_category_;
256
257 internal::TaskQueueImpl* currently_executing_task_queue_; // NOT OWNED
258
259 Observer* observer_; // NOT OWNED
260 scoped_refptr<DeletionSentinel> deletion_sentinel_;
261 base::WeakPtrFactory<TaskQueueManager> weak_factory_;
262
263 DISALLOW_COPY_AND_ASSIGN(TaskQueueManager);
264 };
265
266 } // namespace scheduler
267
268 #endif // CONTENT_RENDERER_SCHEDULER_BASE_TASK_QUEUE_MANAGER_H_
OLDNEW
« no previous file with comments | « components/scheduler/base/task_queue_impl.cc ('k') | components/scheduler/base/task_queue_manager.cc » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698