mirror of
https://github.com/mollyim/webrtc.git
synced 2025-05-14 06:10:40 +01:00
52fd96fb73
Bug: webrtc:11567 Change-Id: I26fb07bf84ed197ce667290aa0bf4816bc9c5c06 Reviewed-on: https://webrtc-review.googlesource.com/c/src/+/178818 Reviewed-by: Sebastian Jansson <srte@webrtc.org> Commit-Queue: Markus Handell <handellm@webrtc.org> Cr-Commit-Position: refs/heads/master@{#31660}
82 lines
2.4 KiB
C++
82 lines
2.4 KiB
C++
/*
|
|
* Copyright (c) 2020 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 "test/time_controller/simulated_task_queue.h"
|
|
|
|
#include <algorithm>
|
|
#include <utility>
|
|
|
|
namespace webrtc {
|
|
|
|
SimulatedTaskQueue::SimulatedTaskQueue(
|
|
sim_time_impl::SimulatedTimeControllerImpl* handler,
|
|
absl::string_view name)
|
|
: handler_(handler), name_(new char[name.size()]) {
|
|
std::copy_n(name.begin(), name.size(), name_);
|
|
}
|
|
|
|
SimulatedTaskQueue::~SimulatedTaskQueue() {
|
|
handler_->Unregister(this);
|
|
delete[] name_;
|
|
}
|
|
|
|
void SimulatedTaskQueue::Delete() {
|
|
{
|
|
MutexLock lock(&lock_);
|
|
ready_tasks_.clear();
|
|
delayed_tasks_.clear();
|
|
}
|
|
delete this;
|
|
}
|
|
|
|
void SimulatedTaskQueue::RunReady(Timestamp at_time) {
|
|
MutexLock lock(&lock_);
|
|
for (auto it = delayed_tasks_.begin();
|
|
it != delayed_tasks_.end() && it->first <= at_time;
|
|
it = delayed_tasks_.erase(it)) {
|
|
for (auto& task : it->second) {
|
|
ready_tasks_.emplace_back(std::move(task));
|
|
}
|
|
}
|
|
CurrentTaskQueueSetter set_current(this);
|
|
while (!ready_tasks_.empty()) {
|
|
std::unique_ptr<QueuedTask> ready = std::move(ready_tasks_.front());
|
|
ready_tasks_.pop_front();
|
|
lock_.Unlock();
|
|
bool delete_task = ready->Run();
|
|
if (delete_task) {
|
|
ready.reset();
|
|
} else {
|
|
ready.release();
|
|
}
|
|
lock_.Lock();
|
|
}
|
|
if (!delayed_tasks_.empty()) {
|
|
next_run_time_ = delayed_tasks_.begin()->first;
|
|
} else {
|
|
next_run_time_ = Timestamp::PlusInfinity();
|
|
}
|
|
}
|
|
|
|
void SimulatedTaskQueue::PostTask(std::unique_ptr<QueuedTask> task) {
|
|
MutexLock lock(&lock_);
|
|
ready_tasks_.emplace_back(std::move(task));
|
|
next_run_time_ = Timestamp::MinusInfinity();
|
|
}
|
|
|
|
void SimulatedTaskQueue::PostDelayedTask(std::unique_ptr<QueuedTask> task,
|
|
uint32_t milliseconds) {
|
|
MutexLock lock(&lock_);
|
|
Timestamp target_time =
|
|
handler_->CurrentTime() + TimeDelta::Millis(milliseconds);
|
|
delayed_tasks_[target_time].push_back(std::move(task));
|
|
next_run_time_ = std::min(next_run_time_, target_time);
|
|
}
|
|
|
|
} // namespace webrtc
|