OLD | NEW |
1 // Copyright (c) 2015 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2015 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 MEDIA_BASE_BIND_TO_CURRENT_LOOP_H_ | 5 #ifndef MEDIA_BASE_BIND_TO_CURRENT_LOOP_H_ |
6 #define MEDIA_BASE_BIND_TO_CURRENT_LOOP_H_ | 6 #define MEDIA_BASE_BIND_TO_CURRENT_LOOP_H_ |
7 | 7 |
8 #include <memory> | 8 #include <memory> |
| 9 #include <utility> |
9 | 10 |
10 #include "base/bind.h" | 11 #include "base/bind.h" |
11 #include "base/location.h" | 12 #include "base/location.h" |
12 #include "base/memory/ptr_util.h" | 13 #include "base/memory/ptr_util.h" |
13 #include "base/single_thread_task_runner.h" | 14 #include "base/single_thread_task_runner.h" |
14 #include "base/threading/thread_task_runner_handle.h" | 15 #include "base/threading/thread_task_runner_handle.h" |
15 | 16 |
16 // This is a helper utility for base::Bind()ing callbacks to the current | 17 // This is a helper utility for base::Bind()ing callbacks to the current |
17 // MessageLoop. The typical use is when |a| (of class |A|) wants to hand a | 18 // MessageLoop. The typical use is when |a| (of class |A|) wants to hand a |
18 // callback such as base::Bind(&A::AMethod, a) to |b|, but needs to ensure that | 19 // callback such as base::Bind(&A::AMethod, a) to |b|, but needs to ensure that |
(...skipping 31 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
50 | 51 |
51 ~TrampolineHelper() { | 52 ~TrampolineHelper() { |
52 task_runner_->PostTask( | 53 task_runner_->PostTask( |
53 posted_from_, | 54 posted_from_, |
54 base::Bind(&TrampolineHelper::ClearCallbackOnTargetTaskRunner, | 55 base::Bind(&TrampolineHelper::ClearCallbackOnTargetTaskRunner, |
55 base::Passed(&callback_))); | 56 base::Passed(&callback_))); |
56 } | 57 } |
57 | 58 |
58 private: | 59 private: |
59 static void ClearCallbackOnTargetTaskRunner(CallbackType) {} | 60 static void ClearCallbackOnTargetTaskRunner(CallbackType) {} |
60 static void RunOnceClosure(base::OnceClosure cb) { std::move(cb).Run(); } | |
61 | 61 |
62 tracked_objects::Location posted_from_; | 62 tracked_objects::Location posted_from_; |
63 scoped_refptr<base::SequencedTaskRunner> task_runner_; | 63 scoped_refptr<base::SequencedTaskRunner> task_runner_; |
64 CallbackType callback_; | 64 CallbackType callback_; |
65 }; | 65 }; |
66 | 66 |
67 template <> | 67 template <> |
68 inline void TrampolineHelper<void()>::Run() { | 68 inline void TrampolineHelper<void()>::Run() { |
69 task_runner_->PostTask(posted_from_, callback_); | 69 task_runner_->PostTask(posted_from_, callback_); |
70 } | 70 } |
71 | 71 |
72 template <typename... Args> | 72 template <typename... Args> |
73 inline void TrampolineHelper<void(Args...)>::Run(Args... args) { | 73 inline void TrampolineHelper<void(Args...)>::Run(Args... args) { |
74 // TODO(tzik): Use OnceCallback directly without RunOnceClosure, once | |
75 // TaskRunner::PostTask migrates to OnceClosure. | |
76 base::OnceClosure cb = base::BindOnce(callback_, std::forward<Args>(args)...); | 74 base::OnceClosure cb = base::BindOnce(callback_, std::forward<Args>(args)...); |
77 task_runner_->PostTask( | 75 task_runner_->PostTask(posted_from_, std::move(cb)); |
78 posted_from_, | 76 } |
79 base::Bind(&TrampolineHelper::RunOnceClosure, base::Passed(&cb))); | 77 |
| 78 // First, tell the compiler OnceTrampolineHelper is a struct template with one |
| 79 // type parameter. Then define specializations where the type is a function |
| 80 // returning void and taking zero or more arguments. |
| 81 template <typename Signature> |
| 82 class OnceTrampolineHelper; |
| 83 |
| 84 template <typename... Args> |
| 85 class OnceTrampolineHelper<void(Args...)> { |
| 86 public: |
| 87 using CallbackType = base::OnceCallback<void(Args...)>; |
| 88 |
| 89 OnceTrampolineHelper(const tracked_objects::Location& posted_from, |
| 90 scoped_refptr<base::SequencedTaskRunner> task_runner, |
| 91 CallbackType callback) |
| 92 : posted_from_(posted_from), |
| 93 task_runner_(std::move(task_runner)), |
| 94 callback_(std::move(callback)) { |
| 95 DCHECK(task_runner_); |
| 96 DCHECK(callback_); |
| 97 } |
| 98 |
| 99 inline void Run(Args... args); |
| 100 |
| 101 ~OnceTrampolineHelper() { |
| 102 if (callback_) { |
| 103 task_runner_->PostTask( |
| 104 posted_from_, |
| 105 base::BindOnce(&OnceTrampolineHelper::ClearCallbackOnTargetTaskRunner, |
| 106 std::move(callback_))); |
| 107 } |
| 108 } |
| 109 |
| 110 private: |
| 111 static void ClearCallbackOnTargetTaskRunner(CallbackType) {} |
| 112 |
| 113 tracked_objects::Location posted_from_; |
| 114 scoped_refptr<base::SequencedTaskRunner> task_runner_; |
| 115 CallbackType callback_; |
| 116 }; |
| 117 |
| 118 template <> |
| 119 inline void OnceTrampolineHelper<void()>::Run() { |
| 120 task_runner_->PostTask(posted_from_, std::move(callback_)); |
| 121 } |
| 122 |
| 123 template <typename... Args> |
| 124 inline void OnceTrampolineHelper<void(Args...)>::Run(Args... args) { |
| 125 base::OnceClosure cb = |
| 126 base::BindOnce(std::move(callback_), std::forward<Args>(args)...); |
| 127 task_runner_->PostTask(posted_from_, std::move(cb)); |
80 } | 128 } |
81 | 129 |
82 } // namespace internal | 130 } // namespace internal |
83 | 131 |
84 template <typename T> | 132 template <typename T> |
| 133 inline base::OnceCallback<T> BindToCurrentLoop(base::OnceCallback<T> cb) { |
| 134 return base::BindOnce( |
| 135 &internal::OnceTrampolineHelper<T>::Run, |
| 136 base::MakeUnique<internal::OnceTrampolineHelper<T>>( |
| 137 FROM_HERE, // TODO(tzik): Propagate FROM_HERE from the caller. |
| 138 base::ThreadTaskRunnerHandle::Get(), std::move(cb))); |
| 139 } |
| 140 |
| 141 template <typename T> |
85 inline base::Callback<T> BindToCurrentLoop(base::Callback<T> cb) { | 142 inline base::Callback<T> BindToCurrentLoop(base::Callback<T> cb) { |
86 return base::Bind( | 143 return base::Bind( |
87 &internal::TrampolineHelper<T>::Run, | 144 &internal::TrampolineHelper<T>::Run, |
88 base::MakeUnique<internal::TrampolineHelper<T>>( | 145 base::MakeUnique<internal::TrampolineHelper<T>>( |
89 FROM_HERE, // TODO(tzik): Propagate FROM_HERE from the caller. | 146 FROM_HERE, // TODO(tzik): Propagate FROM_HERE from the caller. |
90 base::ThreadTaskRunnerHandle::Get(), std::move(cb))); | 147 base::ThreadTaskRunnerHandle::Get(), std::move(cb))); |
91 } | 148 } |
92 | 149 |
93 } // namespace media | 150 } // namespace media |
94 | 151 |
95 #endif // MEDIA_BASE_BIND_TO_CURRENT_LOOP_H_ | 152 #endif // MEDIA_BASE_BIND_TO_CURRENT_LOOP_H_ |
OLD | NEW |