mirror of https://github.com/Icinga/icinga2.git
60 lines
1.5 KiB
C++
60 lines
1.5 KiB
C++
/* Icinga 2 | (c) 2012 Icinga GmbH | GPLv2+ */
|
|
|
|
#include "remote/jsonrpcconnection.hpp"
|
|
#include "remote/messageorigin.hpp"
|
|
#include "remote/apifunction.hpp"
|
|
#include "base/initialize.hpp"
|
|
#include "base/configtype.hpp"
|
|
#include "base/logger.hpp"
|
|
#include "base/utility.hpp"
|
|
#include <boost/asio/spawn.hpp>
|
|
#include <boost/date_time/posix_time/posix_time_duration.hpp>
|
|
#include <boost/system/system_error.hpp>
|
|
|
|
using namespace icinga;
|
|
|
|
REGISTER_APIFUNCTION(Heartbeat, event, &JsonRpcConnection::HeartbeatAPIHandler);
|
|
|
|
void JsonRpcConnection::HandleAndWriteHeartbeats(boost::asio::yield_context yc)
|
|
{
|
|
boost::system::error_code ec;
|
|
|
|
for (;;) {
|
|
m_HeartbeatTimer.expires_from_now(boost::posix_time::seconds(10));
|
|
m_HeartbeatTimer.async_wait(yc[ec]);
|
|
|
|
if (m_ShuttingDown) {
|
|
break;
|
|
}
|
|
|
|
if (m_NextHeartbeat != 0 && m_NextHeartbeat < Utility::GetTime()) {
|
|
Log(LogWarning, "JsonRpcConnection")
|
|
<< "Client for endpoint '" << m_Endpoint->GetName() << "' has requested "
|
|
<< "heartbeat message but hasn't responded in time. Closing connection.";
|
|
|
|
Disconnect();
|
|
break;
|
|
}
|
|
|
|
SendMessageInternal(new Dictionary({
|
|
{ "jsonrpc", "2.0" },
|
|
{ "method", "event::Heartbeat" },
|
|
{ "params", new Dictionary({
|
|
{ "timeout", 120 }
|
|
}) }
|
|
}));
|
|
}
|
|
}
|
|
|
|
Value JsonRpcConnection::HeartbeatAPIHandler(const MessageOrigin::Ptr& origin, const Dictionary::Ptr& params)
|
|
{
|
|
Value vtimeout = params->Get("timeout");
|
|
|
|
if (!vtimeout.IsEmpty()) {
|
|
origin->FromClient->m_NextHeartbeat = Utility::GetTime() + vtimeout;
|
|
}
|
|
|
|
return Empty;
|
|
}
|
|
|