view mod_host_status_check/mod_host_status_check.lua @ 5381:32a9817c7516

mod_firewall: Initialize compiled chunk just once for all handlers This should fix a case where some stateful dependencies (such as throttles) produce separate instances for every call to new_handler(), leading to surprising behaviour (e.g. rules executed via JUMP CHAIN vs ::deliver would have separate rate limits). This also adds better error handling in case the compiled code fails to run for some reason.
author Matthew Wild <mwild1@gmail.com>
date Fri, 28 Apr 2023 13:27:06 +0100
parents 3d80f8dba886
children
line wrap: on
line source

local time = require "socket".gettime;

local heartbeats = module:shared("/*/host_status_check/heartbeats");
local connection_events = module:shared("/*/host_status_check/connection_events");

if prosody.hosts[module.host].type == "component" and module:get_option_string("component_module") == "component" then
	module:hook("component-authenticated", function ()
		connection_events[module.host] = { connected = true; timestamp = time() };
	end);

	-- Note: this event is not in 0.9, and requires a recent 0.10 or trunk build
	module:hook("component-disconnected", function ()
		connection_events[module.host] = { connected = false; timestamp = time() };
	end);

	module:hook("stanza/xmpp:prosody.im/heartbeat:heartbeat", function ()
		heartbeats[module.host] = time();
		return true;
	end);
else
	connection_events[module.host] = { connected = true, timestamp = time() };
end

function module.unload()
	connection_events[module.host] = { connected = false, timestamp = time() };
	heartbeats[module.host] = nil;
end