mpsc_ring_buffer.h

Go to the documentation of this file.

 
#ifndef CPPCOMMON_THREADS_MPSC_RING_BUFFER_H
#define CPPCOMMON_THREADS_MPSC_RING_BUFFER_H
 
#include "threads/spin_lock.h"
#include "threads/spsc_ring_buffer.h"
#include "time/timestamp.h"
 
#include <cstdio>
#include <memory>
#include <thread>
#include <vector>
 
namespace CppCommon {
 
 
class MPSCRingBuffer
{
public:
 
    explicit MPSCRingBuffer(size_t capacity, size_t concurrency = std::thread::hardware_concurrency());
    MPSCRingBuffer(const MPSCRingBuffer&) = delete;
    MPSCRingBuffer(MPSCRingBuffer&&) = delete;
    ~MPSCRingBuffer() = default;
 
    MPSCRingBuffer& operator=(const MPSCRingBuffer&) = delete;
    MPSCRingBuffer& operator=(MPSCRingBuffer&&) = delete;
 
    explicit operator bool() const noexcept { return !empty(); }
 
    bool empty() const noexcept { return (size() == 0); }
    size_t capacity() const noexcept { return _capacity; }
    size_t concurrency() const noexcept { return _concurrency; }
    size_t size() const noexcept;
 
 
    bool Enqueue(const void* data, size_t size);
 
 
    bool Dequeue(void* data, size_t& size);
 
private:
    struct Producer
    {
        SpinLock lock;
        SPSCRingBuffer buffer;
 
        Producer(size_t capacity) : buffer(capacity) {}
    };
 
    size_t _capacity;
    size_t _concurrency;
    std::vector<std::shared_ptr<Producer>> _producers;
    size_t _consumer;
};
 
} // namespace CppCommon
 
#include "mpsc_ring_buffer.inl"
 
#endif // CPPCOMMON_THREADS_MPSC_RING_BUFFER_H