feat(app): add graceful shutdown support to MessageQueue
This commit is contained in:
@@ -2,6 +2,7 @@
|
||||
|
||||
#include <condition_variable>
|
||||
#include <mutex>
|
||||
#include <optional>
|
||||
#include <queue>
|
||||
|
||||
#include "log_message.hpp"
|
||||
@@ -22,6 +23,8 @@ public:
|
||||
*
|
||||
* Метод является потокобезопасным.
|
||||
*
|
||||
* После остановки очереди новые сообщения не принимаются.
|
||||
*
|
||||
* @param message Сообщение для передачи.
|
||||
*/
|
||||
void push(LogMessage message);
|
||||
@@ -29,15 +32,28 @@ public:
|
||||
/**
|
||||
* @brief Извлекает сообщение из очереди.
|
||||
*
|
||||
* Если очередь пуста, метод блокирует вызывающий
|
||||
* поток до появления нового сообщения.
|
||||
* Если очередь пуста, метод блокирует вызывающий поток
|
||||
* до появления нового сообщения или до остановки очереди.
|
||||
*
|
||||
* После остановки очереди и обработки всех оставшихся сообщений
|
||||
* возвращается std::nullopt.
|
||||
*
|
||||
* Метод является потокобезопасным.
|
||||
*
|
||||
* @return Следующее сообщение из очереди.
|
||||
* @return Следующее сообщение или std::nullopt, если очередь остановлена.
|
||||
*/
|
||||
[[nodiscard]]
|
||||
LogMessage pop();
|
||||
std::optional<LogMessage> pop();
|
||||
|
||||
/**
|
||||
* @brief Останавливает очередь.
|
||||
*
|
||||
* Пробуждает все ожидающие потоки. После вызова stop()
|
||||
* новые сообщения не принимаются.
|
||||
*
|
||||
* Метод является потокобезопасным.
|
||||
*/
|
||||
void stop();
|
||||
|
||||
/**
|
||||
* @brief Проверяет, пуста ли очередь.
|
||||
@@ -56,6 +72,11 @@ private:
|
||||
*/
|
||||
std::queue<LogMessage> queue_;
|
||||
|
||||
/**
|
||||
* @brief Признак остановки очереди.
|
||||
*/
|
||||
bool stopped_{false};
|
||||
|
||||
/**
|
||||
* @brief Мьютекс для синхронизации доступа к очереди.
|
||||
*/
|
||||
|
||||
@@ -5,15 +5,24 @@ namespace app {
|
||||
void MessageQueue::push(LogMessage message) {
|
||||
std::lock_guard<std::mutex> lock(mutex_);
|
||||
|
||||
if (stopped_) {
|
||||
return;
|
||||
}
|
||||
|
||||
queue_.push(std::move(message));
|
||||
|
||||
conditionVariable_.notify_one();
|
||||
}
|
||||
|
||||
LogMessage MessageQueue::pop() {
|
||||
std::optional<LogMessage> MessageQueue::pop() {
|
||||
std::unique_lock<std::mutex> lock(mutex_);
|
||||
|
||||
conditionVariable_.wait(lock, [this]() { return !queue_.empty(); });
|
||||
conditionVariable_.wait(lock,
|
||||
[this]() { return stopped_ || !queue_.empty(); });
|
||||
|
||||
if (queue_.empty()) {
|
||||
return std::nullopt;
|
||||
}
|
||||
|
||||
auto message = std::move(queue_.front());
|
||||
|
||||
@@ -22,6 +31,16 @@ LogMessage MessageQueue::pop() {
|
||||
return message;
|
||||
}
|
||||
|
||||
void MessageQueue::stop() {
|
||||
{
|
||||
std::lock_guard<std::mutex> lock(mutex_);
|
||||
|
||||
stopped_ = true;
|
||||
}
|
||||
|
||||
conditionVariable_.notify_all();
|
||||
}
|
||||
|
||||
bool MessageQueue::empty() const {
|
||||
std::lock_guard<std::mutex> lock(mutex_);
|
||||
|
||||
|
||||
Reference in New Issue
Block a user