ThreadSafeQueue.h

Go to the documentation of this file.

#pragma once
 
#include <queue>
#include <mutex>
#include <condition_variable>
#include <chrono>
 
template<typename T>
class ThreadSafeQueue {
private:
    std::queue<T> queue_;
    mutable std::mutex mutex_;
    std::condition_variable cv_;
    bool stopped_ = false;
 
public:
    void push(T item) {
        {
            std::lock_guard<std::mutex> lock(mutex_);
            if (stopped_) return;
            queue_.push(std::move(item));
        }
        cv_.notify_one();
    }
 
    bool pop(T& item, std::chrono::milliseconds timeout = std::chrono::milliseconds(100)) {
        std::unique_lock<std::mutex> lock(mutex_);
 
        if (!cv_.wait_for(lock, timeout, [this] {
            return !queue_.empty() || stopped_;
        })) {
            return false; // Timeout
        }
 
        if (stopped_ && queue_.empty()) {
            return false;
        }
 
        item = std::move(queue_.front());
        queue_.pop();
        return true;
    }
 
    void stop() {
        {
            std::lock_guard<std::mutex> lock(mutex_);
            stopped_ = true;
        }
        cv_.notify_all();
    }
 
    size_t size() const {
        std::lock_guard<std::mutex> lock(mutex_);
        return queue_.size();
    }
};