Mercurial > prosody-modules
view mod_measure_stanza_counts/mod_measure_stanza_counts.lua @ 5381:32a9817c7516
mod_firewall: Initialize compiled chunk just once for all handlers
This should fix a case where some stateful dependencies (such as throttles)
produce separate instances for every call to new_handler(), leading to
surprising behaviour (e.g. rules executed via JUMP CHAIN vs ::deliver would
have separate rate limits).
This also adds better error handling in case the compiled code fails to run
for some reason.
author | Matthew Wild <mwild1@gmail.com> |
---|---|
date | Fri, 28 Apr 2023 13:27:06 +0100 |
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);