message_pool.hpp

Go to the documentation of this file.

#ifndef KAGOME_MESSAGE_POOL_HPP
#define KAGOME_MESSAGE_POOL_HPP

#include <cstddef>
#include <cstdint>
#include <optional>
#include <string>
#include <unordered_set>

#include <boost/asio/buffer.hpp>
#include "common/spin_lock.hpp"

namespace kagome::telemetry {

  using MessageHandle = std::size_t;

  class MessagePool {
   public:
    using RefCount = int16_t;
    MessagePool(std::size_t entry_size_bytes, std::size_t entries_count);

    std::optional<MessageHandle> push(const std::string &message,
                                      int16_t ref_count);

    RefCount add_ref(MessageHandle handle);

    RefCount release(MessageHandle handle);

    boost::asio::mutable_buffer operator[](MessageHandle handle) const;

    std::size_t capacity() const;

   private:
    std::optional<MessageHandle> nextFreeSlot();

    struct Record {
      std::vector<uint8_t> data;
      std::size_t data_size = 0;
      std::size_t ref_count = 0;
    };

    const std::size_t entry_size_;
    const std::size_t entries_count_;
    std::vector<Record> pool_;
    std::unordered_set<std::size_t> free_slots_;
    common::spin_lock mutex_;
  };

}  // namespace kagome::telemetry

#endif  // KAGOME_MESSAGE_POOL_HPP