view mod_measure_stanza_counts/mod_measure_stanza_counts.lua @ 5473:e4382f6e3564

mod_http_oauth2: Tweak formatting of log message No need to `or ""` anymore since Prosody rev e88db5668cfb (0.11.0) and the %q format should produce either (nil) or "http://example"
author Kim Alvefur <zash@zash.se>
date Thu, 18 May 2023 13:27:27 +0200
parents 33b1b6ff23d8
children
line wrap: on
line source

module:set_global()

local filters = require"util.filters";

local stanzas_in = module:metric(
	"counter", "received", "",
	"Stanzas received by Prosody",
	{ "session_type", "stanza_kind" }
)
local stanzas_out = module:metric(
	"counter", "sent", "",
	"Stanzas sent by prosody",
	{ "session_type", "stanza_kind" }
)

local stanza_kinds = { message = true, presence = true, iq = true };

local function rate(metric_family)
	return function (stanza, session)
		if stanza.attr and not stanza.attr.xmlns and stanza_kinds[stanza.name] then
			metric_family:with_labels(session.type, stanza.name):add(1);
		end
		return stanza;
	end
end

local function measure_stanza_counts(session)
	filters.add_filter(session, "stanzas/in",  rate(stanzas_in));
	filters.add_filter(session, "stanzas/out", rate(stanzas_out));
end

filters.add_filter_hook(measure_stanza_counts);