Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(530)

Unified Diff: webrtc/base/task_queue_win.cc

Issue 2877023002: Move webrtc/{base => rtc_base} (Closed)
Patch Set: update presubmit.py and DEPS include rules Created 3 years, 6 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
« no previous file with comments | « webrtc/base/task_queue_unittest.cc ('k') | webrtc/base/template_util.h » ('j') | no next file with comments »
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
Index: webrtc/base/task_queue_win.cc
diff --git a/webrtc/base/task_queue_win.cc b/webrtc/base/task_queue_win.cc
deleted file mode 100644
index a149dd88a6b05e0ee402ce5c78713c4042db2c6c..0000000000000000000000000000000000000000
--- a/webrtc/base/task_queue_win.cc
+++ /dev/null
@@ -1,415 +0,0 @@
-/*
- * Copyright 2016 The WebRTC Project Authors. All rights reserved.
- *
- * Use of this source code is governed by a BSD-style license
- * that can be found in the LICENSE file in the root of the source
- * tree. An additional intellectual property rights grant can be found
- * in the file PATENTS. All contributing project authors may
- * be found in the AUTHORS file in the root of the source tree.
- */
-
-#include "webrtc/base/task_queue.h"
-
-#include <mmsystem.h>
-#include <string.h>
-
-#include <algorithm>
-#include <queue>
-
-#include "webrtc/base/arraysize.h"
-#include "webrtc/base/checks.h"
-#include "webrtc/base/logging.h"
-#include "webrtc/base/safe_conversions.h"
-#include "webrtc/base/timeutils.h"
-
-namespace rtc {
-namespace {
-#define WM_RUN_TASK WM_USER + 1
-#define WM_QUEUE_DELAYED_TASK WM_USER + 2
-
-using Priority = TaskQueue::Priority;
-
-DWORD g_queue_ptr_tls = 0;
-
-BOOL CALLBACK InitializeTls(PINIT_ONCE init_once, void* param, void** context) {
- g_queue_ptr_tls = TlsAlloc();
- return TRUE;
-}
-
-DWORD GetQueuePtrTls() {
- static INIT_ONCE init_once = INIT_ONCE_STATIC_INIT;
- ::InitOnceExecuteOnce(&init_once, InitializeTls, nullptr, nullptr);
- return g_queue_ptr_tls;
-}
-
-struct ThreadStartupData {
- Event* started;
- void* thread_context;
-};
-
-void CALLBACK InitializeQueueThread(ULONG_PTR param) {
- MSG msg;
- ::PeekMessage(&msg, nullptr, WM_USER, WM_USER, PM_NOREMOVE);
- ThreadStartupData* data = reinterpret_cast<ThreadStartupData*>(param);
- ::TlsSetValue(GetQueuePtrTls(), data->thread_context);
- data->started->Set();
-}
-
-ThreadPriority TaskQueuePriorityToThreadPriority(Priority priority) {
- switch (priority) {
- case Priority::HIGH:
- return kRealtimePriority;
- case Priority::LOW:
- return kLowPriority;
- case Priority::NORMAL:
- return kNormalPriority;
- default:
- RTC_NOTREACHED();
- break;
- }
- return kNormalPriority;
-}
-
-int64_t GetTick() {
- static const UINT kPeriod = 1;
- bool high_res = (timeBeginPeriod(kPeriod) == TIMERR_NOERROR);
- int64_t ret = TimeMillis();
- if (high_res)
- timeEndPeriod(kPeriod);
- return ret;
-}
-
-class DelayedTaskInfo {
- public:
- // Default ctor needed to support priority_queue::pop().
- DelayedTaskInfo() {}
- DelayedTaskInfo(uint32_t milliseconds, std::unique_ptr<QueuedTask> task)
- : due_time_(GetTick() + milliseconds), task_(std::move(task)) {}
- DelayedTaskInfo(DelayedTaskInfo&&) = default;
-
- // Implement for priority_queue.
- bool operator>(const DelayedTaskInfo& other) const {
- return due_time_ > other.due_time_;
- }
-
- // Required by priority_queue::pop().
- DelayedTaskInfo& operator=(DelayedTaskInfo&& other) = default;
-
- // See below for why this method is const.
- void Run() const {
- RTC_DCHECK(due_time_);
- task_->Run() ? task_.reset() : static_cast<void>(task_.release());
- }
-
- int64_t due_time() const { return due_time_; }
-
- private:
- int64_t due_time_ = 0; // Absolute timestamp in milliseconds.
-
- // |task| needs to be mutable because std::priority_queue::top() returns
- // a const reference and a key in an ordered queue must not be changed.
- // There are two basic workarounds, one using const_cast, which would also
- // make the key (|due_time|), non-const and the other is to make the non-key
- // (|task|), mutable.
- // Because of this, the |task| variable is made private and can only be
- // mutated by calling the |Run()| method.
- mutable std::unique_ptr<QueuedTask> task_;
-};
-
-class MultimediaTimer {
- public:
- // Note: We create an event that requires manual reset.
- MultimediaTimer() : event_(::CreateEvent(nullptr, true, false, nullptr)) {}
-
- ~MultimediaTimer() {
- Cancel();
- ::CloseHandle(event_);
- }
-
- bool StartOneShotTimer(UINT delay_ms) {
- RTC_DCHECK_EQ(0, timer_id_);
- RTC_DCHECK(event_ != nullptr);
- timer_id_ =
- ::timeSetEvent(delay_ms, 0, reinterpret_cast<LPTIMECALLBACK>(event_), 0,
- TIME_ONESHOT | TIME_CALLBACK_EVENT_SET);
- return timer_id_ != 0;
- }
-
- void Cancel() {
- ::ResetEvent(event_);
- if (timer_id_) {
- ::timeKillEvent(timer_id_);
- timer_id_ = 0;
- }
- }
-
- HANDLE* event_for_wait() { return &event_; }
-
- private:
- HANDLE event_ = nullptr;
- MMRESULT timer_id_ = 0;
-
- RTC_DISALLOW_COPY_AND_ASSIGN(MultimediaTimer);
-};
-
-} // namespace
-
-class TaskQueue::ThreadState {
- public:
- explicit ThreadState(HANDLE in_queue) : in_queue_(in_queue) {}
- ~ThreadState() {}
-
- void RunThreadMain();
-
- private:
- bool ProcessQueuedMessages();
- void RunDueTasks();
- void ScheduleNextTimer();
- void CancelTimers();
-
- // Since priority_queue<> by defult orders items in terms of
- // largest->smallest, using std::less<>, and we want smallest->largest,
- // we would like to use std::greater<> here. Alas it's only available in
- // C++14 and later, so we roll our own compare template that that relies on
- // operator<().
- template <typename T>
- struct greater {
- bool operator()(const T& l, const T& r) { return l > r; }
- };
-
- MultimediaTimer timer_;
- std::priority_queue<DelayedTaskInfo,
- std::vector<DelayedTaskInfo>,
- greater<DelayedTaskInfo>>
- timer_tasks_;
- UINT_PTR timer_id_ = 0;
- HANDLE in_queue_;
-};
-
-TaskQueue::TaskQueue(const char* queue_name, Priority priority /*= NORMAL*/)
- : thread_(&TaskQueue::ThreadMain,
- this,
- queue_name,
- TaskQueuePriorityToThreadPriority(priority)),
- in_queue_(::CreateEvent(nullptr, true, false, nullptr)) {
- RTC_DCHECK(queue_name);
- RTC_DCHECK(in_queue_);
- thread_.Start();
- Event event(false, false);
- ThreadStartupData startup = {&event, this};
- RTC_CHECK(thread_.QueueAPC(&InitializeQueueThread,
- reinterpret_cast<ULONG_PTR>(&startup)));
- event.Wait(Event::kForever);
-}
-
-TaskQueue::~TaskQueue() {
- RTC_DCHECK(!IsCurrent());
- while (!::PostThreadMessage(thread_.GetThreadRef(), WM_QUIT, 0, 0)) {
- RTC_CHECK_EQ(ERROR_NOT_ENOUGH_QUOTA, ::GetLastError());
- Sleep(1);
- }
- thread_.Stop();
- ::CloseHandle(in_queue_);
-}
-
-// static
-TaskQueue* TaskQueue::Current() {
- return static_cast<TaskQueue*>(::TlsGetValue(GetQueuePtrTls()));
-}
-
-// static
-bool TaskQueue::IsCurrent(const char* queue_name) {
- TaskQueue* current = Current();
- return current && current->thread_.name().compare(queue_name) == 0;
-}
-
-bool TaskQueue::IsCurrent() const {
- return IsThreadRefEqual(thread_.GetThreadRef(), CurrentThreadRef());
-}
-
-void TaskQueue::PostTask(std::unique_ptr<QueuedTask> task) {
- rtc::CritScope lock(&pending_lock_);
- pending_.push(std::move(task));
- ::SetEvent(in_queue_);
-}
-
-void TaskQueue::PostDelayedTask(std::unique_ptr<QueuedTask> task,
- uint32_t milliseconds) {
- if (!milliseconds) {
- PostTask(std::move(task));
- return;
- }
-
- // TODO(tommi): Avoid this allocation. It is currently here since
- // the timestamp stored in the task info object, is a 64bit timestamp
- // and WPARAM is 32bits in 32bit builds. Otherwise, we could pass the
- // task pointer and timestamp as LPARAM and WPARAM.
- auto* task_info = new DelayedTaskInfo(milliseconds, std::move(task));
- if (!::PostThreadMessage(thread_.GetThreadRef(), WM_QUEUE_DELAYED_TASK, 0,
- reinterpret_cast<LPARAM>(task_info))) {
- delete task_info;
- }
-}
-
-void TaskQueue::PostTaskAndReply(std::unique_ptr<QueuedTask> task,
- std::unique_ptr<QueuedTask> reply,
- TaskQueue* reply_queue) {
- QueuedTask* task_ptr = task.release();
- QueuedTask* reply_task_ptr = reply.release();
- DWORD reply_thread_id = reply_queue->thread_.GetThreadRef();
- PostTask([task_ptr, reply_task_ptr, reply_thread_id]() {
- if (task_ptr->Run())
- delete task_ptr;
- // If the thread's message queue is full, we can't queue the task and will
- // have to drop it (i.e. delete).
- if (!::PostThreadMessage(reply_thread_id, WM_RUN_TASK, 0,
- reinterpret_cast<LPARAM>(reply_task_ptr))) {
- delete reply_task_ptr;
- }
- });
-}
-
-void TaskQueue::PostTaskAndReply(std::unique_ptr<QueuedTask> task,
- std::unique_ptr<QueuedTask> reply) {
- return PostTaskAndReply(std::move(task), std::move(reply), Current());
-}
-
-void TaskQueue::RunPendingTasks() {
- while (true) {
- std::unique_ptr<QueuedTask> task;
- {
- rtc::CritScope lock(&pending_lock_);
- if (pending_.empty())
- break;
- task = std::move(pending_.front());
- pending_.pop();
- }
-
- if (!task->Run())
- task.release();
- }
-}
-
-// static
-void TaskQueue::ThreadMain(void* context) {
- ThreadState state(static_cast<TaskQueue*>(context)->in_queue_);
- state.RunThreadMain();
-}
-
-void TaskQueue::ThreadState::RunThreadMain() {
- HANDLE handles[2] = { *timer_.event_for_wait(), in_queue_ };
- while (true) {
- // Make sure we do an alertable wait as that's required to allow APCs to run
- // (e.g. required for InitializeQueueThread and stopping the thread in
- // PlatformThread).
- DWORD result = ::MsgWaitForMultipleObjectsEx(
- arraysize(handles), handles, INFINITE, QS_ALLEVENTS, MWMO_ALERTABLE);
- RTC_CHECK_NE(WAIT_FAILED, result);
- if (result == (WAIT_OBJECT_0 + 2)) {
- // There are messages in the message queue that need to be handled.
- if (!ProcessQueuedMessages())
- break;
- }
-
- if (result == WAIT_OBJECT_0 || (!timer_tasks_.empty() &&
- ::WaitForSingleObject(*timer_.event_for_wait(), 0) == WAIT_OBJECT_0)) {
- // The multimedia timer was signaled.
- timer_.Cancel();
- RunDueTasks();
- ScheduleNextTimer();
- }
-
- if (result == (WAIT_OBJECT_0 + 1)) {
- ::ResetEvent(in_queue_);
- TaskQueue::Current()->RunPendingTasks();
- }
- }
-}
-
-bool TaskQueue::ThreadState::ProcessQueuedMessages() {
- MSG msg = {};
- // To protect against overly busy message queues, we limit the time
- // we process tasks to a few milliseconds. If we don't do that, there's
- // a chance that timer tasks won't ever run.
- static const int kMaxTaskProcessingTimeMs = 500;
- auto start = GetTick();
- while (::PeekMessage(&msg, nullptr, 0, 0, PM_REMOVE) &&
- msg.message != WM_QUIT) {
- if (!msg.hwnd) {
- switch (msg.message) {
- // TODO(tommi): Stop using this way of queueing tasks.
- case WM_RUN_TASK: {
- QueuedTask* task = reinterpret_cast<QueuedTask*>(msg.lParam);
- if (task->Run())
- delete task;
- break;
- }
- case WM_QUEUE_DELAYED_TASK: {
- std::unique_ptr<DelayedTaskInfo> info(
- reinterpret_cast<DelayedTaskInfo*>(msg.lParam));
- bool need_to_schedule_timers =
- timer_tasks_.empty() ||
- timer_tasks_.top().due_time() > info->due_time();
- timer_tasks_.emplace(std::move(*info.get()));
- if (need_to_schedule_timers) {
- CancelTimers();
- ScheduleNextTimer();
- }
- break;
- }
- case WM_TIMER: {
- RTC_DCHECK_EQ(timer_id_, msg.wParam);
- ::KillTimer(nullptr, msg.wParam);
- timer_id_ = 0;
- RunDueTasks();
- ScheduleNextTimer();
- break;
- }
- default:
- RTC_NOTREACHED();
- break;
- }
- } else {
- ::TranslateMessage(&msg);
- ::DispatchMessage(&msg);
- }
-
- if (GetTick() > start + kMaxTaskProcessingTimeMs)
- break;
- }
- return msg.message != WM_QUIT;
-}
-
-void TaskQueue::ThreadState::RunDueTasks() {
- RTC_DCHECK(!timer_tasks_.empty());
- auto now = GetTick();
- do {
- const auto& top = timer_tasks_.top();
- if (top.due_time() > now)
- break;
- top.Run();
- timer_tasks_.pop();
- } while (!timer_tasks_.empty());
-}
-
-void TaskQueue::ThreadState::ScheduleNextTimer() {
- RTC_DCHECK_EQ(timer_id_, 0);
- if (timer_tasks_.empty())
- return;
-
- const auto& next_task = timer_tasks_.top();
- int64_t delay_ms = std::max(0ll, next_task.due_time() - GetTick());
- uint32_t milliseconds = rtc::dchecked_cast<uint32_t>(delay_ms);
- if (!timer_.StartOneShotTimer(milliseconds))
- timer_id_ = ::SetTimer(nullptr, 0, milliseconds, nullptr);
-}
-
-void TaskQueue::ThreadState::CancelTimers() {
- timer_.Cancel();
- if (timer_id_) {
- ::KillTimer(nullptr, timer_id_);
- timer_id_ = 0;
- }
-}
-
-} // namespace rtc
« no previous file with comments | « webrtc/base/task_queue_unittest.cc ('k') | webrtc/base/template_util.h » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698