ThreadSafeQueue.hpp

Go to the documentation of this file.

/*
** EPITECH PROJECT, 2025
** Created by mael on 08/12/2025.
** File description:
** ThreadSafeQueue.hpp
*/
 
#pragma once
 
#include <condition_variable>
#include <mutex>
#include <optional>
#include <queue>
 
template <typename T>
class ThreadSafeQueue {
   public:
    ThreadSafeQueue() = default;
    ~ThreadSafeQueue() = default;
 
    // Delete copy constructor and assignment
    ThreadSafeQueue(const ThreadSafeQueue &) = delete;
    ThreadSafeQueue &operator=(const ThreadSafeQueue &) = delete;
 
    void push(T item) {
        std::lock_guard<std::mutex> lock(_mutex);
        _queue.push(std::move(item));
        _cv.notify_one();
    }
 
    std::optional<T> tryPop() {
        std::lock_guard<std::mutex> lock(_mutex);
        if (_queue.empty()) {
            return std::nullopt;
        }
        T item = std::move(_queue.front());
        _queue.pop();
        return item;
    }
 
    T pop() {
        std::unique_lock<std::mutex> lock(_mutex);
        _cv.wait(lock, [this] { return !_queue.empty(); });
        T item = std::move(_queue.front());
        _queue.pop();
        return item;
    }
 
    bool empty() const {
        std::lock_guard<std::mutex> lock(_mutex);
        return _queue.empty();
    }
 
    size_t size() const {
        std::lock_guard<std::mutex> lock(_mutex);
        return _queue.size();
    }
 
    void clear() {
        std::lock_guard<std::mutex> lock(_mutex);
        std::queue<T> emptyQueue;
        std::swap(_queue, emptyQueue);
    }
 
   private:
    mutable std::mutex _mutex;
    std::condition_variable _cv;
    std::queue<T> _queue;
};