123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156 |
- // Copyright (c) 2011 The Chromium Authors. All rights reserved.
- // Use of this source code is governed by a BSD-style license that can be
- // found in the LICENSE file.
- //
- // CancelableCallback is a wrapper around base::Callback that allows
- // cancellation of a callback. CancelableCallback takes a reference on the
- // wrapped callback until this object is destroyed or Reset()/Cancel() are
- // called.
- //
- // NOTE:
- //
- // Calling CancelableCallback::Cancel() brings the object back to its natural,
- // default-constructed state, i.e., CancelableCallback::callback() will return
- // a null callback.
- //
- // THREAD-SAFETY:
- //
- // CancelableCallback objects must be created on, posted to, cancelled on, and
- // destroyed on the same thread.
- //
- //
- // EXAMPLE USAGE:
- //
- // In the following example, the test is verifying that RunIntensiveTest()
- // Quit()s the message loop within 4 seconds. The cancelable callback is posted
- // to the message loop, the intensive test runs, the message loop is run,
- // then the callback is cancelled.
- //
- // RunLoop run_loop;
- //
- // void TimeoutCallback(const std::string& timeout_message) {
- // FAIL() << timeout_message;
- // run_loop.QuitWhenIdle();
- // }
- //
- // CancelableOnceClosure timeout(
- // base::BindOnce(&TimeoutCallback, "Test timed out."));
- // ThreadTaskRunnerHandle::Get()->PostDelayedTask(FROM_HERE, timeout.callback(),
- // TimeDelta::FromSeconds(4));
- // RunIntensiveTest();
- // run_loop.Run();
- // timeout.Cancel(); // Hopefully this is hit before the timeout callback runs.
- //
- #ifndef BASE_CANCELABLE_CALLBACK_H_
- #define BASE_CANCELABLE_CALLBACK_H_
- #include <utility>
- #include "base/base_export.h"
- #include "base/bind.h"
- #include "base/callback.h"
- #include "base/callback_internal.h"
- #include "base/check.h"
- #include "base/compiler_specific.h"
- #include "base/memory/weak_ptr.h"
- namespace base {
- namespace internal {
- template <typename CallbackType>
- class CancelableCallbackImpl {
- public:
- CancelableCallbackImpl() = default;
- CancelableCallbackImpl(const CancelableCallbackImpl&) = delete;
- CancelableCallbackImpl& operator=(const CancelableCallbackImpl&) = delete;
- // |callback| must not be null.
- explicit CancelableCallbackImpl(CallbackType callback)
- : callback_(std::move(callback)) {
- DCHECK(callback_);
- }
- ~CancelableCallbackImpl() = default;
- // Cancels and drops the reference to the wrapped callback.
- void Cancel() {
- weak_ptr_factory_.InvalidateWeakPtrs();
- callback_.Reset();
- }
- // Returns true if the wrapped callback has been cancelled.
- bool IsCancelled() const {
- return callback_.is_null();
- }
- // Sets |callback| as the closure that may be cancelled. |callback| may not
- // be null. Outstanding and any previously wrapped callbacks are cancelled.
- void Reset(CallbackType callback) {
- DCHECK(callback);
- // Outstanding tasks (e.g., posted to a message loop) must not be called.
- Cancel();
- callback_ = std::move(callback);
- }
- // Returns a callback that can be disabled by calling Cancel().
- CallbackType callback() const {
- if (!callback_)
- return CallbackType();
- CallbackType forwarder;
- MakeForwarder(&forwarder);
- return forwarder;
- }
- private:
- template <typename... Args>
- void MakeForwarder(RepeatingCallback<void(Args...)>* out) const {
- using ForwarderType = void (CancelableCallbackImpl::*)(Args...);
- ForwarderType forwarder = &CancelableCallbackImpl::ForwardRepeating;
- *out = BindRepeating(forwarder, weak_ptr_factory_.GetWeakPtr());
- }
- template <typename... Args>
- void MakeForwarder(OnceCallback<void(Args...)>* out) const {
- using ForwarderType = void (CancelableCallbackImpl::*)(Args...);
- ForwarderType forwarder = &CancelableCallbackImpl::ForwardOnce;
- *out = BindOnce(forwarder, weak_ptr_factory_.GetWeakPtr());
- }
- template <typename... Args>
- void ForwardRepeating(Args... args) {
- callback_.Run(std::forward<Args>(args)...);
- }
- template <typename... Args>
- void ForwardOnce(Args... args) {
- weak_ptr_factory_.InvalidateWeakPtrs();
- std::move(callback_).Run(std::forward<Args>(args)...);
- }
- // The stored closure that may be cancelled.
- CallbackType callback_;
- mutable base::WeakPtrFactory<CancelableCallbackImpl> weak_ptr_factory_{this};
- };
- } // namespace internal
- // Consider using base::WeakPtr directly instead of base::CancelableCallback for
- // the task cancellation.
- template <typename Signature>
- using CancelableOnceCallback =
- internal::CancelableCallbackImpl<OnceCallback<Signature>>;
- using CancelableOnceClosure = CancelableOnceCallback<void()>;
- template <typename Signature>
- using CancelableRepeatingCallback =
- internal::CancelableCallbackImpl<RepeatingCallback<Signature>>;
- using CancelableRepeatingClosure = CancelableRepeatingCallback<void()>;
- template <typename Signature>
- using CancelableCallback = CancelableRepeatingCallback<Signature>;
- using CancelableClosure = CancelableCallback<void()>;
- } // namespace base
- #endif // BASE_CANCELABLE_CALLBACK_H_
|