92 lines
2.7 KiB
C++
92 lines
2.7 KiB
C++
|
/*
|
||
|
|
* Copyright 2019 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 "rtc_base/task_utils/repeating_task.h"
|
||
|
|
|
||
|
|
#include "absl/memory/memory.h"
|
||
|
|
#include "rtc_base/logging.h"
|
||
|
|
#include "rtc_base/task_utils/to_queued_task.h"
|
||
|
|
#include "rtc_base/time_utils.h"
|
||
|
|
|
||
|
|
namespace webrtc {
|
||
|
|
namespace webrtc_repeating_task_impl {
|
||
|
|
|
||
|
|
RepeatingTaskBase::RepeatingTaskBase(TaskQueueBase* task_queue,
|
||
|
|
TimeDelta first_delay,
|
||
|
|
Clock* clock)
|
||
|
|
: task_queue_(task_queue),
|
||
|
|
clock_(clock),
|
||
|
|
next_run_time_(clock_->CurrentTime() + first_delay) {}
|
||
|
|
|
||
|
|
RepeatingTaskBase::~RepeatingTaskBase() = default;
|
||
|
|
|
||
|
|
bool RepeatingTaskBase::Run() {
|
||
|
|
RTC_DCHECK_RUN_ON(task_queue_);
|
||
|
|
// Return true to tell the TaskQueue to destruct this object.
|
||
|
|
if (next_run_time_.IsPlusInfinity())
|
||
|
|
return true;
|
||
|
|
|
||
|
|
TimeDelta delay = RunClosure();
|
||
|
|
|
||
|
|
// The closure might have stopped this task, in which case we return true to
|
||
|
|
// destruct this object.
|
||
|
|
if (next_run_time_.IsPlusInfinity())
|
||
|
|
return true;
|
||
|
|
|
||
|
|
RTC_DCHECK(delay.IsFinite());
|
||
|
|
TimeDelta lost_time = clock_->CurrentTime() - next_run_time_;
|
||
|
|
next_run_time_ += delay;
|
||
|
|
delay -= lost_time;
|
||
|
|
delay = std::max(delay, TimeDelta::Zero());
|
||
|
|
|
||
|
|
task_queue_->PostDelayedTask(absl::WrapUnique(this), delay.ms());
|
||
|
|
|
||
|
|
// Return false to tell the TaskQueue to not destruct this object since we
|
||
|
|
// have taken ownership with absl::WrapUnique.
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
void RepeatingTaskBase::Stop() {
|
||
|
|
RTC_DCHECK_RUN_ON(task_queue_);
|
||
|
|
RTC_DCHECK(next_run_time_.IsFinite());
|
||
|
|
next_run_time_ = Timestamp::PlusInfinity();
|
||
|
|
}
|
||
|
|
|
||
|
|
} // namespace webrtc_repeating_task_impl
|
||
|
|
|
||
|
|
RepeatingTaskHandle::RepeatingTaskHandle(RepeatingTaskHandle&& other)
|
||
|
|
: repeating_task_(other.repeating_task_) {
|
||
|
|
other.repeating_task_ = nullptr;
|
||
|
|
}
|
||
|
|
|
||
|
|
RepeatingTaskHandle& RepeatingTaskHandle::operator=(
|
||
|
|
RepeatingTaskHandle&& other) {
|
||
|
|
repeating_task_ = other.repeating_task_;
|
||
|
|
other.repeating_task_ = nullptr;
|
||
|
|
return *this;
|
||
|
|
}
|
||
|
|
|
||
|
|
RepeatingTaskHandle::RepeatingTaskHandle(
|
||
|
|
webrtc_repeating_task_impl::RepeatingTaskBase* repeating_task)
|
||
|
|
: repeating_task_(repeating_task) {}
|
||
|
|
|
||
|
|
void RepeatingTaskHandle::Stop() {
|
||
|
|
if (repeating_task_) {
|
||
|
|
repeating_task_->Stop();
|
||
|
|
repeating_task_ = nullptr;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
bool RepeatingTaskHandle::Running() const {
|
||
|
|
return repeating_task_ != nullptr;
|
||
|
|
}
|
||
|
|
|
||
|
|
} // namespace webrtc
|