tdlight/tdutils/td/utils/WaitFreeVector.h

//
// Copyright Aliaksei Levin (levlam@telegram.org), Arseny Smirnov (arseny30@gmail.com) 2014-2024
//
// 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