| OLD | NEW |
| 1 /* | 1 /* |
| 2 * Copyright 2014 The WebRTC Project Authors. All rights reserved. | 2 * Copyright 2014 The WebRTC Project Authors. All rights reserved. |
| 3 * | 3 * |
| 4 * Use of this source code is governed by a BSD-style license | 4 * Use of this source code is governed by a BSD-style license |
| 5 * that can be found in the LICENSE file in the root of the source | 5 * that can be found in the LICENSE file in the root of the source |
| 6 * tree. An additional intellectual property rights grant can be found | 6 * tree. An additional intellectual property rights grant can be found |
| 7 * in the file PATENTS. All contributing project authors may | 7 * in the file PATENTS. All contributing project authors may |
| 8 * be found in the AUTHORS file in the root of the source tree. | 8 * be found in the AUTHORS file in the root of the source tree. |
| 9 */ | 9 */ |
| 10 | 10 |
| 11 #include "webrtc/base/asyncinvoker.h" | 11 #include "webrtc/base/asyncinvoker.h" |
| 12 | 12 |
| 13 #include "webrtc/base/atomicops.h" | 13 #include "webrtc/base/atomicops.h" |
| 14 #include "webrtc/base/checks.h" | 14 #include "webrtc/base/checks.h" |
| 15 #include "webrtc/base/logging.h" | 15 #include "webrtc/base/logging.h" |
| 16 | 16 |
| 17 namespace rtc { | 17 namespace rtc { |
| 18 | 18 |
| 19 AsyncInvoker::AsyncInvoker() : invocation_complete_(false, false) {} | 19 AsyncInvoker::AsyncInvoker() : invocation_complete_(false, false) {} |
| 20 | 20 |
| 21 AsyncInvoker::~AsyncInvoker() { | 21 AsyncInvoker::~AsyncInvoker() { |
| 22 destroying_ = true; | 22 destroying_ = true; |
| 23 SignalInvokerDestroyed(); | |
| 24 // Messages for this need to be cleared *before* our destructor is complete. | 23 // Messages for this need to be cleared *before* our destructor is complete. |
| 25 MessageQueueManager::Clear(this); | 24 MessageQueueManager::Clear(this); |
| 26 // And we need to wait for any invocations that are still in progress on | 25 // And we need to wait for any invocations that are still in progress on |
| 27 // other threads. | 26 // other threads. |
| 28 while (AtomicOps::AcquireLoad(&pending_invocations_)) { | 27 while (AtomicOps::AcquireLoad(&pending_invocations_)) { |
| 29 // If the destructor was called while AsyncInvoke was being called by | 28 // If the destructor was called while AsyncInvoke was being called by |
| 30 // another thread, WITHIN an AsyncInvoked functor, it may do another | 29 // another thread, WITHIN an AsyncInvoked functor, it may do another |
| 31 // Thread::Post even after we called MessageQueueManager::Clear(this). So | 30 // Thread::Post even after we called MessageQueueManager::Clear(this). So |
| 32 // we need to keep calling Clear to discard these posts. | 31 // we need to keep calling Clear to discard these posts. |
| 33 MessageQueueManager::Clear(this); | 32 MessageQueueManager::Clear(this); |
| (...skipping 85 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 119 | 118 |
| 120 NotifyingAsyncClosureBase::NotifyingAsyncClosureBase( | 119 NotifyingAsyncClosureBase::NotifyingAsyncClosureBase( |
| 121 AsyncInvoker* invoker, | 120 AsyncInvoker* invoker, |
| 122 const Location& callback_posted_from, | 121 const Location& callback_posted_from, |
| 123 Thread* calling_thread) | 122 Thread* calling_thread) |
| 124 : AsyncClosure(invoker), | 123 : AsyncClosure(invoker), |
| 125 callback_posted_from_(callback_posted_from), | 124 callback_posted_from_(callback_posted_from), |
| 126 calling_thread_(calling_thread) { | 125 calling_thread_(calling_thread) { |
| 127 calling_thread->SignalQueueDestroyed.connect( | 126 calling_thread->SignalQueueDestroyed.connect( |
| 128 this, &NotifyingAsyncClosureBase::CancelCallback); | 127 this, &NotifyingAsyncClosureBase::CancelCallback); |
| 129 invoker->SignalInvokerDestroyed.connect( | 128 // Note: We don't need to listen for the invoker being destroyed, because it |
| 130 this, &NotifyingAsyncClosureBase::CancelCallback); | 129 // will wait for this closure to be destroyed (and pending_invocations_ to go |
| 130 // to 0) before its destructor completes. |
| 131 } | 131 } |
| 132 | 132 |
| 133 NotifyingAsyncClosureBase::~NotifyingAsyncClosureBase() { | 133 NotifyingAsyncClosureBase::~NotifyingAsyncClosureBase() { |
| 134 disconnect_all(); | 134 disconnect_all(); |
| 135 } | 135 } |
| 136 | 136 |
| 137 void NotifyingAsyncClosureBase::TriggerCallback() { | 137 void NotifyingAsyncClosureBase::TriggerCallback() { |
| 138 CritScope cs(&crit_); | 138 CritScope cs(&crit_); |
| 139 if (!CallbackCanceled() && !callback_.empty()) { | 139 if (!CallbackCanceled() && !callback_.empty()) { |
| 140 invoker_->AsyncInvoke<void>(callback_posted_from_, calling_thread_, | 140 invoker_->AsyncInvoke<void>(callback_posted_from_, calling_thread_, |
| 141 callback_); | 141 callback_); |
| 142 } | 142 } |
| 143 } | 143 } |
| 144 | 144 |
| 145 void NotifyingAsyncClosureBase::CancelCallback() { | 145 void NotifyingAsyncClosureBase::CancelCallback() { |
| 146 // If the callback is triggering when this is called, block the | 146 // If the callback is triggering when this is called, block the |
| 147 // destructor of the dying object here by waiting until the callback | 147 // destructor of the dying object here by waiting until the callback |
| 148 // is done triggering. | 148 // is done triggering. |
| 149 CritScope cs(&crit_); | 149 CritScope cs(&crit_); |
| 150 // calling_thread_ == nullptr means do not trigger the callback. | 150 // calling_thread_ == nullptr means do not trigger the callback. |
| 151 calling_thread_ = nullptr; | 151 calling_thread_ = nullptr; |
| 152 } | 152 } |
| 153 | 153 |
| 154 } // namespace rtc | 154 } // namespace rtc |
| OLD | NEW |