buffer_pool.h

Go to the documentation of this file.

// Copyright (c) Space-ng, inc. All rights reserved.
 
#pragma once
 
#include "sol3/cpp/thread_safe_value.h"
 
#include <cstddef>
#include <cstdint>
#include <memory>
#include <vector>
 
namespace sol3::net {
using sol3::cpp::ThreadSafeValue;
 
class BufferPool {
 public:
  explicit BufferPool(std::size_t reserve_bytes);
  BufferPool(BufferPool const&) = delete;
  BufferPool& operator=(BufferPool const&) = delete;
  BufferPool(BufferPool&&) = delete;
  BufferPool& operator=(BufferPool&&) = delete;
  ~BufferPool() = default;
 
  // The returned vector has size() == 0 and capacity() >= reserve_bytes_
  std::shared_ptr<std::vector<uint8_t>> acquire();
 
 private:
  using FreeList = std::vector<std::unique_ptr<std::vector<uint8_t>>>;
  using ThreadSafeFreeList = ThreadSafeValue<FreeList>;
 
  std::shared_ptr<ThreadSafeFreeList> free_list_ =
      std::make_shared<ThreadSafeFreeList>();
  std::size_t reserve_bytes_;
};
 
}  // namespace sol3::net