Adding syncqueue

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