tyler.durden/icinga2
1
0
Fork 0
mirror of https://github.com/Icinga/icinga2.git synced 2025-04-08 17:05:25 +02:00
Code Issues Packages Projects Releases Wiki Activity

Count still queued messages per JsonRpcConnection

Browse Source
This commit is contained in:
Alexander A. Klimov 2024-12-11 16:57:50 +01:00
parent 2b4abee33b
commit 1a723a477a
2 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
lib/remote/jsonrpcconnection.cpp
View File

@ -154,6 +154,7 @@ void JsonRpcConnection::WriteOutgoingMessages(boost::asio::yield_context yc)
}
size_t bytesSent = JsonRpc::SendRawMessage(m_Stream, message, yc);
m_PendingOutgoingMessages--;
if (m_Endpoint) {
m_Endpoint->AddMessageSent(bytesSent);
@ -230,6 +231,7 @@ void JsonRpcConnection::SendRawMessage(const String& message)
m_OutgoingMessagesQueue.emplace_back(message);
m_OutgoingMessagesQueued.Set();
m_PendingOutgoingMessages++;
});
}
@ -241,6 +243,7 @@ void JsonRpcConnection::SendMessageInternal(const Dictionary::Ptr& message)
m_OutgoingMessagesQueue.emplace_back(JsonEncode(message));
m_OutgoingMessagesQueued.Set();
m_PendingOutgoingMessages++;
}
void JsonRpcConnection::Disconnect()

6
lib/remote/jsonrpcconnection.hpp
View File

@ -54,6 +54,11 @@ public:
Shared<AsioTlsStream>::Ptr GetStream() const;
ConnectionRole GetRole() const;
auto GetPendingOutgoingMessages() const noexcept
{
return m_PendingOutgoingMessages.load();
}
void Disconnect();
void SendMessage(const Dictionary::Ptr& request);
@ -76,6 +81,7 @@ private:
boost::asio::io_context::strand m_IoStrand;
std::vector<String> m_OutgoingMessagesQueue;
AsioConditionVariable m_OutgoingMessagesQueued;
Atomic<decltype(m_OutgoingMessagesQueue)::size_type, std::memory_order_relaxed> m_PendingOutgoingMessages {0};
AsioConditionVariable m_WriterDone;
Atomic<bool> m_ShuttingDown;
boost::asio::deadline_timer m_CheckLivenessTimer, m_HeartbeatTimer;