MCRingBuffer5.hpp

#pragma once

#include <atomic>
#include <array>
#include <cstddef>
#include <new>
#include <type_traits>

template<typename T, std::size_t SIZE>
struct MCRingBuffer5 {
    using value_type = T;
    alignas(128) std::array<std::byte, sizeof(T) * SIZE> buffer;
    alignas(128) std::atomic_size_t tail{ 0 };
    mutable size_t head_cache{ 0 };
    alignas(128) std::atomic_size_t head{ 0 };
    mutable size_t tail_cache{ 0 };

    template<typename... Args>
    int Enqueue(Args&&... args) {
        size_t t = tail.load(std::memory_order_relaxed);
        size_t n = (t + 1);
        if (n == SIZE) n = 0;
        size_t h = head_cache;
        if (n == h && n == (head_cache = head.load(std::memory_order_acquire)))
            return 0;
        new(buffer.data() + t * sizeof(T)) T(std::forward<Args>(args)...);
        tail.store(n, std::memory_order_release);
        return 1;
    }

    template<typename Callable>
    int Dequeue(Callable&& f) {
        size_t h = head.load(std::memory_order_relaxed);
        size_t t = tail_cache;
        if (h == t && h == (tail_cache = tail.load(std::memory_order_acquire)))
            return 0;
        T* elem = std::launder(reinterpret_cast<T*>(
            buffer.data() + h * sizeof(T)));
        std::invoke(std::forward<Callable>(f), std::move(*elem));
        elem->~T();
        h += 1;
        if (h == SIZE) h = 0;
        head.store(h, std::memory_order_release);
        return 1;
    }
};