• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
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
3 // found in the LICENSE file.
4 
5 #ifndef BASE_TASK_SCHEDULER_POST_TASK_H_
6 #define BASE_TASK_SCHEDULER_POST_TASK_H_
7 
8 #include <utility>
9 
10 #include "base/base_export.h"
11 #include "base/bind.h"
12 #include "base/callback.h"
13 #include "base/location.h"
14 #include "base/memory/ref_counted.h"
15 #include "base/post_task_and_reply_with_result_internal.h"
16 #include "base/sequenced_task_runner.h"
17 #include "base/single_thread_task_runner.h"
18 #include "base/task_runner.h"
19 #include "base/task_scheduler/single_thread_task_runner_thread_mode.h"
20 #include "base/task_scheduler/task_traits.h"
21 #include "base/time/time.h"
22 #include "build/build_config.h"
23 
24 namespace base {
25 
26 // This is the preferred interface to post tasks to the TaskScheduler.
27 //
28 // To post a simple one-off task with default traits:
29 //     PostTask(FROM_HERE, Bind(...));
30 //
31 // To post a high priority one-off task to respond to a user interaction:
32 //     PostTaskWithTraits(
33 //         FROM_HERE,
34 //         {TaskPriority::USER_BLOCKING},
35 //         Bind(...));
36 //
37 // To post tasks that must run in sequence with default traits:
38 //     scoped_refptr<SequencedTaskRunner> task_runner =
39 //         CreateSequencedTaskRunnerWithTraits(TaskTraits());
40 //     task_runner.PostTask(FROM_HERE, Bind(...));
41 //     task_runner.PostTask(FROM_HERE, Bind(...));
42 //
43 // To post tasks that may block, must run in sequence and can be skipped on
44 // shutdown:
45 //     scoped_refptr<SequencedTaskRunner> task_runner =
46 //         CreateSequencedTaskRunnerWithTraits(
47 //             {MayBlock(), TaskShutdownBehavior::SKIP_ON_SHUTDOWN});
48 //     task_runner.PostTask(FROM_HERE, Bind(...));
49 //     task_runner.PostTask(FROM_HERE, Bind(...));
50 //
51 // The default traits apply to tasks that:
52 //     (1) don't block (ref. MayBlock() and WithBaseSyncPrimitives()),
53 //     (2) prefer inheriting the current priority to specifying their own, and
54 //     (3) can either block shutdown or be skipped on shutdown
55 //         (TaskScheduler implementation is free to choose a fitting default).
56 // Explicit traits must be specified for tasks for which these loose
57 // requirements are not sufficient.
58 //
59 // Tasks posted through functions below will run on threads owned by the
60 // registered TaskScheduler (i.e. not on the main thread). Tasks posted through
61 // functions below with a delay may be coalesced (i.e. delays may be adjusted to
62 // reduce the number of wakeups and hence power consumption).
63 //
64 // Prerequisite: A TaskScheduler must have been registered for the current
65 // process via TaskScheduler::SetInstance() before the functions below are
66 // valid. This is typically done during the initialization phase in each
67 // process. If your code is not running in that phase, you most likely don't
68 // have to worry about this. You will encounter DCHECKs or nullptr dereferences
69 // if this is violated. For tests, prefer base::test::ScopedTaskEnvironment.
70 
71 // Posts |task| to the TaskScheduler. Calling this is equivalent to calling
72 // PostTaskWithTraits with plain TaskTraits.
73 BASE_EXPORT void PostTask(const Location& from_here, OnceClosure task);
74 
75 // Posts |task| to the TaskScheduler. |task| will not run before |delay|
76 // expires. Calling this is equivalent to calling PostDelayedTaskWithTraits with
77 // plain TaskTraits.
78 //
79 // Use PostDelayedTaskWithTraits to specify a BACKGROUND priority if the task
80 // doesn't have to run as soon as |delay| expires.
81 BASE_EXPORT void PostDelayedTask(const Location& from_here,
82                                  OnceClosure task,
83                                  TimeDelta delay);
84 
85 // Posts |task| to the TaskScheduler and posts |reply| on the caller's execution
86 // context (i.e. same sequence or thread and same TaskTraits if applicable) when
87 // |task| completes. Calling this is equivalent to calling
88 // PostTaskWithTraitsAndReply with plain TaskTraits. Can only be called when
89 // SequencedTaskRunnerHandle::IsSet().
90 BASE_EXPORT void PostTaskAndReply(const Location& from_here,
91                                   OnceClosure task,
92                                   OnceClosure reply);
93 
94 // Posts |task| to the TaskScheduler and posts |reply| with the return value of
95 // |task| as argument on the caller's execution context (i.e. same sequence or
96 // thread and same TaskTraits if applicable) when |task| completes. Calling this
97 // is equivalent to calling PostTaskWithTraitsAndReplyWithResult with plain
98 // TaskTraits. Can only be called when SequencedTaskRunnerHandle::IsSet().
99 template <typename TaskReturnType, typename ReplyArgType>
PostTaskAndReplyWithResult(const Location & from_here,OnceCallback<TaskReturnType ()> task,OnceCallback<void (ReplyArgType)> reply)100 void PostTaskAndReplyWithResult(const Location& from_here,
101                                 OnceCallback<TaskReturnType()> task,
102                                 OnceCallback<void(ReplyArgType)> reply) {
103   PostTaskWithTraitsAndReplyWithResult(from_here, TaskTraits(), std::move(task),
104                                        std::move(reply));
105 }
106 
107 // Callback version of PostTaskAndReplyWithResult above.
108 // Though RepeatingCallback is convertible to OnceCallback, we need this since
109 // we can not use template deduction and object conversion at once on the
110 // overload resolution.
111 // TODO(tzik): Update all callers of the Callback version to use OnceCallback.
112 template <typename TaskReturnType, typename ReplyArgType>
PostTaskAndReplyWithResult(const Location & from_here,Callback<TaskReturnType ()> task,Callback<void (ReplyArgType)> reply)113 void PostTaskAndReplyWithResult(const Location& from_here,
114                                 Callback<TaskReturnType()> task,
115                                 Callback<void(ReplyArgType)> reply) {
116   PostTaskAndReplyWithResult(
117       from_here, OnceCallback<TaskReturnType()>(std::move(task)),
118       OnceCallback<void(ReplyArgType)>(std::move(reply)));
119 }
120 
121 // Posts |task| with specific |traits| to the TaskScheduler.
122 BASE_EXPORT void PostTaskWithTraits(const Location& from_here,
123                                     const TaskTraits& traits,
124                                     OnceClosure task);
125 
126 // Posts |task| with specific |traits| to the TaskScheduler. |task| will not run
127 // before |delay| expires.
128 //
129 // Specify a BACKGROUND priority via |traits| if the task doesn't have to run as
130 // soon as |delay| expires.
131 BASE_EXPORT void PostDelayedTaskWithTraits(const Location& from_here,
132                                            const TaskTraits& traits,
133                                            OnceClosure task,
134                                            TimeDelta delay);
135 
136 // Posts |task| with specific |traits| to the TaskScheduler and posts |reply| on
137 // the caller's execution context (i.e. same sequence or thread and same
138 // TaskTraits if applicable) when |task| completes. Can only be called when
139 // SequencedTaskRunnerHandle::IsSet().
140 BASE_EXPORT void PostTaskWithTraitsAndReply(const Location& from_here,
141                                             const TaskTraits& traits,
142                                             OnceClosure task,
143                                             OnceClosure reply);
144 
145 // Posts |task| with specific |traits| to the TaskScheduler and posts |reply|
146 // with the return value of |task| as argument on the caller's execution context
147 // (i.e. same sequence or thread and same TaskTraits if applicable) when |task|
148 // completes. Can only be called when SequencedTaskRunnerHandle::IsSet().
149 template <typename TaskReturnType, typename ReplyArgType>
PostTaskWithTraitsAndReplyWithResult(const Location & from_here,const TaskTraits & traits,OnceCallback<TaskReturnType ()> task,OnceCallback<void (ReplyArgType)> reply)150 void PostTaskWithTraitsAndReplyWithResult(
151     const Location& from_here,
152     const TaskTraits& traits,
153     OnceCallback<TaskReturnType()> task,
154     OnceCallback<void(ReplyArgType)> reply) {
155   TaskReturnType* result = new TaskReturnType();
156   return PostTaskWithTraitsAndReply(
157       from_here, traits,
158       BindOnce(&internal::ReturnAsParamAdapter<TaskReturnType>, std::move(task),
159                result),
160       BindOnce(&internal::ReplyAdapter<TaskReturnType, ReplyArgType>,
161                std::move(reply), Owned(result)));
162 }
163 
164 // Callback version of PostTaskWithTraitsAndReplyWithResult above.
165 // Though RepeatingCallback is convertible to OnceCallback, we need this since
166 // we can not use template deduction and object conversion at once on the
167 // overload resolution.
168 // TODO(tzik): Update all callers of the Callback version to use OnceCallback.
169 template <typename TaskReturnType, typename ReplyArgType>
PostTaskWithTraitsAndReplyWithResult(const Location & from_here,const TaskTraits & traits,Callback<TaskReturnType ()> task,Callback<void (ReplyArgType)> reply)170 void PostTaskWithTraitsAndReplyWithResult(const Location& from_here,
171                                           const TaskTraits& traits,
172                                           Callback<TaskReturnType()> task,
173                                           Callback<void(ReplyArgType)> reply) {
174   PostTaskWithTraitsAndReplyWithResult(
175       from_here, traits, OnceCallback<TaskReturnType()>(std::move(task)),
176       OnceCallback<void(ReplyArgType)>(std::move(reply)));
177 }
178 
179 // Returns a TaskRunner whose PostTask invocations result in scheduling tasks
180 // using |traits|. Tasks may run in any order and in parallel.
181 BASE_EXPORT scoped_refptr<TaskRunner> CreateTaskRunnerWithTraits(
182     const TaskTraits& traits);
183 
184 // Returns a SequencedTaskRunner whose PostTask invocations result in scheduling
185 // tasks using |traits|. Tasks run one at a time in posting order.
186 BASE_EXPORT scoped_refptr<SequencedTaskRunner>
187 CreateSequencedTaskRunnerWithTraits(const TaskTraits& traits);
188 
189 // Returns a SingleThreadTaskRunner whose PostTask invocations result in
190 // scheduling tasks using |traits| on a thread determined by |thread_mode|. See
191 // base/task_scheduler/single_thread_task_runner_thread_mode.h for |thread_mode|
192 // details. Tasks run on a single thread in posting order.
193 //
194 // If all you need is to make sure that tasks don't run concurrently (e.g.
195 // because they access a data structure which is not thread-safe), use
196 // CreateSequencedTaskRunnerWithTraits(). Only use this if you rely on a thread-
197 // affine API (it might be safer to assume thread-affinity when dealing with
198 // under-documented third-party APIs, e.g. other OS') or share data across tasks
199 // using thread-local storage.
200 BASE_EXPORT scoped_refptr<SingleThreadTaskRunner>
201 CreateSingleThreadTaskRunnerWithTraits(
202     const TaskTraits& traits,
203     SingleThreadTaskRunnerThreadMode thread_mode =
204         SingleThreadTaskRunnerThreadMode::SHARED);
205 
206 #if defined(OS_WIN)
207 // Returns a SingleThreadTaskRunner whose PostTask invocations result in
208 // scheduling tasks using |traits| in a COM Single-Threaded Apartment on a
209 // thread determined by |thread_mode|. See
210 // base/task_scheduler/single_thread_task_runner_thread_mode.h for |thread_mode|
211 // details. Tasks run in the same Single-Threaded Apartment in posting order for
212 // the returned SingleThreadTaskRunner. There is not necessarily a one-to-one
213 // correspondence between SingleThreadTaskRunners and Single-Threaded
214 // Apartments. The implementation is free to share apartments or create new
215 // apartments as necessary. In either case, care should be taken to make sure
216 // COM pointers are not smuggled across apartments.
217 BASE_EXPORT scoped_refptr<SingleThreadTaskRunner>
218 CreateCOMSTATaskRunnerWithTraits(const TaskTraits& traits,
219                                  SingleThreadTaskRunnerThreadMode thread_mode =
220                                      SingleThreadTaskRunnerThreadMode::SHARED);
221 #endif  // defined(OS_WIN)
222 
223 }  // namespace base
224 
225 #endif  // BASE_TASK_SCHEDULER_POST_TASK_H_
226