view mod_throttle_presence/mod_throttle_presence.lua @ 1408:ed2a9c00b5c4

mod_smacks: Impose a limit on number of unacked stanzas (Thanks Holger)
author Kim Alvefur <zash@zash.se>
date Wed, 07 May 2014 16:18:05 +0200
parents fe9c4daee076
children d8c50448d0e7
line wrap: on
line source

local add_filter = require "util.filters".add_filter;
local add_task = require "util.timer".add_task;

local buffer_seconds = module:get_option_number("flush_presence_seconds");

local function throttle_session(data)
	local session = data.session;
	local buffer, flushing = {}, false;
	local timer_active = false;
	local function flush_buffer()
		module:log("debug", "Flushing buffer for %s", session.full_jid);
		flushing = true;
		for jid, presence in pairs(buffer) do
			session.send(presence);
		end
		flushing = false;
	end
	local function throttle_presence(stanza)
		if stanza.name ~= "presence" or (stanza.attr.type and stanza.attr.type ~= "unavailable") then
			module:log("debug", "Non-presence stanza for %s: %s", session.full_jid, tostring(stanza));
			flush_buffer();
		elseif not flushing then
			module:log("debug", "Buffering presence stanza from %s to %s", stanza.attr.from, session.full_jid);
			buffer[stanza.attr.from] = stanza;
			if not timer_active and buffer_seconds then
				timer_active = true;
				add_task(buffer_seconds, flush_buffer);
			end
			return nil; -- Drop this stanza (we've stored it for later)
		end
		return stanza;
	end
	add_filter(session, "stanzas/out", throttle_presence);
end


module:hook("resource-bind", throttle_session);