/****************************************************************************** * 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. * ******************************************************************************/ #ifndef ENDPOINTMANAGER_H #define ENDPOINTMANAGER_H namespace icinga { /** * Event arguments for the "new endpoint registered" event. * * @ingroup icinga */ struct I2_ICINGA_API NewEndpointEventArgs : public EventArgs { icinga::Endpoint::Ptr Endpoint; /**< The new endpoint. */ }; struct NewResponseEventArgs; /** * Information about a pending API request. * * @ingroup icinga */ struct I2_ICINGA_API PendingRequest { time_t Timeout; RequestMessage Request; function Callback; bool HasTimedOut(void) const { return time(NULL) > Timeout; } }; /** * Event arguments for the "new response" event. * * @ingroup icinga */ struct I2_ICINGA_API NewResponseEventArgs : public EventArgs { Endpoint::Ptr Sender; RequestMessage Request; ResponseMessage Response; bool TimedOut; }; /** * Forwards messages between endpoints. * * @ingroup icinga */ class I2_ICINGA_API EndpointManager : public Object { public: typedef shared_ptr Ptr; typedef weak_ptr WeakPtr; EndpointManager(void) : m_NextMessageID(0) { } void SetIdentity(string identity); string GetIdentity(void) const; void SetSSLContext(shared_ptr sslContext); shared_ptr GetSSLContext(void) const; void AddListener(string service); void AddConnection(string node, string service); void RegisterEndpoint(Endpoint::Ptr endpoint); void UnregisterEndpoint(Endpoint::Ptr endpoint); void SendUnicastMessage(Endpoint::Ptr sender, Endpoint::Ptr recipient, const MessagePart& message); void SendAnycastMessage(Endpoint::Ptr sender, const RequestMessage& message); void SendMulticastMessage(Endpoint::Ptr sender, const RequestMessage& message); void SendAPIMessage(Endpoint::Ptr sender, RequestMessage& message, function callback, time_t timeout = 10); void ProcessResponseMessage(const Endpoint::Ptr& sender, const ResponseMessage& message); void ForEachEndpoint(function callback); Endpoint::Ptr GetEndpointByIdentity(string identity) const; Observable OnNewEndpoint; private: string m_Identity; shared_ptr m_SSLContext; vector m_Servers; vector m_Endpoints; long m_NextMessageID; map m_Requests; Timer::Ptr m_RequestTimer; void RegisterServer(JsonRpcServer::Ptr server); void UnregisterServer(JsonRpcServer::Ptr server); static bool RequestTimeoutLessComparer(const pair& a, const pair& b); void RescheduleRequestTimer(void); int RequestTimerHandler(const TimerEventArgs& ea); int NewClientHandler(const NewClientEventArgs& ncea); }; } #endif /* ENDPOINTMANAGER_H */