mirror of
https://github.com/ton-blockchain/ton
synced 2025-02-12 11:12:16 +00:00
550c28d7db
* Improve DHT store/load, pinging overlay peers * Fix speed limits in storage * Use keyStoreTypeDirectory in rldp-http-proxy and storage-daemon Mainly for caching synced block in tonlib. --------- Co-authored-by: SpyCheese <mikle98@yandex.ru>
83 lines
2.4 KiB
C++
83 lines
2.4 KiB
C++
/*
|
|
This file is part of TON Blockchain Library.
|
|
|
|
TON Blockchain Library is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU Lesser General Public License as published by
|
|
the Free Software Foundation, either version 2 of the License, or
|
|
(at your option) any later version.
|
|
|
|
TON Blockchain Library is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU Lesser General Public License for more details.
|
|
|
|
You should have received a copy of the GNU Lesser General Public License
|
|
along with TON Blockchain Library. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include "SpeedLimiter.h"
|
|
#include "common/errorcode.h"
|
|
|
|
namespace ton {
|
|
|
|
SpeedLimiter::SpeedLimiter(double max_speed) : max_speed_(max_speed) {
|
|
}
|
|
|
|
void SpeedLimiter::set_max_speed(double max_speed) {
|
|
max_speed_ = max_speed;
|
|
auto old_queue = std::move(queue_);
|
|
unlock_at_ = (queue_.empty() ? td::Timestamp::now() : queue_.front().execute_at_);
|
|
queue_ = {};
|
|
while (!old_queue.empty()) {
|
|
auto &e = old_queue.front();
|
|
enqueue(e.size_, e.timeout_, std::move(e.promise_));
|
|
old_queue.pop();
|
|
}
|
|
process_queue();
|
|
}
|
|
|
|
void SpeedLimiter::enqueue(double size, td::Timestamp timeout, td::Promise<td::Unit> promise) {
|
|
if (max_speed_ < 0.0) {
|
|
promise.set_result(td::Unit());
|
|
return;
|
|
}
|
|
if (max_speed_ == 0.0) {
|
|
promise.set_error(td::Status::Error(ErrorCode::timeout, "Speed limit is 0"));
|
|
return;
|
|
}
|
|
if (timeout < unlock_at_) {
|
|
promise.set_error(td::Status::Error(ErrorCode::timeout, "Timeout caused by speed limit"));
|
|
return;
|
|
}
|
|
if (queue_.empty() && unlock_at_.is_in_past()) {
|
|
unlock_at_ = td::Timestamp::now();
|
|
promise.set_result(td::Unit());
|
|
} else {
|
|
queue_.push({unlock_at_, size, timeout, std::move(promise)});
|
|
}
|
|
unlock_at_ = td::Timestamp::in(size / max_speed_, unlock_at_);
|
|
if (!queue_.empty()) {
|
|
alarm_timestamp() = queue_.front().execute_at_;
|
|
}
|
|
}
|
|
|
|
void SpeedLimiter::alarm() {
|
|
process_queue();
|
|
}
|
|
|
|
void SpeedLimiter::process_queue() {
|
|
while (!queue_.empty()) {
|
|
auto &e = queue_.front();
|
|
if (e.execute_at_.is_in_past()) {
|
|
e.promise_.set_result(td::Unit());
|
|
queue_.pop();
|
|
} else {
|
|
break;
|
|
}
|
|
}
|
|
if (!queue_.empty()) {
|
|
alarm_timestamp() = queue_.front().execute_at_;
|
|
}
|
|
}
|
|
|
|
} // namespace ton
|