diff options
Diffstat (limited to 'yage/util/syncqueue.h')
-rw-r--r-- | yage/util/syncqueue.h | 81 |
1 files changed, 81 insertions, 0 deletions
diff --git a/yage/util/syncqueue.h b/yage/util/syncqueue.h new file mode 100644 index 00000000..a23b1857 --- /dev/null +++ b/yage/util/syncqueue.h @@ -0,0 +1,81 @@ +#ifndef YAGE_UTIL_SYNCQUEUE_H +#define YAGE_UTIL_SYNCQUEUE_H + +#include <condition_variable> +#include <mutex> +#include <queue> + +namespace yage +{ + +template <typename T> +class SyncQueue +{ +public: + SyncQueue() = default; + SyncQueue(const SyncQueue &) = delete; + SyncQueue &operator=(const SyncQueue &) = delete; + + T pop(); + void pop(T &item); + void push(const T &item); + void push(T &&item); + +private: + std::queue<T> queue_; + std::mutex mutex_; + std::condition_variable cond_; +}; + +// Template definitions + +template <typename T> +T SyncQueue<T>::pop() +{ + std::unique_lock<std::mutex> mlock(mutex_); + + while (queue_.empty()) { + cond_.wait(mlock); + } + + auto item = queue_.front(); + queue_.pop(); + return item; +} + +template <typename T> +void SyncQueue<T>::pop(T &item) +{ + std::unique_lock<std::mutex> mlock(mutex_); + + while (queue_.empty()) { + cond_.wait(mlock); + } + + item = queue_.front(); + queue_.pop(); +} + +template <typename T> +void SyncQueue<T>::push(const T &item) +{ + std::unique_lock<std::mutex> mlock(mutex_); + + queue_.push(item); + mlock.unlock(); + cond_.notify_one(); +} + +template <typename T> +void SyncQueue<T>::push(T &&item) +{ + std::unique_lock<std::mutex> mlock(mutex_); + + queue_.push(std::move(item)); + mlock.unlock(); + cond_.notify_one(); +} + +} // namespace yage + +#endif |