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);