+
−module:set_global();
+
−local stats = module:shared"stats";
+
−local iter = require "util.iterators";
+
−local count, keys = iter.count, iter.keys;
+
−
+
−stats.stats = stats.stats or {};
+
−stats.conns = stats.conns or {};
+
−
+
−setmetatable(stats, {
+
−	__index = {
+
−
+
−		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;
+
−	};
+
−});
+
−
+
−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
+
−
+
−module:add_timer(1, function ()
+
−	stats:set("s2s-in", count(keys(prosody.incoming_s2s)));
+
−	return math.random(10, 20);
+
−end);
+
−module:add_timer(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);
+
−
+
−
+
−function module.add_host(module)
+
−	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:set("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);
+
−end
+
−
+
−module:provides("net", {
+
−	default_port = 5444;
+
−	listener = network;
+
−});