/****************************************************************************** * 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 { /** * Forwards messages between endpoints. * * @ingroup icinga */ class I2_ICINGA_API EndpointManager : public Object { public: typedef shared_ptr Ptr; typedef weak_ptr WeakPtr; typedef map::iterator Iterator; EndpointManager(void); static EndpointManager::Ptr GetInstance(void); void SetIdentity(const String& identity); String GetIdentity(void) const; void SetSSLContext(const shared_ptr& sslContext); shared_ptr GetSSLContext(void) const; void AddListener(const String& service); void AddConnection(const String& node, const String& service); void RegisterEndpoint(const Endpoint::Ptr& endpoint); void UnregisterEndpoint(const Endpoint::Ptr& endpoint); void SendUnicastMessage(const Endpoint::Ptr& sender, const Endpoint::Ptr& recipient, const MessagePart& message); void SendAnycastMessage(const Endpoint::Ptr& sender, const RequestMessage& message); void SendMulticastMessage(const Endpoint::Ptr& sender, const RequestMessage& message); void SendAPIMessage(const Endpoint::Ptr& sender, const Endpoint::Ptr& recipient, RequestMessage& message, function callback, double timeout = 30); void ProcessResponseMessage(const Endpoint::Ptr& sender, const ResponseMessage& message); void ForEachEndpoint(function callback); Iterator Begin(void); Iterator End(void); Endpoint::Ptr GetEndpointByIdentity(const String& identity) const; boost::signal OnNewEndpoint; private: String m_Identity; shared_ptr m_SSLContext; vector m_Servers; vector m_PendingEndpoints; map m_Endpoints; /** * Information about a pending API request. * * @ingroup icinga */ struct I2_ICINGA_API PendingRequest { double Timeout; RequestMessage Request; function Callback; bool HasTimedOut(void) const { return Utility::GetTime() > Timeout; } }; long m_NextMessageID; map m_Requests; Timer::Ptr m_RequestTimer; void RegisterServer(const JsonRpcServer::Ptr& server); void UnregisterServer(const JsonRpcServer::Ptr& server); static bool RequestTimeoutLessComparer(const pair& a, const pair& b); void RequestTimerHandler(void); void NewClientHandler(const TcpClient::Ptr& client); }; } #endif /* ENDPOINTMANAGER_H */