icinga2/lib/remote/jsonrpcconnection-heartbeat.cpp

/* 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(20));
		m_HeartbeatTimer.async_wait(yc[ec]);

		if (m_ShuttingDown) {
			break;
		}

		if (m_NextHeartbeat != 0 && m_NextHeartbeat < Utility::GetTime()) {
			{
				Log logMsg (LogWarning, "JsonRpcConnection");

				if (m_Endpoint) {
					logMsg << "Client for endpoint '" << m_Endpoint->GetName() << "'";
				} else {
					logMsg << "Anonymous client";
				}

				logMsg << " has requested heartbeat message but hasn't responded in time. Closing connection.";
			}

			Disconnect();
			break;
		}

		if (m_Endpoint) {
			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;
}