view mod_measure_client_identities/mod_measure_client_identities.lua @ 5807:f08818b8b87d

mod_pubsub_serverinfo: Update documentation Removed a 'feature yet to be implemented' that now has been implemented.
author Guus der Kinderen <guus.der.kinderen@gmail.com>
date Thu, 04 Jan 2024 12:33:34 +0100
parents fdbf7c2aed7b
children
line wrap: on
line source

module:set_global();

local measure = require"core.statsmanager".measure;

local counters = {
	unknown = measure("amount", "client_identities.unknown"),
};

module:hook("stats-update", function ()
	local buckets = {
		unknown = 0,
	};
	for _, session in pairs(prosody.full_sessions) do
		if session.caps_cache ~= nil then
			local node_string = session.caps_cache.attr.node;
			local node = node_string:match("([^#]+)");
			if buckets[node] == nil then
				buckets[node] = 0;
			end
			buckets[node] = buckets[node] + 1;
		else
			buckets.unknown = buckets.unknown + 1;
		end
	end
	local visited = {};
	for bucket, count in pairs(buckets) do
		if counters[bucket] == nil then
			counters[bucket] = measure("amount", "client_identities."..bucket);
		end
		counters[bucket](count);
		visited[bucket] = true;
	end
	for bucket, counter in pairs(counters) do
		if not visited[bucket] then
			counter(0);
		end
	end
end)