Mercurial > prosody-modules
changeset 1110:97e238ce37ce
mod_s2s_keepalive: Initial commit, poke s2s connections with pings and whitespace
author | Kim Alvefur <zash@zash.se> |
---|---|
date | Fri, 12 Jul 2013 01:28:48 +0200 |
parents | d988f2db9773 |
children | bdbf76730f49 |
files | mod_s2s_keepalive/mod_s2s_keepalive.lua |
diffstat | 1 files changed, 42 insertions(+), 0 deletions(-) [+] |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/mod_s2s_keepalive/mod_s2s_keepalive.lua Fri Jul 12 01:28:48 2013 +0200 @@ -0,0 +1,42 @@ +local st = require "util.stanza"; + +local keepalive_servers = module:get_option_set("keepalive_servers"); +local keepalive_interval = module:get_option_number("keepalive_interval", 60); + +local host = module.host; + +local function send_pings() + for remote_domain, session in pairs(hosts[host].s2sout) do + if session.type == "s2sout" -- as opposed to _unauthed + and (not(keepalive_servers) or keepalive_servers:contains(remote_domain)) then + session.sends2s(st.iq({ to = remote_domain, type = "get", from = host, id = "keepalive" }) + :tag("ping", { xmlns = "urn:xmpp:ping" }) + ); + -- Note: We don't actually check if this comes back. + end + end + + for session in pairs(prosody.incoming_s2s) do + if session.type == "s2sin" -- as opposed to _unauthed + and (not(keepalive_servers) or keepalive_servers:contains(session.from_host)) then + session.conn:send " "; + -- If the connection is dead, this should make it time out. + end + end + return keepalive_interval; +end + +if module.add_timer then -- 0.9 + module:add_timer(keepalive_interval, send_pings); +else -- 0.8 + local timer = require "util.timer"; + local unloaded; + timer.add_task(keepalive_interval, function() + if not unloaded then + return send_pings() + end + end); + function module.unload() + unloaded = true + end +end