Diff

mod_streamstats/mod_streamstats.lua @ 259:d137515e0701

mod_streamstats: New module that streams live stats from Prosody over a TCP stream
author Matthew Wild <mwild1@gmail.com>
date Thu, 07 Oct 2010 16:20:28 +0100
child 639:d4f3754c4286
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/mod_streamstats/mod_streamstats.lua	Thu Oct 07 16:20:28 2010 +0100
@@ -0,0 +1,96 @@
+local stats = prosody.stats;
+
+if not stats then
+	stats = {
+		stats = {}; conns = {};
+		
+		broadcast = function (self, stat)
+			local value = self.stats[stat];
+			for conn in pairs(self.conns) do
+				conn:write(stat..":"..value.."\n");
+			end
+		end;
+
+		adjust = function (self, stat, delta)
+			if delta == 0 then return; end
+			self.stats[stat] = (self.stats[stat] or 0) + delta;
+			self:broadcast(stat);
+		end;
+
+		set = function (self, stat, value)
+			if value == self.stats[stat] then return; end
+			self.stats[stat] = value;
+			self:broadcast(stat);
+		end;
+		
+		add_conn = function (self, conn)
+			self.conns[conn] = true;
+			for stat, value in pairs(self.stats) do
+				conn:write(stat..":"..value.."\n");
+			end
+		end;
+		
+		remove_conn = function (self, conn)
+			self.conns[conn] = nil;
+		end;
+	};
+	prosody.stats = stats;
+	
+	local network = {};
+	
+	function network.onconnect(conn)
+		stats:add_conn(conn);
+	end
+	
+	function network.onincoming(conn, data)
+	end
+	
+	function network.ondisconnect(conn, reason)
+		stats:remove_conn(conn);
+	end
+	
+	require "util.iterators";
+	require "util.timer".add_task(1, function ()
+		stats:set("s2s-in", count(keys(prosody.incoming_s2s)));
+		return math.random(10, 20);
+	end);
+	require "util.timer".add_task(3, function ()
+		local s2sout_count = 0;
+		for _, host in pairs(prosody.hosts) do
+			s2sout_count = s2sout_count + count(keys(host.s2sout));
+		end
+		stats:set("s2s-out", s2sout_count);
+		return math.random(10, 20);
+	end);
+	
+	require "net.connlisteners".register("stats", network);
+	require "net.connlisteners".start("stats", { port = module:get_option("stats_ports") or 5444, interface = "127.0.0.1" });
+end
+
+module:hook("resource-bind", function ()
+	stats:adjust("c2s", 1);
+end);
+module:hook("resource-unbind", function ()
+	stats:adjust("c2s", -1);
+end);
+
+local c2s_count = 0;
+for username, user in pairs(hosts[module.host].sessions or {}) do
+	for resource, session in pairs(user.sessions or {}) do
+		c2s_count = c2s_count + 1;
+	end
+end
+stats:adjust("c2s", c2s_count);
+
+module:hook("s2sin-established", function (event)
+	stats:adjust("s2s-in", 1);
+end);
+module:hook("s2sin-destroyed", function (event)
+	stats:adjust("s2s-in", -1);
+end);
+module:hook("s2sout-established", function (event)
+	stats:adjust("s2s-out", 1);
+end);
+module:hook("s2sout-destroyed", function (event)
+	stats:adjust("s2s-out", -1);
+end);