Mercurial > prosody-modules
view mod_measure_stanza_counts/mod_measure_stanza_counts.lua @ 4542:fb4a50bf60f1
mod_prometheus: Invoke stats collection if in 'manual' mode
Since 10d13e0554f9 a special value for statistics_interval "manual"
exists, where a module is expected to invoke processing in connection to
collection of stats. This makes internal collection and exporting to
Prometheus happens at the same time with no chance of timers getting out
of sync.
author | Kim Alvefur <zash@zash.se> |
---|---|
date | Tue, 13 Apr 2021 23:53:53 +0200 |
parents | 512405077709 |
children | 33b1b6ff23d8 |
line wrap: on
line source
module:set_global() local filters = require"util.filters"; local stanza_kinds = { message = true, presence = true, iq = true }; local function rate(measures, dir) return function (stanza, session) measures[dir](); measures[dir .. "_" .. session.type](); if stanza.attr and not stanza.attr.xmlns and stanza_kinds[stanza.name] then measures[dir .. "_" .. session.type .. "_" .. stanza.name](); end return stanza; end end local measures = setmetatable({}, { __index = function (t, name) local m = module:measure(name, "rate"); t[name] = m; return m; end }); local function measure_stanza_counts(session) filters.add_filter(session, "stanzas/in", rate(measures, "incoming")); filters.add_filter(session, "stanzas/out", rate(measures, "outgoing")); end filters.add_filter_hook(measure_stanza_counts);