tdlight/tdutils/td/utils/WaitFreeVector.h
2023-01-01 00:28:08 +03:00

70 lines
1.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 <utility>
namespace td {
template <class T>
class WaitFreeVector {
static constexpr size_t MAX_VECTOR_SIZE = (1 << 15) - 10;
vector<vector<T>> storage_;
public:
template <class... ArgsT>
void emplace_back(ArgsT &&...args) {
if (storage_.empty() || storage_.back().size() == MAX_VECTOR_SIZE) {
storage_.emplace_back();
}
storage_.back().emplace_back(std::forward<ArgsT>(args)...);
}
void pop_back() {
storage_.back().pop_back();
if (storage_.back().empty()) {
storage_.pop_back();
}
}
void push_back(T &&value) {
emplace_back(std::move(value));
}
void push_back(const T &value) {
emplace_back(value);
}
const T &back() const {
return storage_.back().back();
}
T &operator[](size_t index) {
return storage_[index / MAX_VECTOR_SIZE][index % MAX_VECTOR_SIZE];
}
const T &operator[](size_t index) const {
return storage_[index / MAX_VECTOR_SIZE][index % MAX_VECTOR_SIZE];
}
size_t size() const {
if (storage_.empty()) {
return 0;
}
return (storage_.size() - 1) * MAX_VECTOR_SIZE + storage_.back().size();
}
bool empty() const {
return storage_.empty();
}
};
} // namespace td