icinga2/components/demo/democomponent.cpp

#include "i2-demo.h"

using namespace icinga;

IcingaApplication::Ptr DemoComponent::GetIcingaApplication(void)
{
	return static_pointer_cast<IcingaApplication>(GetApplication());
}

string DemoComponent::GetName(void) const
{
	return "democomponent";
}

void DemoComponent::Start(void)
{
	m_DemoEndpoint = make_shared<VirtualEndpoint>();
	m_DemoEndpoint->RegisterMethodHandler("demo::HelloWorld",
	    bind_weak(&DemoComponent::HelloWorldRequestHAndler, shared_from_this()));
	m_DemoEndpoint->RegisterMethodSource("demo::HelloWorld");

	EndpointManager::Ptr endpointManager = GetIcingaApplication()->GetEndpointManager();
	endpointManager->RegisterEndpoint(m_DemoEndpoint);

	endpointManager->OnNewEndpoint += bind_weak(&DemoComponent::NewEndpointHandler, shared_from_this());
	endpointManager->ForeachEndpoint(bind(&DemoComponent::NewEndpointHandler, this, _1));

	m_DemoTimer = make_shared<Timer>();
	m_DemoTimer->SetInterval(1);
	m_DemoTimer->OnTimerExpired += bind_weak(&DemoComponent::DemoTimerHandler, shared_from_this());
	m_DemoTimer->Start();
}

void DemoComponent::Stop(void)
{
	IcingaApplication::Ptr app = GetIcingaApplication();

	if (app) {
		EndpointManager::Ptr endpointManager = app->GetEndpointManager();
		endpointManager->UnregisterEndpoint(m_DemoEndpoint);
	}
}

int DemoComponent::NewEndpointHandler(const NewEndpointEventArgs& neea)
{
	/* Allow sending/receiving demo messages without authentication */
	neea.Endpoint->AddAllowedMethodSinkPrefix("demo::");
	neea.Endpoint->AddAllowedMethodSourcePrefix("demo::");

	return 0;
}

int DemoComponent::DemoTimerHandler(const TimerEventArgs& tea)
{
	cout << "Sending multicast 'hello world' message." << endl;

	JsonRpcRequest request;
	request.SetMethod("demo::HelloWorld");

	EndpointManager::Ptr endpointManager = GetIcingaApplication()->GetEndpointManager();
	endpointManager->SendMulticastRequest(m_DemoEndpoint, request);

	return 0;
}

int DemoComponent::HelloWorldRequestHAndler(const NewRequestEventArgs& nrea)
{
	cout << "Got Hello World from " << nrea.Sender->GetAddress() << endl;

	return 0;
}

EXPORT_COMPONENT(DemoComponent);