SafeQueue.hpp

Go to the documentation of this file.

/*
** EPITECH PROJECT, 2023
** Plazza
** File description:
** SafeQueue
*/
 
#ifndef SAFEQUEUE_HPP_
#define SAFEQUEUE_HPP_
#include "ISafeQueue.hpp"
#include <condition_variable>
#include <mutex>
#include <queue>
 
template <typename T>
class SafeQueue : public ISafeQueue<T>
{
  public:
    SafeQueue() = default;
    ~SafeQueue() = default;
    void push(T value) override
    {
        std::unique_lock<std::mutex> l(mutex);
        queue.push(value);
        cond.notify_one();
    }
 
    bool tryPop(T &value) override
    {
        std::unique_lock<std::mutex> l(mutex);
        if (queue.size() == 0)
            return false;
        value = queue.front();
        queue.pop();
        return true;
    }
 
    T pop() override
    {
        std::unique_lock<std::mutex> l(mutex);
        if (queue.size() == 0)
            cond.wait(l);
        T value = queue.front();
        queue.pop();
        return value;
    }
 
    size_t size() const { return queue.size(); }
 
  private:
    std::queue<T> queue;
    std::mutex mutex;
    std::condition_variable cond;
};
#endif