diff --git a/src/core/message_queue.cpp b/src/core/message_queue.cpp index 502d89a..e9f38cb 100644 --- a/src/core/message_queue.cpp +++ b/src/core/message_queue.cpp @@ -27,12 +27,16 @@ void message_queue::add(const message &message) { const std::lock_guard guard(m_mutex); m_queue.push(message); + for (const message_queue_size_callback &callback : m_callbacks) + callback(m_queue.size()); } void message_queue::add(message &&message) { const std::lock_guard guard(m_mutex); m_queue.push(message); + for (const message_queue_size_callback &callback : m_callbacks) + callback(m_queue.size()); } void message_queue::process_direct_message_event(dpp::cluster *bot, bot::settings::settings *settings, const dpp::message_context_menu_t &event) diff --git a/src/core/submit_queue.cpp b/src/core/submit_queue.cpp index 74962e9..3fe5057 100644 --- a/src/core/submit_queue.cpp +++ b/src/core/submit_queue.cpp @@ -26,12 +26,16 @@ void submit_queue::add(const translated_message &message) { const std::lock_guard guard(m_mutex); m_queue.push(message); + for (const submit_queue_size_callback &callback : m_callbacks) + callback(m_queue.size()); } void submit_queue::add(translated_message &&message) { const std::lock_guard guard(m_mutex); m_queue.push(message); + for (const submit_queue_size_callback &callback : m_callbacks) + callback(m_queue.size()); } void submit_queue::run(dpp::cluster *bot)