mirror of
https://github.com/mollyim/webrtc.git
synced 2025-05-13 13:50:40 +01:00
3b205da7e4
Bug: b/239155933 Change-Id: I1b90a969b9f781fe2902aa822020590683a04b7d Reviewed-on: https://webrtc-review.googlesource.com/c/src/+/270923 Reviewed-by: Artem Titov <titovartem@webrtc.org> Reviewed-by: Victor Boivie <boivie@webrtc.org> Commit-Queue: Mirko Bonadei <mbonadei@webrtc.org> Cr-Commit-Position: refs/heads/main@{#37719}
89 lines
2.8 KiB
C++
89 lines
2.8 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() {
|
|
// Need to destroy the tasks outside of the lock because task destruction
|
|
// can lead to re-entry in SimulatedTaskQueue via custom destructors.
|
|
std::deque<absl::AnyInvocable<void() &&>> ready_tasks;
|
|
std::map<Timestamp, std::vector<absl::AnyInvocable<void() &&>>> delayed_tasks;
|
|
{
|
|
MutexLock lock(&lock_);
|
|
ready_tasks_.swap(ready_tasks);
|
|
delayed_tasks_.swap(delayed_tasks);
|
|
}
|
|
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_.push_back(std::move(task));
|
|
}
|
|
}
|
|
CurrentTaskQueueSetter set_current(this);
|
|
while (!ready_tasks_.empty()) {
|
|
absl::AnyInvocable<void()&&> ready = std::move(ready_tasks_.front());
|
|
ready_tasks_.pop_front();
|
|
lock_.Unlock();
|
|
std::move(ready)();
|
|
ready = nullptr;
|
|
lock_.Lock();
|
|
}
|
|
if (!delayed_tasks_.empty()) {
|
|
next_run_time_ = delayed_tasks_.begin()->first;
|
|
} else {
|
|
next_run_time_ = Timestamp::PlusInfinity();
|
|
}
|
|
}
|
|
|
|
void SimulatedTaskQueue::PostTask(absl::AnyInvocable<void() &&> task) {
|
|
MutexLock lock(&lock_);
|
|
ready_tasks_.push_back(std::move(task));
|
|
next_run_time_ = Timestamp::MinusInfinity();
|
|
}
|
|
|
|
void SimulatedTaskQueue::PostDelayedTask(absl::AnyInvocable<void() &&> task,
|
|
TimeDelta delay) {
|
|
PostDelayedHighPrecisionTask(std::move(task), delay);
|
|
}
|
|
|
|
void SimulatedTaskQueue::PostDelayedHighPrecisionTask(
|
|
absl::AnyInvocable<void() &&> task,
|
|
TimeDelta delay) {
|
|
MutexLock lock(&lock_);
|
|
Timestamp target_time = handler_->CurrentTime() + delay;
|
|
delayed_tasks_[target_time].push_back(std::move(task));
|
|
next_run_time_ = std::min(next_run_time_, target_time);
|
|
}
|
|
|
|
} // namespace webrtc
|