1 // Copyright (c) 2011 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.
5 // CancelableCallback is a wrapper around base::Callback that allows
6 // cancellation of a callback. CancelableCallback takes a reference on the
7 // wrapped callback until this object is destroyed or Reset()/Cancel() are
12 // Calling CancelableCallback::Cancel() brings the object back to its natural,
13 // default-constructed state, i.e., CancelableCallback::callback() will return
18 // CancelableCallback objects must be created on, posted to, cancelled on, and
19 // destroyed on the same thread.
24 // In the following example, the test is verifying that RunIntensiveTest()
25 // Quit()s the message loop within 4 seconds. The cancelable callback is posted
26 // to the message loop, the intensive test runs, the message loop is run,
27 // then the callback is cancelled.
29 // void TimeoutCallback(const std::string& timeout_message) {
30 // FAIL() << timeout_message;
31 // MessageLoop::current()->QuitWhenIdle();
34 // CancelableClosure timeout(base::Bind(&TimeoutCallback, "Test timed out."));
35 // MessageLoop::current()->PostDelayedTask(FROM_HERE, timeout.callback(),
36 // 4000) // 4 seconds to run.
37 // RunIntensiveTest();
38 // MessageLoop::current()->Run();
39 // timeout.Cancel(); // Hopefully this is hit before the timeout callback runs.
42 #ifndef BASE_CANCELABLE_CALLBACK_H_
43 #define BASE_CANCELABLE_CALLBACK_H_
45 #include "base/base_export.h"
46 #include "base/bind.h"
47 #include "base/callback.h"
48 #include "base/callback_internal.h"
49 #include "base/compiler_specific.h"
50 #include "base/logging.h"
51 #include "base/memory/weak_ptr.h"
55 template <typename Sig
>
56 class CancelableCallback
;
59 class CancelableCallback
<void(void)> {
61 CancelableCallback() : weak_factory_(this) {}
63 // |callback| must not be null.
64 explicit CancelableCallback(const base::Callback
<void(void)>& callback
)
65 : weak_factory_(this),
67 DCHECK(!callback
.is_null());
68 InitializeForwarder();
71 ~CancelableCallback() {}
73 // Cancels and drops the reference to the wrapped callback.
75 weak_factory_
.InvalidateWeakPtrs();
80 // Returns true if the wrapped callback has been cancelled.
81 bool IsCancelled() const {
82 return callback_
.is_null();
85 // Sets |callback| as the closure that may be cancelled. |callback| may not
86 // be null. Outstanding and any previously wrapped callbacks are cancelled.
87 void Reset(const base::Callback
<void(void)>& callback
) {
88 DCHECK(!callback
.is_null());
90 // Outstanding tasks (e.g., posted to a message loop) must not be called.
93 // |forwarder_| is no longer valid after Cancel(), so re-bind.
94 InitializeForwarder();
99 // Returns a callback that can be disabled by calling Cancel().
100 const base::Callback
<void(void)>& callback() const {
109 // Helper method to bind |forwarder_| using a weak pointer from
111 void InitializeForwarder() {
112 forwarder_
= base::Bind(&CancelableCallback
<void(void)>::Forward
,
113 weak_factory_
.GetWeakPtr());
116 // Used to ensure Forward() is not run when this object is destroyed.
117 base::WeakPtrFactory
<CancelableCallback
<void(void)> > weak_factory_
;
119 // The wrapper closure.
120 base::Callback
<void(void)> forwarder_
;
122 // The stored closure that may be cancelled.
123 base::Callback
<void(void)> callback_
;
125 DISALLOW_COPY_AND_ASSIGN(CancelableCallback
);
128 template <typename A1
>
129 class CancelableCallback
<void(A1
)> {
131 CancelableCallback() : weak_factory_(this) {}
133 // |callback| must not be null.
134 explicit CancelableCallback(const base::Callback
<void(A1
)>& callback
)
135 : weak_factory_(this),
136 callback_(callback
) {
137 DCHECK(!callback
.is_null());
138 InitializeForwarder();
141 ~CancelableCallback() {}
143 // Cancels and drops the reference to the wrapped callback.
145 weak_factory_
.InvalidateWeakPtrs();
150 // Returns true if the wrapped callback has been cancelled.
151 bool IsCancelled() const {
152 return callback_
.is_null();
155 // Sets |callback| as the closure that may be cancelled. |callback| may not
156 // be null. Outstanding and any previously wrapped callbacks are cancelled.
157 void Reset(const base::Callback
<void(A1
)>& callback
) {
158 DCHECK(!callback
.is_null());
160 // Outstanding tasks (e.g., posted to a message loop) must not be called.
163 // |forwarder_| is no longer valid after Cancel(), so re-bind.
164 InitializeForwarder();
166 callback_
= callback
;
169 // Returns a callback that can be disabled by calling Cancel().
170 const base::Callback
<void(A1
)>& callback() const {
175 void Forward(A1 a1
) const {
179 // Helper method to bind |forwarder_| using a weak pointer from
181 void InitializeForwarder() {
182 forwarder_
= base::Bind(&CancelableCallback
<void(A1
)>::Forward
,
183 weak_factory_
.GetWeakPtr());
186 // Used to ensure Forward() is not run when this object is destroyed.
187 base::WeakPtrFactory
<CancelableCallback
<void(A1
)> > weak_factory_
;
189 // The wrapper closure.
190 base::Callback
<void(A1
)> forwarder_
;
192 // The stored closure that may be cancelled.
193 base::Callback
<void(A1
)> callback_
;
195 DISALLOW_COPY_AND_ASSIGN(CancelableCallback
);
198 template <typename A1
, typename A2
>
199 class CancelableCallback
<void(A1
, A2
)> {
201 CancelableCallback() : weak_factory_(this) {}
203 // |callback| must not be null.
204 explicit CancelableCallback(const base::Callback
<void(A1
, A2
)>& callback
)
205 : weak_factory_(this),
206 callback_(callback
) {
207 DCHECK(!callback
.is_null());
208 InitializeForwarder();
211 ~CancelableCallback() {}
213 // Cancels and drops the reference to the wrapped callback.
215 weak_factory_
.InvalidateWeakPtrs();
220 // Returns true if the wrapped callback has been cancelled.
221 bool IsCancelled() const {
222 return callback_
.is_null();
225 // Sets |callback| as the closure that may be cancelled. |callback| may not
226 // be null. Outstanding and any previously wrapped callbacks are cancelled.
227 void Reset(const base::Callback
<void(A1
, A2
)>& callback
) {
228 DCHECK(!callback
.is_null());
230 // Outstanding tasks (e.g., posted to a message loop) must not be called.
233 // |forwarder_| is no longer valid after Cancel(), so re-bind.
234 InitializeForwarder();
236 callback_
= callback
;
239 // Returns a callback that can be disabled by calling Cancel().
240 const base::Callback
<void(A1
, A2
)>& callback() const {
245 void Forward(A1 a1
, A2 a2
) const {
246 callback_
.Run(a1
, a2
);
249 // Helper method to bind |forwarder_| using a weak pointer from
251 void InitializeForwarder() {
252 forwarder_
= base::Bind(&CancelableCallback
<void(A1
, A2
)>::Forward
,
253 weak_factory_
.GetWeakPtr());
256 // The wrapper closure.
257 base::Callback
<void(A1
, A2
)> forwarder_
;
259 // The stored closure that may be cancelled.
260 base::Callback
<void(A1
, A2
)> callback_
;
262 // Used to ensure Forward() is not run when this object is destroyed.
263 base::WeakPtrFactory
<CancelableCallback
<void(A1
, A2
)> > weak_factory_
;
264 DISALLOW_COPY_AND_ASSIGN(CancelableCallback
);
267 typedef CancelableCallback
<void(void)> CancelableClosure
;
271 #endif // BASE_CANCELABLE_CALLBACK_H_