100 lines
2.5 KiB
C++
100 lines
2.5 KiB
C++
//
|
|
// Copyright Aliaksei Levin (levlam@telegram.org), Arseny Smirnov (arseny30@gmail.com) 2014-2023
|
|
//
|
|
// Distributed under the Boost Software License, Version 1.0. (See accompanying
|
|
// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
|
|
//
|
|
#pragma once
|
|
|
|
#include "td/utils/common.h"
|
|
#include "td/utils/logging.h"
|
|
|
|
#include <utility>
|
|
|
|
namespace td {
|
|
|
|
// Process states in order defined by their SeqNo
|
|
template <class DataT>
|
|
class OrderedEventsProcessor {
|
|
public:
|
|
using SeqNo = uint64;
|
|
|
|
OrderedEventsProcessor() = default;
|
|
explicit OrderedEventsProcessor(SeqNo offset) : offset_(offset), begin_(offset_), end_(offset_) {
|
|
}
|
|
|
|
template <class FunctionT>
|
|
void clear(FunctionT &&function) {
|
|
for (auto &it : data_array_) {
|
|
if (it.second) {
|
|
function(std::move(it.first));
|
|
}
|
|
}
|
|
*this = OrderedEventsProcessor();
|
|
}
|
|
void clear() {
|
|
*this = OrderedEventsProcessor();
|
|
}
|
|
template <class FromDataT, class FunctionT>
|
|
void add(SeqNo seq_no, FromDataT &&data, FunctionT &&function) {
|
|
LOG_CHECK(seq_no >= begin_) << seq_no << ">=" << begin_; // or ignore?
|
|
|
|
if (seq_no == begin_) { // run now
|
|
begin_++;
|
|
function(seq_no, std::forward<FromDataT>(data));
|
|
|
|
while (begin_ < end_) {
|
|
auto &data_flag = data_array_[static_cast<size_t>(begin_ - offset_)];
|
|
if (!data_flag.second) {
|
|
break;
|
|
}
|
|
function(begin_, std::move(data_flag.first));
|
|
data_flag.second = false;
|
|
begin_++;
|
|
}
|
|
if (begin_ > end_) {
|
|
end_ = begin_;
|
|
}
|
|
if (begin_ == end_) {
|
|
offset_ = begin_;
|
|
}
|
|
|
|
// try_compactify
|
|
auto begin_pos = static_cast<size_t>(begin_ - offset_);
|
|
if (begin_pos > 5 && begin_pos * 2 > data_array_.size()) {
|
|
data_array_.erase(data_array_.begin(), data_array_.begin() + begin_pos);
|
|
offset_ = begin_;
|
|
}
|
|
} else {
|
|
auto pos = static_cast<size_t>(seq_no - offset_);
|
|
auto need_size = pos + 1;
|
|
if (data_array_.size() < need_size) {
|
|
data_array_.resize(need_size);
|
|
}
|
|
data_array_[pos].first = std::forward<FromDataT>(data);
|
|
data_array_[pos].second = true;
|
|
if (end_ < seq_no + 1) {
|
|
end_ = seq_no + 1;
|
|
}
|
|
}
|
|
}
|
|
|
|
bool has_events() const {
|
|
return begin_ != end_;
|
|
}
|
|
SeqNo max_unfinished_seq_no() {
|
|
return end_ - 1;
|
|
}
|
|
SeqNo max_finished_seq_no() {
|
|
return begin_ - 1;
|
|
}
|
|
|
|
private:
|
|
SeqNo offset_ = 1;
|
|
SeqNo begin_ = 1;
|
|
SeqNo end_ = 1;
|
|
std::vector<std::pair<DataT, bool>> data_array_;
|
|
};
|
|
|
|
} // namespace td
|