view mod_statistics/mod_statistics.lua @ 3532:85c357b69eec

mod_csi_muc_priorities: Reduce importance of group chat messages This helps if you are in more noisy public channels than small private group chats. The short term plan is to give users the ability to set MUC JIDs as either high or low priority and use that. Long term it would be great to be able to automatically classify MUCs as public channels vs private group chats.
author Kim Alvefur <zash@zash.se>
date Mon, 01 Apr 2019 00:15:13 +0200
parents 66bda434d476
children
line wrap: on
line source

module:set_global();

local stats = module:require("mod_statistics/stats");
local filters = require "util.filters";
local serialize = require "util.serialization".serialize;

local cached_values = {};

local sessions = {};

local function push_stat(conn, name, value)
	local value_str = serialize(value);
	return conn:write((("STAT %q (%s)\n"):format(name, value_str):gsub("\\\n", "\\n")));
end

local function push_stat_to_all(name, value)
	for conn in pairs(sessions) do
		push_stat(conn, name, value);
	end
end

local session_stats_tpl = ([[{
	message_in = %d, message_out = %d;
	presence_in = %d, presence_out = %d;
	iq_in = %d, iq_out = %d;
	bytes_in = %d, bytes_out = %d;
}]]):gsub("%s", "");


local jid_fields = {
	c2s = "full_jid";
	s2sin = "from_host";
	s2sout = "to_host";
	component = "host";
};

local function push_session_to_all(session, stats)
	local id = tostring(session):match("[a-f0-9]+$"); -- FIXME: Better id? :/
	local stanzas_in, stanzas_out = stats.stanzas_in, stats.stanzas_out;
	local s = (session_stats_tpl):format(
		stanzas_in.message, stanzas_out.message,
		stanzas_in.presence, stanzas_out.presence,
		stanzas_in.iq, stanzas_out.iq,
		stats.bytes_in, stats.bytes_out);
	local jid = session[jid_fields[session.type]] or "";
	for conn in pairs(sessions) do
		conn:write(("SESS %q %q %s\n"):format(id, jid, s));
	end
end

local available_stats = stats.stats;
local active_sessions = stats.active_sessions;

-- Handle statistics provided by other modules
local function item_handlers(host)
	host = host and (host.."/") or "";

	return function (event) -- Added
		local stats = event.item.statistics;
		local group = host..(stats.name and (stats.name.."::") or "");
		for name, stat in pairs(stats) do
			available_stats[group..name] = stat;
		end
	end, function (event) -- Removed
		local stats = event.item.statistics;
		local group = host..(stats.name and (stats.name.."::") or "");
		for name, stat in pairs(stats) do
			available_stats[group..name] = nil;
		end
	end;
end

module:handle_items("statistics-provider", item_handlers());
function module.add_host(module)
	module:handle_items("statistics-provider", item_handlers(module.host));
end

-- Network listener
local listener = {};

function listener.onconnect(conn)
	sessions[conn] = {};
	push_stat(conn, "version", prosody.version);
	for name, value in pairs(cached_values) do
		push_stat(conn, name, value);
	end
	conn:write("\n"); -- Signal end of first batch (for non-streaming clients)
end

function listener.onincoming(conn, data)
end

function listener.ondisconnect(conn)
	sessions[conn] = nil;
end

function listener.onreadtimeout()
	return true;
end

function module.load()
	if not(prosody and prosody.arg) then
		return;
	end
	filters.add_filter_hook(stats.filter_hook);

	module:add_timer(1, function ()
		for stat_name, stat in pairs(available_stats) do
			if stat.get then
				local cached = cached_values[stat_name];
				local new_value = stat.get();
				if new_value ~= cached then
					push_stat_to_all(stat_name, new_value);
					cached_values[stat_name] = new_value;
				end
			end
		end
		for session, session_stats in pairs(active_sessions) do
			active_sessions[session] = nil;
			push_session_to_all(session, session_stats);
		end
		return 1;
	end);

end
function module.unload()
	filters.remove_filter_hook(stats.filter_hook);
end
function module.command( args )
	local command = args[1];
	if command == "top" then
		local dir = module:get_directory();
		package.path = dir.."/?.lua;"..dir.."/?.lib.lua;"..package.path;
		local prosodytop = module:require "prosodytop";
		prosodytop.run();
	end
end

if prosody and prosody.arg then
	module:provides("net", {
		default_port = 5782;
		listener = listener;
		private = true;
	});
end