Mercurial > prosody-modules
comparison mod_s2s_keepalive/mod_s2s_keepalive.lua @ 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 | |
children | 2db2c03dfb95 |
comparison
equal
deleted
inserted
replaced
1109:d988f2db9773 | 1110:97e238ce37ce |
---|---|
1 local st = require "util.stanza"; | |
2 | |
3 local keepalive_servers = module:get_option_set("keepalive_servers"); | |
4 local keepalive_interval = module:get_option_number("keepalive_interval", 60); | |
5 | |
6 local host = module.host; | |
7 | |
8 local function send_pings() | |
9 for remote_domain, session in pairs(hosts[host].s2sout) do | |
10 if session.type == "s2sout" -- as opposed to _unauthed | |
11 and (not(keepalive_servers) or keepalive_servers:contains(remote_domain)) then | |
12 session.sends2s(st.iq({ to = remote_domain, type = "get", from = host, id = "keepalive" }) | |
13 :tag("ping", { xmlns = "urn:xmpp:ping" }) | |
14 ); | |
15 -- Note: We don't actually check if this comes back. | |
16 end | |
17 end | |
18 | |
19 for session in pairs(prosody.incoming_s2s) do | |
20 if session.type == "s2sin" -- as opposed to _unauthed | |
21 and (not(keepalive_servers) or keepalive_servers:contains(session.from_host)) then | |
22 session.conn:send " "; | |
23 -- If the connection is dead, this should make it time out. | |
24 end | |
25 end | |
26 return keepalive_interval; | |
27 end | |
28 | |
29 if module.add_timer then -- 0.9 | |
30 module:add_timer(keepalive_interval, send_pings); | |
31 else -- 0.8 | |
32 local timer = require "util.timer"; | |
33 local unloaded; | |
34 timer.add_task(keepalive_interval, function() | |
35 if not unloaded then | |
36 return send_pings() | |
37 end | |
38 end); | |
39 function module.unload() | |
40 unloaded = true | |
41 end | |
42 end |