2013-08-27 12:21:41 +02:00
|
|
|
/******************************************************************************
|
|
|
|
* Icinga 2 *
|
|
|
|
* Copyright (C) 2012 Icinga Development Team (http://www.icinga.org/) *
|
|
|
|
* *
|
|
|
|
* This program is free software; you can redistribute it and/or *
|
|
|
|
* modify it under the terms of the GNU General Public License *
|
|
|
|
* as published by the Free Software Foundation; either version 2 *
|
|
|
|
* of the License, or (at your option) any later version. *
|
|
|
|
* *
|
|
|
|
* This program is distributed in the hope that it will be useful, *
|
|
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of *
|
|
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the *
|
|
|
|
* GNU General Public License for more details. *
|
|
|
|
* *
|
|
|
|
* You should have received a copy of the GNU General Public License *
|
|
|
|
* along with this program; if not, write to the Free Software Foundation *
|
|
|
|
* Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA. *
|
|
|
|
******************************************************************************/
|
|
|
|
|
|
|
|
#include "cluster/clustercomponent.h"
|
|
|
|
#include "cluster/endpoint.h"
|
|
|
|
#include "base/dynamictype.h"
|
|
|
|
#include "base/logger_fwd.h"
|
|
|
|
#include "base/objectlock.h"
|
|
|
|
#include "base/networkstream.h"
|
|
|
|
#include <boost/smart_ptr/make_shared.hpp>
|
|
|
|
|
|
|
|
using namespace icinga;
|
|
|
|
|
|
|
|
REGISTER_TYPE(ClusterComponent);
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Starts the component.
|
|
|
|
*/
|
|
|
|
void ClusterComponent::Start(void)
|
|
|
|
{
|
|
|
|
DynamicObject::Start();
|
|
|
|
|
|
|
|
/* set up SSL context */
|
|
|
|
shared_ptr<X509> cert = GetX509Certificate(GetCertificateFile());
|
|
|
|
m_Identity = GetCertificateCN(cert);
|
|
|
|
Log(LogInformation, "cluster", "My identity: " + m_Identity);
|
|
|
|
|
|
|
|
m_SSLContext = MakeSSLContext(GetCertificateFile(), GetCertificateFile(), GetCAFile());
|
|
|
|
|
|
|
|
/* create the primary JSON-RPC listener */
|
|
|
|
if (!GetBindPort().IsEmpty())
|
|
|
|
AddListener(GetBindPort());
|
|
|
|
|
2013-08-30 09:34:58 +02:00
|
|
|
m_ClusterTimer = boost::make_shared<Timer>();
|
|
|
|
m_ClusterTimer->OnTimerExpired.connect(boost::bind(&ClusterComponent::ClusterTimerHandler, this));
|
|
|
|
m_ClusterTimer->SetInterval(5);
|
|
|
|
m_ClusterTimer->Start();
|
2013-08-27 12:21:41 +02:00
|
|
|
|
2013-08-28 11:12:20 +02:00
|
|
|
Service::OnNewCheckResult.connect(bind(&ClusterComponent::CheckResultHandler, this, _1, _2, _3));
|
|
|
|
Service::OnNextCheckChanged.connect(bind(&ClusterComponent::NextCheckChangedHandler, this, _1, _2, _3));
|
2013-08-28 14:59:41 +02:00
|
|
|
Notification::OnNextNotificationChanged.connect(bind(&ClusterComponent::NextNotificationChangedHandler, this, _1, _2, _3));
|
2013-08-28 11:12:20 +02:00
|
|
|
Service::OnForceNextCheckChanged.connect(bind(&ClusterComponent::ForceNextCheckChangedHandler, this, _1, _2, _3));
|
2013-08-29 11:37:51 +02:00
|
|
|
Service::OnForceNextNotificationChanged.connect(bind(&ClusterComponent::ForceNextNotificationChangedHandler, this, _1, _2, _3));
|
2013-08-28 11:12:20 +02:00
|
|
|
Service::OnEnableActiveChecksChanged.connect(bind(&ClusterComponent::EnableActiveChecksChangedHandler, this, _1, _2, _3));
|
|
|
|
Service::OnEnablePassiveChecksChanged.connect(bind(&ClusterComponent::EnablePassiveChecksChangedHandler, this, _1, _2, _3));
|
2013-08-29 13:06:36 +02:00
|
|
|
Service::OnEnableNotificationsChanged.connect(bind(&ClusterComponent::EnableNotificationsChangedHandler, this, _1, _2, _3));
|
|
|
|
Service::OnEnableFlappingChanged.connect(bind(&ClusterComponent::EnableFlappingChangedHandler, this, _1, _2, _3));
|
2013-08-28 14:59:41 +02:00
|
|
|
Service::OnCommentAdded.connect(bind(&ClusterComponent::CommentAddedHandler, this, _1, _2, _3));
|
|
|
|
Service::OnCommentRemoved.connect(bind(&ClusterComponent::CommentRemovedHandler, this, _1, _2, _3));
|
2013-08-28 16:08:22 +02:00
|
|
|
Service::OnDowntimeAdded.connect(bind(&ClusterComponent::DowntimeAddedHandler, this, _1, _2, _3));
|
|
|
|
Service::OnDowntimeRemoved.connect(bind(&ClusterComponent::DowntimeRemovedHandler, this, _1, _2, _3));
|
2013-08-29 13:48:18 +02:00
|
|
|
Service::OnAcknowledgementSet.connect(bind(&ClusterComponent::AcknowledgementSetHandler, this, _1, _2, _3, _4, _5, _6));
|
|
|
|
Service::OnAcknowledgementCleared.connect(bind(&ClusterComponent::AcknowledgementClearedHandler, this, _1, _2));
|
2013-08-28 11:12:20 +02:00
|
|
|
|
2013-08-27 12:21:41 +02:00
|
|
|
Endpoint::OnMessageReceived.connect(bind(&ClusterComponent::MessageHandler, this, _1, _2));
|
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Stops the component.
|
|
|
|
*/
|
|
|
|
void ClusterComponent::Stop(void)
|
|
|
|
{
|
|
|
|
/* Nothing to do here. */
|
|
|
|
}
|
|
|
|
|
|
|
|
String ClusterComponent::GetCertificateFile(void) const
|
|
|
|
{
|
|
|
|
ObjectLock olock(this);
|
|
|
|
|
|
|
|
return m_CertPath;
|
|
|
|
}
|
|
|
|
|
|
|
|
String ClusterComponent::GetCAFile(void) const
|
|
|
|
{
|
|
|
|
ObjectLock olock(this);
|
|
|
|
|
|
|
|
return m_CAPath;
|
|
|
|
}
|
|
|
|
|
|
|
|
String ClusterComponent::GetBindHost(void) const
|
|
|
|
{
|
|
|
|
ObjectLock olock(this);
|
|
|
|
|
|
|
|
return m_BindHost;
|
|
|
|
}
|
|
|
|
|
|
|
|
String ClusterComponent::GetBindPort(void) const
|
|
|
|
{
|
|
|
|
ObjectLock olock(this);
|
|
|
|
|
|
|
|
return m_BindPort;
|
|
|
|
}
|
|
|
|
|
|
|
|
Array::Ptr ClusterComponent::GetPeers(void) const
|
|
|
|
{
|
|
|
|
ObjectLock olock(this);
|
|
|
|
|
|
|
|
return m_Peers;
|
|
|
|
}
|
|
|
|
|
|
|
|
shared_ptr<SSL_CTX> ClusterComponent::GetSSLContext(void) const
|
|
|
|
{
|
|
|
|
ObjectLock olock(this);
|
|
|
|
|
|
|
|
return m_SSLContext;
|
|
|
|
}
|
|
|
|
|
|
|
|
String ClusterComponent::GetIdentity(void) const
|
|
|
|
{
|
|
|
|
ObjectLock olock(this);
|
|
|
|
|
|
|
|
return m_Identity;
|
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Creates a new JSON-RPC listener on the specified port.
|
|
|
|
*
|
|
|
|
* @param service The port to listen on.
|
|
|
|
*/
|
|
|
|
void ClusterComponent::AddListener(const String& service)
|
|
|
|
{
|
|
|
|
ObjectLock olock(this);
|
|
|
|
|
|
|
|
shared_ptr<SSL_CTX> sslContext = m_SSLContext;
|
|
|
|
|
|
|
|
if (!sslContext)
|
|
|
|
BOOST_THROW_EXCEPTION(std::logic_error("SSL context is required for AddListener()"));
|
|
|
|
|
|
|
|
std::ostringstream s;
|
|
|
|
s << "Adding new listener: port " << service;
|
|
|
|
Log(LogInformation, "cluster", s.str());
|
|
|
|
|
|
|
|
TcpSocket::Ptr server = boost::make_shared<TcpSocket>();
|
|
|
|
server->Bind(service, AF_INET6);
|
|
|
|
|
|
|
|
boost::thread thread(boost::bind(&ClusterComponent::ListenerThreadProc, this, server));
|
|
|
|
thread.detach();
|
|
|
|
|
|
|
|
m_Servers.insert(server);
|
|
|
|
}
|
|
|
|
|
|
|
|
void ClusterComponent::ListenerThreadProc(const Socket::Ptr& server)
|
|
|
|
{
|
|
|
|
server->Listen();
|
|
|
|
|
|
|
|
for (;;) {
|
|
|
|
Socket::Ptr client = server->Accept();
|
|
|
|
|
|
|
|
try {
|
|
|
|
NewClientHandler(client, TlsRoleServer);
|
|
|
|
} catch (const std::exception& ex) {
|
|
|
|
std::stringstream message;
|
|
|
|
message << "Error for new JSON-RPC socket: " << boost::diagnostic_information(ex);
|
|
|
|
Log(LogInformation, "cluster", message.str());
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Creates a new JSON-RPC client and connects to the specified host and port.
|
|
|
|
*
|
|
|
|
* @param node The remote host.
|
|
|
|
* @param service The remote port.
|
|
|
|
*/
|
|
|
|
void ClusterComponent::AddConnection(const String& node, const String& service) {
|
|
|
|
{
|
|
|
|
ObjectLock olock(this);
|
|
|
|
|
|
|
|
shared_ptr<SSL_CTX> sslContext = m_SSLContext;
|
|
|
|
|
|
|
|
if (!sslContext)
|
|
|
|
BOOST_THROW_EXCEPTION(std::logic_error("SSL context is required for AddConnection()"));
|
|
|
|
}
|
|
|
|
|
|
|
|
TcpSocket::Ptr client = boost::make_shared<TcpSocket>();
|
|
|
|
|
|
|
|
try {
|
|
|
|
client->Connect(node, service);
|
|
|
|
NewClientHandler(client, TlsRoleClient);
|
|
|
|
} catch (const std::exception& ex) {
|
|
|
|
Log(LogInformation, "cluster", "Could not connect to " + node + ":" + service + ": " + ex.what());
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Processes a new client connection.
|
|
|
|
*
|
|
|
|
* @param client The new client.
|
|
|
|
*/
|
|
|
|
void ClusterComponent::NewClientHandler(const Socket::Ptr& client, TlsRole role)
|
|
|
|
{
|
|
|
|
NetworkStream::Ptr netStream = boost::make_shared<NetworkStream>(client);
|
|
|
|
|
|
|
|
TlsStream::Ptr tlsStream = boost::make_shared<TlsStream>(netStream, role, m_SSLContext);
|
|
|
|
tlsStream->Handshake();
|
|
|
|
|
|
|
|
shared_ptr<X509> cert = tlsStream->GetPeerCertificate();
|
|
|
|
String identity = GetCertificateCN(cert);
|
|
|
|
|
|
|
|
Log(LogInformation, "cluster", "New client connection for identity '" + identity + "'");
|
|
|
|
|
|
|
|
Endpoint::Ptr endpoint = Endpoint::GetByName(identity);
|
|
|
|
|
|
|
|
if (!endpoint) {
|
|
|
|
Log(LogInformation, "cluster", "Closing endpoint '" + identity + "': No configuration available.");
|
|
|
|
tlsStream->Close();
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
endpoint->SetClient(tlsStream);
|
|
|
|
}
|
|
|
|
|
2013-08-30 09:34:58 +02:00
|
|
|
void ClusterComponent::ClusterTimerHandler(void)
|
2013-08-27 12:21:41 +02:00
|
|
|
{
|
2013-08-30 09:34:58 +02:00
|
|
|
/* broadcast a heartbeat message */
|
|
|
|
Dictionary::Ptr message = boost::make_shared<Dictionary>();
|
|
|
|
message->Set("jsonrpc", "2.0");
|
|
|
|
message->Set("method", "cluster::HeartBeat");
|
|
|
|
|
|
|
|
BOOST_FOREACH(const Endpoint::Ptr& endpoint, DynamicType::GetObjects<Endpoint>()) {
|
|
|
|
endpoint->SendMessage(message);
|
|
|
|
}
|
|
|
|
|
2013-08-27 12:21:41 +02:00
|
|
|
Array::Ptr peers = GetPeers();
|
|
|
|
|
|
|
|
if (!peers)
|
|
|
|
return;
|
|
|
|
|
|
|
|
ObjectLock olock(peers);
|
|
|
|
BOOST_FOREACH(const String& peer, peers) {
|
|
|
|
Endpoint::Ptr endpoint = Endpoint::GetByName(peer);
|
|
|
|
|
|
|
|
if (!endpoint)
|
|
|
|
continue;
|
|
|
|
|
|
|
|
if (endpoint->IsConnected())
|
|
|
|
continue;
|
|
|
|
|
|
|
|
String host, port;
|
|
|
|
host = endpoint->GetHost();
|
|
|
|
port = endpoint->GetPort();
|
|
|
|
|
|
|
|
if (host.IsEmpty() || port.IsEmpty()) {
|
|
|
|
Log(LogWarning, "cluster", "Can't reconnect "
|
|
|
|
"to endpoint '" + endpoint->GetName() + "': No "
|
|
|
|
"host/port information.");
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
|
|
|
|
Log(LogInformation, "cluster", "Attempting to reconnect to cluster endpoint '" + endpoint->GetName() + "' via '" + host + ":" + port + "'.");
|
|
|
|
AddConnection(host, port);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-08-28 11:12:20 +02:00
|
|
|
void ClusterComponent::CheckResultHandler(const Service::Ptr& service, const Dictionary::Ptr& cr, const String& authority)
|
2013-08-27 12:21:41 +02:00
|
|
|
{
|
2013-08-28 11:12:20 +02:00
|
|
|
if (!authority.IsEmpty() && authority != GetIdentity())
|
2013-08-27 12:21:41 +02:00
|
|
|
return;
|
|
|
|
|
|
|
|
Dictionary::Ptr params = boost::make_shared<Dictionary>();
|
|
|
|
params->Set("service", service->GetName());
|
|
|
|
params->Set("check_result", cr);
|
|
|
|
|
|
|
|
Dictionary::Ptr message = boost::make_shared<Dictionary>();
|
|
|
|
message->Set("jsonrpc", "2.0");
|
|
|
|
message->Set("method", "cluster::CheckResult");
|
|
|
|
message->Set("params", params);
|
|
|
|
|
|
|
|
BOOST_FOREACH(const Endpoint::Ptr& endpoint, DynamicType::GetObjects<Endpoint>()) {
|
|
|
|
endpoint->SendMessage(message);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-08-28 11:12:20 +02:00
|
|
|
void ClusterComponent::NextCheckChangedHandler(const Service::Ptr& service, double nextCheck, const String& authority)
|
|
|
|
{
|
|
|
|
if (!authority.IsEmpty() && authority != GetIdentity())
|
|
|
|
return;
|
|
|
|
|
|
|
|
Dictionary::Ptr params = boost::make_shared<Dictionary>();
|
|
|
|
params->Set("service", service->GetName());
|
|
|
|
params->Set("next_check", nextCheck);
|
|
|
|
|
|
|
|
Dictionary::Ptr message = boost::make_shared<Dictionary>();
|
|
|
|
message->Set("jsonrpc", "2.0");
|
|
|
|
message->Set("method", "cluster::SetNextCheck");
|
|
|
|
message->Set("params", params);
|
|
|
|
|
|
|
|
BOOST_FOREACH(const Endpoint::Ptr& endpoint, DynamicType::GetObjects<Endpoint>()) {
|
|
|
|
endpoint->SendMessage(message);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-08-28 14:59:41 +02:00
|
|
|
void ClusterComponent::NextNotificationChangedHandler(const Notification::Ptr& notification, double nextNotification, const String& authority)
|
|
|
|
{
|
|
|
|
if (!authority.IsEmpty() && authority != GetIdentity())
|
|
|
|
return;
|
|
|
|
|
|
|
|
Dictionary::Ptr params = boost::make_shared<Dictionary>();
|
|
|
|
params->Set("notification", notification->GetName());
|
|
|
|
params->Set("next_notification", nextNotification);
|
|
|
|
|
|
|
|
Dictionary::Ptr message = boost::make_shared<Dictionary>();
|
|
|
|
message->Set("jsonrpc", "2.0");
|
|
|
|
message->Set("method", "cluster::SetNextNotification");
|
|
|
|
message->Set("params", params);
|
|
|
|
|
|
|
|
BOOST_FOREACH(const Endpoint::Ptr& endpoint, DynamicType::GetObjects<Endpoint>()) {
|
|
|
|
endpoint->SendMessage(message);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-08-28 11:12:20 +02:00
|
|
|
void ClusterComponent::ForceNextCheckChangedHandler(const Service::Ptr& service, bool forced, const String& authority)
|
|
|
|
{
|
|
|
|
if (!authority.IsEmpty() && authority != GetIdentity())
|
|
|
|
return;
|
|
|
|
|
|
|
|
Dictionary::Ptr params = boost::make_shared<Dictionary>();
|
|
|
|
params->Set("service", service->GetName());
|
|
|
|
params->Set("forced", forced);
|
|
|
|
|
|
|
|
Dictionary::Ptr message = boost::make_shared<Dictionary>();
|
|
|
|
message->Set("jsonrpc", "2.0");
|
|
|
|
message->Set("method", "cluster::SetForceNextCheck");
|
|
|
|
message->Set("params", params);
|
|
|
|
|
|
|
|
BOOST_FOREACH(const Endpoint::Ptr& endpoint, DynamicType::GetObjects<Endpoint>()) {
|
|
|
|
endpoint->SendMessage(message);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-08-29 11:37:51 +02:00
|
|
|
void ClusterComponent::ForceNextNotificationChangedHandler(const Service::Ptr& service, bool forced, const String& authority)
|
|
|
|
{
|
|
|
|
if (!authority.IsEmpty() && authority != GetIdentity())
|
|
|
|
return;
|
|
|
|
|
|
|
|
Dictionary::Ptr params = boost::make_shared<Dictionary>();
|
|
|
|
params->Set("service", service->GetName());
|
|
|
|
params->Set("forced", forced);
|
|
|
|
|
|
|
|
Dictionary::Ptr message = boost::make_shared<Dictionary>();
|
|
|
|
message->Set("jsonrpc", "2.0");
|
|
|
|
message->Set("method", "cluster::SetForceNextNotification");
|
|
|
|
message->Set("params", params);
|
|
|
|
|
|
|
|
BOOST_FOREACH(const Endpoint::Ptr& endpoint, DynamicType::GetObjects<Endpoint>()) {
|
|
|
|
endpoint->SendMessage(message);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-08-28 11:12:20 +02:00
|
|
|
void ClusterComponent::EnableActiveChecksChangedHandler(const Service::Ptr& service, bool enabled, const String& authority)
|
|
|
|
{
|
|
|
|
if (!authority.IsEmpty() && authority != GetIdentity())
|
|
|
|
return;
|
|
|
|
|
|
|
|
Dictionary::Ptr params = boost::make_shared<Dictionary>();
|
|
|
|
params->Set("service", service->GetName());
|
|
|
|
params->Set("enabled", enabled);
|
|
|
|
|
|
|
|
Dictionary::Ptr message = boost::make_shared<Dictionary>();
|
|
|
|
message->Set("jsonrpc", "2.0");
|
|
|
|
message->Set("method", "cluster::SetEnableActiveChecks");
|
|
|
|
message->Set("params", params);
|
|
|
|
|
|
|
|
BOOST_FOREACH(const Endpoint::Ptr& endpoint, DynamicType::GetObjects<Endpoint>()) {
|
|
|
|
endpoint->SendMessage(message);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
void ClusterComponent::EnablePassiveChecksChangedHandler(const Service::Ptr& service, bool enabled, const String& authority)
|
|
|
|
{
|
|
|
|
if (!authority.IsEmpty() && authority != GetIdentity())
|
|
|
|
return;
|
|
|
|
|
|
|
|
Dictionary::Ptr params = boost::make_shared<Dictionary>();
|
|
|
|
params->Set("service", service->GetName());
|
|
|
|
params->Set("enabled", enabled);
|
|
|
|
|
|
|
|
Dictionary::Ptr message = boost::make_shared<Dictionary>();
|
|
|
|
message->Set("jsonrpc", "2.0");
|
|
|
|
message->Set("method", "cluster::SetEnablePassiveChecks");
|
|
|
|
message->Set("params", params);
|
|
|
|
|
|
|
|
BOOST_FOREACH(const Endpoint::Ptr& endpoint, DynamicType::GetObjects<Endpoint>()) {
|
|
|
|
endpoint->SendMessage(message);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-08-29 13:06:36 +02:00
|
|
|
void ClusterComponent::EnableNotificationsChangedHandler(const Service::Ptr& service, bool enabled, const String& authority)
|
|
|
|
{
|
|
|
|
if (!authority.IsEmpty() && authority != GetIdentity())
|
|
|
|
return;
|
|
|
|
|
|
|
|
Dictionary::Ptr params = boost::make_shared<Dictionary>();
|
|
|
|
params->Set("service", service->GetName());
|
|
|
|
params->Set("enabled", enabled);
|
|
|
|
|
|
|
|
Dictionary::Ptr message = boost::make_shared<Dictionary>();
|
|
|
|
message->Set("jsonrpc", "2.0");
|
|
|
|
message->Set("method", "cluster::SetEnableNotifications");
|
|
|
|
message->Set("params", params);
|
|
|
|
|
|
|
|
BOOST_FOREACH(const Endpoint::Ptr& endpoint, DynamicType::GetObjects<Endpoint>()) {
|
|
|
|
endpoint->SendMessage(message);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
void ClusterComponent::EnableFlappingChangedHandler(const Service::Ptr& service, bool enabled, const String& authority)
|
|
|
|
{
|
|
|
|
if (!authority.IsEmpty() && authority != GetIdentity())
|
|
|
|
return;
|
|
|
|
|
|
|
|
Dictionary::Ptr params = boost::make_shared<Dictionary>();
|
|
|
|
params->Set("service", service->GetName());
|
|
|
|
params->Set("enabled", enabled);
|
|
|
|
|
|
|
|
Dictionary::Ptr message = boost::make_shared<Dictionary>();
|
|
|
|
message->Set("jsonrpc", "2.0");
|
|
|
|
message->Set("method", "cluster::SetEnableFlapping");
|
|
|
|
message->Set("params", params);
|
|
|
|
|
|
|
|
BOOST_FOREACH(const Endpoint::Ptr& endpoint, DynamicType::GetObjects<Endpoint>()) {
|
|
|
|
endpoint->SendMessage(message);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-08-28 14:59:41 +02:00
|
|
|
void ClusterComponent::CommentAddedHandler(const Service::Ptr& service, const Dictionary::Ptr& comment, const String& authority)
|
|
|
|
{
|
|
|
|
if (!authority.IsEmpty() && authority != GetIdentity())
|
|
|
|
return;
|
|
|
|
|
|
|
|
Dictionary::Ptr params = boost::make_shared<Dictionary>();
|
|
|
|
params->Set("service", service->GetName());
|
|
|
|
params->Set("comment", comment);
|
|
|
|
|
|
|
|
Dictionary::Ptr message = boost::make_shared<Dictionary>();
|
|
|
|
message->Set("jsonrpc", "2.0");
|
|
|
|
message->Set("method", "cluster::AddComment");
|
|
|
|
message->Set("params", params);
|
|
|
|
|
|
|
|
BOOST_FOREACH(const Endpoint::Ptr& endpoint, DynamicType::GetObjects<Endpoint>()) {
|
|
|
|
endpoint->SendMessage(message);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
void ClusterComponent::CommentRemovedHandler(const Service::Ptr& service, const Dictionary::Ptr& comment, const String& authority)
|
|
|
|
{
|
|
|
|
if (!authority.IsEmpty() && authority != GetIdentity())
|
|
|
|
return;
|
|
|
|
|
|
|
|
Dictionary::Ptr params = boost::make_shared<Dictionary>();
|
|
|
|
params->Set("service", service->GetName());
|
|
|
|
params->Set("id", comment->Get("id"));
|
|
|
|
|
|
|
|
Dictionary::Ptr message = boost::make_shared<Dictionary>();
|
|
|
|
message->Set("jsonrpc", "2.0");
|
|
|
|
message->Set("method", "cluster::RemoveComment");
|
|
|
|
message->Set("params", params);
|
|
|
|
|
|
|
|
BOOST_FOREACH(const Endpoint::Ptr& endpoint, DynamicType::GetObjects<Endpoint>()) {
|
|
|
|
endpoint->SendMessage(message);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-08-28 16:08:22 +02:00
|
|
|
void ClusterComponent::DowntimeAddedHandler(const Service::Ptr& service, const Dictionary::Ptr& downtime, const String& authority)
|
|
|
|
{
|
|
|
|
if (!authority.IsEmpty() && authority != GetIdentity())
|
|
|
|
return;
|
|
|
|
|
|
|
|
Dictionary::Ptr params = boost::make_shared<Dictionary>();
|
|
|
|
params->Set("service", service->GetName());
|
|
|
|
params->Set("downtime", downtime);
|
|
|
|
|
|
|
|
Dictionary::Ptr message = boost::make_shared<Dictionary>();
|
|
|
|
message->Set("jsonrpc", "2.0");
|
|
|
|
message->Set("method", "cluster::AddDowntime");
|
|
|
|
message->Set("params", params);
|
|
|
|
|
|
|
|
BOOST_FOREACH(const Endpoint::Ptr& endpoint, DynamicType::GetObjects<Endpoint>()) {
|
|
|
|
endpoint->SendMessage(message);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
void ClusterComponent::DowntimeRemovedHandler(const Service::Ptr& service, const Dictionary::Ptr& downtime, const String& authority)
|
|
|
|
{
|
|
|
|
if (!authority.IsEmpty() && authority != GetIdentity())
|
|
|
|
return;
|
|
|
|
|
|
|
|
Dictionary::Ptr params = boost::make_shared<Dictionary>();
|
|
|
|
params->Set("service", service->GetName());
|
|
|
|
params->Set("id", downtime->Get("id"));
|
|
|
|
|
|
|
|
Dictionary::Ptr message = boost::make_shared<Dictionary>();
|
|
|
|
message->Set("jsonrpc", "2.0");
|
|
|
|
message->Set("method", "cluster::RemoveDowntime");
|
|
|
|
message->Set("params", params);
|
|
|
|
|
|
|
|
BOOST_FOREACH(const Endpoint::Ptr& endpoint, DynamicType::GetObjects<Endpoint>()) {
|
|
|
|
endpoint->SendMessage(message);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-08-29 13:48:18 +02:00
|
|
|
void ClusterComponent::AcknowledgementSetHandler(const Service::Ptr& service, const String& author, const String& comment, AcknowledgementType type, double expiry, const String& authority)
|
|
|
|
{
|
|
|
|
if (!authority.IsEmpty() && authority != GetIdentity())
|
|
|
|
return;
|
|
|
|
|
|
|
|
Dictionary::Ptr params = boost::make_shared<Dictionary>();
|
|
|
|
params->Set("service", service->GetName());
|
|
|
|
params->Set("author", author);
|
|
|
|
params->Set("comment", comment);
|
|
|
|
params->Set("type", type);
|
|
|
|
params->Set("expiry", expiry);
|
|
|
|
|
|
|
|
Dictionary::Ptr message = boost::make_shared<Dictionary>();
|
|
|
|
message->Set("jsonrpc", "2.0");
|
|
|
|
message->Set("method", "cluster::SetAcknowledgement");
|
|
|
|
message->Set("params", params);
|
|
|
|
|
|
|
|
BOOST_FOREACH(const Endpoint::Ptr& endpoint, DynamicType::GetObjects<Endpoint>()) {
|
|
|
|
endpoint->SendMessage(message);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
void ClusterComponent::AcknowledgementClearedHandler(const Service::Ptr& service, const String& authority)
|
|
|
|
{
|
|
|
|
if (!authority.IsEmpty() && authority != GetIdentity())
|
|
|
|
return;
|
|
|
|
|
|
|
|
Dictionary::Ptr params = boost::make_shared<Dictionary>();
|
|
|
|
params->Set("service", service->GetName());
|
|
|
|
|
|
|
|
Dictionary::Ptr message = boost::make_shared<Dictionary>();
|
|
|
|
message->Set("jsonrpc", "2.0");
|
|
|
|
message->Set("method", "cluster::ClearAcknowledgement");
|
|
|
|
message->Set("params", params);
|
|
|
|
|
|
|
|
BOOST_FOREACH(const Endpoint::Ptr& endpoint, DynamicType::GetObjects<Endpoint>()) {
|
|
|
|
endpoint->SendMessage(message);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-08-28 11:12:20 +02:00
|
|
|
void ClusterComponent::MessageHandler(const Endpoint::Ptr& sender, const Dictionary::Ptr& message)
|
2013-08-27 12:21:41 +02:00
|
|
|
{
|
2013-08-28 11:12:20 +02:00
|
|
|
BOOST_FOREACH(const Endpoint::Ptr& endpoint, DynamicType::GetObjects<Endpoint>()) {
|
|
|
|
if (sender != endpoint)
|
|
|
|
endpoint->SendMessage(message);
|
|
|
|
}
|
|
|
|
|
2013-08-30 09:34:58 +02:00
|
|
|
if (message->Get("method") == "cluster::HeartBeat") {
|
|
|
|
sender->SetSeen(Utility::GetTime());
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2013-08-28 11:12:20 +02:00
|
|
|
Dictionary::Ptr params = message->Get("params");
|
|
|
|
|
|
|
|
if (!params)
|
|
|
|
return;
|
|
|
|
|
2013-08-27 12:21:41 +02:00
|
|
|
if (message->Get("method") == "cluster::CheckResult") {
|
2013-08-28 11:12:20 +02:00
|
|
|
String svc = params->Get("service");
|
2013-08-27 12:21:41 +02:00
|
|
|
|
2013-08-28 11:12:20 +02:00
|
|
|
Service::Ptr service = Service::GetByName(svc);
|
|
|
|
|
|
|
|
if (!service)
|
2013-08-27 12:21:41 +02:00
|
|
|
return;
|
|
|
|
|
|
|
|
Dictionary::Ptr cr = params->Get("check_result");
|
|
|
|
|
|
|
|
if (!cr)
|
|
|
|
return;
|
|
|
|
|
2013-08-28 11:12:20 +02:00
|
|
|
service->ProcessCheckResult(cr, sender->GetName());
|
|
|
|
} else if (message->Get("method") == "cluster::SetNextCheck") {
|
|
|
|
String svc = params->Get("service");
|
|
|
|
|
|
|
|
Service::Ptr service = Service::GetByName(svc);
|
|
|
|
|
|
|
|
if (!service)
|
|
|
|
return;
|
|
|
|
|
|
|
|
double nextCheck = params->Get("next_check");
|
|
|
|
|
|
|
|
service->SetNextCheck(nextCheck, sender->GetName());
|
|
|
|
} else if (message->Get("method") == "cluster::SetForceNextCheck") {
|
|
|
|
String svc = params->Get("service");
|
|
|
|
|
|
|
|
Service::Ptr service = Service::GetByName(svc);
|
|
|
|
|
|
|
|
if (!service)
|
|
|
|
return;
|
|
|
|
|
|
|
|
bool forced = params->Get("forced");
|
|
|
|
|
|
|
|
service->SetForceNextCheck(forced, sender->GetName());
|
2013-08-29 11:37:51 +02:00
|
|
|
} else if (message->Get("method") == "cluster::SetForceNextNotification") {
|
|
|
|
String svc = params->Get("service");
|
|
|
|
|
|
|
|
Service::Ptr service = Service::GetByName(svc);
|
|
|
|
|
|
|
|
if (!service)
|
|
|
|
return;
|
|
|
|
|
|
|
|
bool forced = params->Get("forced");
|
|
|
|
|
|
|
|
service->SetForceNextNotification(forced, sender->GetName());
|
2013-08-28 11:12:20 +02:00
|
|
|
} else if (message->Get("method") == "cluster::SetEnableActiveChecks") {
|
|
|
|
String svc = params->Get("service");
|
|
|
|
|
|
|
|
Service::Ptr service = Service::GetByName(svc);
|
|
|
|
|
|
|
|
if (!service)
|
|
|
|
return;
|
|
|
|
|
|
|
|
bool enabled = params->Get("enabled");
|
|
|
|
|
|
|
|
service->SetEnableActiveChecks(enabled, sender->GetName());
|
|
|
|
} else if (message->Get("method") == "cluster::SetEnablePassiveChecks") {
|
2013-08-27 12:21:41 +02:00
|
|
|
String svc = params->Get("service");
|
|
|
|
|
|
|
|
Service::Ptr service = Service::GetByName(svc);
|
|
|
|
|
|
|
|
if (!service)
|
|
|
|
return;
|
|
|
|
|
2013-08-28 11:12:20 +02:00
|
|
|
bool enabled = params->Get("enabled");
|
2013-08-27 15:57:00 +02:00
|
|
|
|
2013-08-28 11:12:20 +02:00
|
|
|
service->SetEnablePassiveChecks(enabled, sender->GetName());
|
2013-08-29 13:06:36 +02:00
|
|
|
} else if (message->Get("method") == "cluster::SetEnableNotifications") {
|
|
|
|
String svc = params->Get("service");
|
|
|
|
|
|
|
|
Service::Ptr service = Service::GetByName(svc);
|
|
|
|
|
|
|
|
if (!service)
|
|
|
|
return;
|
|
|
|
|
|
|
|
bool enabled = params->Get("enabled");
|
|
|
|
|
|
|
|
service->SetEnableNotifications(enabled, sender->GetName());
|
|
|
|
} else if (message->Get("method") == "cluster::SetEnableFlapping") {
|
|
|
|
String svc = params->Get("service");
|
|
|
|
|
|
|
|
Service::Ptr service = Service::GetByName(svc);
|
|
|
|
|
|
|
|
if (!service)
|
|
|
|
return;
|
|
|
|
|
|
|
|
bool enabled = params->Get("enabled");
|
|
|
|
|
|
|
|
service->SetEnableFlapping(enabled, sender->GetName());
|
2013-08-28 14:59:41 +02:00
|
|
|
} else if (message->Get("method") == "cluster::SetNextNotification") {
|
|
|
|
String nfc = params->Get("notification");
|
|
|
|
|
|
|
|
Notification::Ptr notification = Notification::GetByName(nfc);
|
|
|
|
|
|
|
|
if (!notification)
|
|
|
|
return;
|
|
|
|
|
|
|
|
bool nextNotification = params->Get("next_notification");
|
|
|
|
|
|
|
|
notification->SetNextNotification(nextNotification, sender->GetName());
|
|
|
|
} else if (message->Get("method") == "cluster::AddComment") {
|
|
|
|
String svc = params->Get("service");
|
|
|
|
|
|
|
|
Service::Ptr service = Service::GetByName(svc);
|
|
|
|
|
|
|
|
if (!service)
|
|
|
|
return;
|
|
|
|
|
|
|
|
Dictionary::Ptr comment = params->Get("comment");
|
|
|
|
|
|
|
|
long type = static_cast<long>(comment->Get("entry_type"));
|
|
|
|
service->AddComment(static_cast<CommentType>(type), comment->Get("author"),
|
|
|
|
comment->Get("text"), comment->Get("expire_time"), comment->Get("id"), sender->GetName());
|
|
|
|
} else if (message->Get("method") == "cluster::RemoveComment") {
|
|
|
|
String svc = params->Get("service");
|
|
|
|
|
|
|
|
Service::Ptr service = Service::GetByName(svc);
|
|
|
|
|
|
|
|
if (!service)
|
|
|
|
return;
|
|
|
|
|
|
|
|
String id = params->Get("id");
|
|
|
|
|
|
|
|
service->RemoveComment(id, sender->GetName());
|
2013-08-28 16:08:22 +02:00
|
|
|
} else if (message->Get("method") == "cluster::AddDowntime") {
|
|
|
|
String svc = params->Get("service");
|
|
|
|
|
|
|
|
Service::Ptr service = Service::GetByName(svc);
|
|
|
|
|
|
|
|
if (!service)
|
|
|
|
return;
|
|
|
|
|
|
|
|
Dictionary::Ptr downtime = params->Get("downtime");
|
|
|
|
|
|
|
|
service->AddDowntime(downtime->Get("comment_id"),
|
|
|
|
downtime->Get("start_time"), downtime->Get("end_time"),
|
|
|
|
downtime->Get("fixed"), downtime->Get("triggered_by"),
|
|
|
|
downtime->Get("duration"), downtime->Get("id"), sender->GetName());
|
|
|
|
} else if (message->Get("method") == "cluster::RemoveDowntime") {
|
|
|
|
String svc = params->Get("service");
|
|
|
|
|
|
|
|
Service::Ptr service = Service::GetByName(svc);
|
|
|
|
|
|
|
|
if (!service)
|
|
|
|
return;
|
|
|
|
|
|
|
|
String id = params->Get("id");
|
|
|
|
|
|
|
|
service->RemoveDowntime(id, sender->GetName());
|
2013-08-29 13:48:18 +02:00
|
|
|
} else if (message->Get("method") == "cluster::SetAcknowledgement") {
|
|
|
|
String svc = params->Get("service");
|
|
|
|
|
|
|
|
Service::Ptr service = Service::GetByName(svc);
|
|
|
|
|
|
|
|
if (!service)
|
|
|
|
return;
|
|
|
|
|
|
|
|
String author = params->Get("author");
|
|
|
|
String comment = params->Get("comment");
|
|
|
|
int type = params->Get("type");
|
|
|
|
double expiry = params->Get("expiry");
|
|
|
|
|
|
|
|
service->AcknowledgeProblem(author, comment, static_cast<AcknowledgementType>(type), expiry, sender->GetName());
|
|
|
|
} else if (message->Get("method") == "cluster::ClearAcknowledgement") {
|
|
|
|
String svc = params->Get("service");
|
|
|
|
|
|
|
|
Service::Ptr service = Service::GetByName(svc);
|
|
|
|
|
|
|
|
if (!service)
|
|
|
|
return;
|
|
|
|
|
|
|
|
ObjectLock olock(service);
|
|
|
|
service->ClearAcknowledgement(sender->GetName());
|
2013-08-27 12:21:41 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
void ClusterComponent::InternalSerialize(const Dictionary::Ptr& bag, int attributeTypes) const
|
|
|
|
{
|
|
|
|
DynamicObject::InternalSerialize(bag, attributeTypes);
|
|
|
|
|
|
|
|
if (attributeTypes & Attribute_Config) {
|
|
|
|
bag->Set("cert_path", m_CertPath);
|
|
|
|
bag->Set("ca_path", m_CAPath);
|
|
|
|
bag->Set("bind_host", m_BindHost);
|
|
|
|
bag->Set("bind_port", m_BindPort);
|
|
|
|
bag->Set("peers", m_Peers);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
void ClusterComponent::InternalDeserialize(const Dictionary::Ptr& bag, int attributeTypes)
|
|
|
|
{
|
|
|
|
DynamicObject::InternalDeserialize(bag, attributeTypes);
|
|
|
|
|
|
|
|
if (attributeTypes & Attribute_Config) {
|
|
|
|
m_CertPath = bag->Get("cert_path");
|
|
|
|
m_CAPath = bag->Get("ca_path");
|
|
|
|
m_BindHost = bag->Get("bind_host");
|
|
|
|
m_BindPort = bag->Get("bind_port");
|
|
|
|
m_Peers = bag->Get("peers");
|
|
|
|
}
|
|
|
|
}
|