prosody-modules
0d3926e49b55
mod_measure_stanza_counts/mod_measure_stanza_counts.lua

Find changesets by keywords by author, files, the commit message, revision number or hash, or revset expression.

Software / code / prosody-modules

File

mod_measure_stanza_counts/mod_measure_stanza_counts.lua @ 3793:0d3926e49b55

mod_pubsub_post: Fix use of 'id' field in JSON object as pubsub item id `data` here is the source string, so there's no `id` field. Also only use the `id` field if it's a string.
author Kim Alvefur <zash@zash.se>
date Wed, 01 Jan 2020 10:11:08 +0100
parent 2788:512405077709
child 4559: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);