Mercurial > prosody-modules
view mod_measure_stanza_counts/mod_measure_stanza_counts.lua @ 3838:9f0c10bd4633
mod_http_upload: Recommend against adding to modules_enabled
Users were confused. Client compatibility is mixed (though I think
most/all are fixed now).
author | Matthew Wild <mwild1@gmail.com> |
---|---|
date | Wed, 08 Jan 2020 10:55:43 +0000 |
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);