Publish local stats to parents every 10s while connected (event::ClusterStats)

Author: Al2Klimov

lib/remote/CMakeLists.txt

@@ -55,6 +55,7 @@ set(remote_SOURCES
   modifyobjecthandler.cpp modifyobjecthandler.hpp
   objectqueryhandler.cpp objectqueryhandler.hpp
   pkiutility.cpp pkiutility.hpp
+  statsreporter.cpp statsreporter.hpp
   statushandler.cpp statushandler.hpp
   templatequeryhandler.cpp templatequeryhandler.hpp
   typequeryhandler.cpp typequeryhandler.hpp

lib/remote/statsreporter.cpp

@@ -0,0 +1,100 @@
+/******************************************************************************
+ * Icinga 2                                                                   *
+ * Copyright (C) 2012-2018 Icinga Development Team (https://www.icinga.com/)  *
+ *                                                                            *
+ * 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 "base/array.hpp"
+#include "base/dictionary.hpp"
+#include "base/function.hpp"
+#include "base/objectlock.hpp"
+#include "base/scriptglobal.hpp"
+#include "base/utility.hpp"
+#include "remote/endpoint.hpp"
+#include "remote/statsreporter.hpp"
+#include <boost/bind.hpp>
+#include <boost/thread/mutex.hpp>
+
+using namespace icinga;
+
+EndpointCache::EndpointCache(StatsReporter& reporter, const Endpoint::Ptr& endpoint)
+{
+	timer = new Timer;
+	timer->OnTimerExpired.connect(std::bind(&StatsReporter::ReportStats, &reporter, endpoint));
+	timer->SetInterval(10);
+	timer->Start();
+	timer->Reschedule(0);
+}
+
+EndpointCache::~EndpointCache()
+{
+	timer->Stop();
+}
+
+StatsReporter StatsReporter::m_Instance;
+
+StatsReporter::StatsReporter()
+{
+	Endpoint::OnConnected.connect(std::bind(&StatsReporter::OnConnected, this, _1));
+	Endpoint::OnDisconnected.connect(std::bind(&StatsReporter::OnDisconnected, this, _1));
+}
+
+void StatsReporter::OnConnected(const Endpoint::Ptr& endpoint)
+{
+	if (Endpoint::GetLocalEndpoint()->GetZone()->IsChildOf(endpoint->GetZone())) {
+		boost::mutex::scoped_lock lock(m_Mutex);
+
+		m_Cache.emplace(endpoint, EndpointCache(*this, endpoint));
+	}
+}
+
+void StatsReporter::OnDisconnected(const Endpoint::Ptr& endpoint)
+{
+	boost::mutex::scoped_lock lock (m_Mutex);
+
+	m_Cache.erase(endpoint);
+}
+
+void StatsReporter::ReportStats(const Endpoint::Ptr& endpoint)
+{
+	Dictionary::Ptr statsFunctions = ScriptGlobal::Get("StatsFunctions", &Empty);
+
+	if (statsFunctions) {
+		Dictionary::Ptr status = new Dictionary();
+
+		{
+			Array::Ptr perfdata = new Array();
+			ObjectLock olock(statsFunctions);
+
+			for (const Dictionary::Pair &kv : statsFunctions) {
+				((Function::Ptr) kv.second)->Invoke({status, perfdata});
+			}
+		}
+
+		for (auto& client : endpoint->GetClients()) {
+			client->SendMessage(new Dictionary({
+				{ "jsonrpc", "2.0" },
+				{ "method", "event::ClusterStats" },
+				{ "params", new Dictionary({
+					{ "ctime", Utility::GetTime() },
+					{ "stats", status }
+				}) }
+			}));
+
+			break;
+		}
+	}
+}

lib/remote/statsreporter.hpp

@@ -0,0 +1,69 @@
+/******************************************************************************
+ * Icinga 2                                                                   *
+ * Copyright (C) 2012-2018 Icinga Development Team (https://www.icinga.com/)  *
+ *                                                                            *
+ * 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 STATSREPORTER_H
+#define STATSREPORTER_H
+
+#include "base/dictionary.hpp"
+#include "base/timer.hpp"
+#include "remote/endpoint.hpp"
+#include <boost/thread/mutex.hpp>
+#include <map>
+
+namespace icinga
+{
+
+class StatsReporter;
+
+/**
+ * @ingroup remote
+ */
+struct EndpointCache
+{
+	EndpointCache(StatsReporter& reporter, const Endpoint::Ptr& endpoint);
+	~EndpointCache();
+
+	Timer::Ptr timer;
+
+	// TODO: actually cache something useful
+};
+
+/**
+* @ingroup remote
+*/
+class StatsReporter
+{
+	friend EndpointCache;
+
+private:
+	StatsReporter();
+
+	void OnConnected(const Endpoint::Ptr& endpoint);
+	void OnDisconnected(const Endpoint::Ptr& endpoint);
+	void ReportStats(const Endpoint::Ptr& endpoint);
+
+	static StatsReporter m_Instance;
+
+	boost::mutex m_Mutex;
+	std::map<Endpoint::Ptr, EndpointCache> m_Cache;
+};
+
+}
+
+#endif /* STATSREPORTER_H */