WorkStealingQueue.h

Go to the documentation of this file.

 
#ifndef LLU_ASYNC_WORKSTEALINGQUEUE_H
#define LLU_ASYNC_WORKSTEALINGQUEUE_H
 
#include <mutex>
 
namespace LLU::Async {
 
    template<typename BaseQueue>
    class WorkStealingQueue {
        using DataType = typename BaseQueue::value_type;
 
        BaseQueue theQueue;
        mutable std::mutex theMutex;
 
    public:
        void push(DataType data) {
            std::lock_guard<std::mutex> lock(theMutex);
            theQueue.push_front(std::move(data));
        }
 
        [[nodiscard]] bool empty() const {
            std::lock_guard<std::mutex> lock(theMutex);
            return theQueue.empty();
        }
 
        [[nodiscard]] bool tryPop(DataType& res) {
            std::lock_guard<std::mutex> lock(theMutex);
            if (theQueue.empty()) {
                return false;
            }
            res = std::move(theQueue.front());
            theQueue.pop_front();
            return true;
        }
 
        [[nodiscard]] bool trySteal(DataType& res) {
            std::lock_guard<std::mutex> lock(theMutex);
            if (theQueue.empty()) {
                return false;
            }
            res = std::move(theQueue.back());
            theQueue.pop_back();
            return true;
        }
    };
} // namespace LLU::Async
 
#endif    // LLU_ASYNC_WORKSTEALINGQUEUE_H