OS lb-8

semester-4/ОС/lb-8/src/task-1/queue.hpp

@@ -0,0 +1,45 @@
+#pragma once
+
+#include <condition_variable>
+#include <mutex>
+#include <optional>
+#include <queue>
+
+namespace th_safe {
+template <typename T> class queue {
+public:
+  void push(const T &value) {
+    {
+      std::lock_guard lock(mutex_);
+      queue_.push(value);
+    }
+    cond_var_.notify_one();
+  }
+
+  std::optional<T> pop() {
+    std::unique_lock lock(mutex_);
+    cond_var_.wait(lock, [this] { return !queue_.empty() || done_; });
+
+    if (queue_.empty())
+      return std::nullopt;
+
+    T value = queue_.front();
+    queue_.pop();
+    return value;
+  }
+
+  void done() {
+    {
+      std::lock_guard lock(mutex_);
+      done_ = true;
+    }
+    cond_var_.notify_all();
+  }
+
+private:
+  std::queue<T> queue_;
+  std::mutex mutex_;
+  std::condition_variable cond_var_;
+  bool done_ = false;
+};
+} // namespace th_safe