| OLD | NEW |
|---|
| (Empty) |
| 1 /* | |
| 2 * Copyright 2014 The WebRTC Project Authors. All rights reserved. | |
| 3 * | |
| 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 | |
| 6 * tree. An additional intellectual property rights grant can be found | |
| 7 * in the file PATENTS. All contributing project authors may | |
| 8 * be found in the AUTHORS file in the root of the source tree. | |
| 9 */ | |
| 10 | |
| 11 #include "webrtc/base/asyncinvoker.h" | |
| 12 | |
| 13 #include "webrtc/base/atomicops.h" | |
| 14 #include "webrtc/base/checks.h" | |
| 15 #include "webrtc/base/logging.h" | |
| 16 | |
| 17 namespace rtc { | |
| 18 | |
| 19 AsyncInvoker::AsyncInvoker() : invocation_complete_(false, false) {} | |
| 20 | |
| 21 AsyncInvoker::~AsyncInvoker() { | |
| 22 destroying_ = true; | |
| 23 // Messages for this need to be cleared *before* our destructor is complete. | |
| 24 MessageQueueManager::Clear(this); | |
| 25 // And we need to wait for any invocations that are still in progress on | |
| 26 // other threads. | |
| 27 while (AtomicOps::AcquireLoad(&pending_invocations_)) { | |
| 28 // If the destructor was called while AsyncInvoke was being called by | |
| 29 // another thread, WITHIN an AsyncInvoked functor, it may do another | |
| 30 // Thread::Post even after we called MessageQueueManager::Clear(this). So | |
| 31 // we need to keep calling Clear to discard these posts. | |
| 32 MessageQueueManager::Clear(this); | |
| 33 invocation_complete_.Wait(Event::kForever); | |
| 34 } | |
| 35 } | |
| 36 | |
| 37 void AsyncInvoker::OnMessage(Message* msg) { | |
| 38 // Get the AsyncClosure shared ptr from this message's data. | |
| 39 ScopedMessageData<AsyncClosure>* data = | |
| 40 static_cast<ScopedMessageData<AsyncClosure>*>(msg->pdata); | |
| 41 // Execute the closure and trigger the return message if needed. | |
| 42 data->inner_data().Execute(); | |
| 43 delete data; | |
| 44 } | |
| 45 | |
| 46 void AsyncInvoker::Flush(Thread* thread, uint32_t id /*= MQID_ANY*/) { | |
| 47 if (destroying_) return; | |
| 48 | |
| 49 // Run this on |thread| to reduce the number of context switches. | |
| 50 if (Thread::Current() != thread) { | |
| 51 thread->Invoke<void>(RTC_FROM_HERE, | |
| 52 Bind(&AsyncInvoker::Flush, this, thread, id)); | |
| 53 return; | |
| 54 } | |
| 55 | |
| 56 MessageList removed; | |
| 57 thread->Clear(this, id, &removed); | |
| 58 for (MessageList::iterator it = removed.begin(); it != removed.end(); ++it) { | |
| 59 // This message was pending on this thread, so run it now. | |
| 60 thread->Send(it->posted_from, it->phandler, it->message_id, it->pdata); | |
| 61 } | |
| 62 } | |
| 63 | |
| 64 void AsyncInvoker::DoInvoke(const Location& posted_from, | |
| 65 Thread* thread, | |
| 66 std::unique_ptr<AsyncClosure> closure, | |
| 67 uint32_t id) { | |
| 68 if (destroying_) { | |
| 69 LOG(LS_WARNING) << "Tried to invoke while destroying the invoker."; | |
| 70 return; | |
| 71 } | |
| 72 AtomicOps::Increment(&pending_invocations_); | |
| 73 thread->Post(posted_from, this, id, | |
| 74 new ScopedMessageData<AsyncClosure>(std::move(closure))); | |
| 75 } | |
| 76 | |
| 77 void AsyncInvoker::DoInvokeDelayed(const Location& posted_from, | |
| 78 Thread* thread, | |
| 79 std::unique_ptr<AsyncClosure> closure, | |
| 80 uint32_t delay_ms, | |
| 81 uint32_t id) { | |
| 82 if (destroying_) { | |
| 83 LOG(LS_WARNING) << "Tried to invoke while destroying the invoker."; | |
| 84 return; | |
| 85 } | |
| 86 AtomicOps::Increment(&pending_invocations_); | |
| 87 thread->PostDelayed(posted_from, delay_ms, this, id, | |
| 88 new ScopedMessageData<AsyncClosure>(std::move(closure))); | |
| 89 } | |
| 90 | |
| 91 GuardedAsyncInvoker::GuardedAsyncInvoker() : thread_(Thread::Current()) { | |
| 92 thread_->SignalQueueDestroyed.connect(this, | |
| 93 &GuardedAsyncInvoker::ThreadDestroyed); | |
| 94 } | |
| 95 | |
| 96 GuardedAsyncInvoker::~GuardedAsyncInvoker() { | |
| 97 } | |
| 98 | |
| 99 bool GuardedAsyncInvoker::Flush(uint32_t id) { | |
| 100 rtc::CritScope cs(&crit_); | |
| 101 if (thread_ == nullptr) | |
| 102 return false; | |
| 103 invoker_.Flush(thread_, id); | |
| 104 return true; | |
| 105 } | |
| 106 | |
| 107 void GuardedAsyncInvoker::ThreadDestroyed() { | |
| 108 rtc::CritScope cs(&crit_); | |
| 109 // We should never get more than one notification about the thread dying. | |
| 110 RTC_DCHECK(thread_ != nullptr); | |
| 111 thread_ = nullptr; | |
| 112 } | |
| 113 | |
| 114 AsyncClosure::~AsyncClosure() { | |
| 115 AtomicOps::Decrement(&invoker_->pending_invocations_); | |
| 116 invoker_->invocation_complete_.Set(); | |
| 117 } | |
| 118 | |
| 119 } // namespace rtc | |
| OLD | NEW |
|---|
Chromium Code Reviews
Issue 2877023002:
Move webrtc/{base => rtc_base} (Closed)
