diff options
| -rw-r--r-- | plugins/mod_debug_reset.lua | 41 |
1 files changed, 41 insertions, 0 deletions
diff --git a/plugins/mod_debug_reset.lua b/plugins/mod_debug_reset.lua new file mode 100644 index 00000000..cd08d960 --- /dev/null +++ b/plugins/mod_debug_reset.lua @@ -0,0 +1,41 @@ +-- This module will "reset" the server when the client connection count drops +-- to zero. This is somewhere between a reload and a full process restart. +-- It is useful to ensure isolation between test runs, for example. It may +-- also be of use for some kinds of manual testing. + +module:set_global(); + +local hostmanager = require "core.hostmanager"; +local modulemanager = require "core.modulemanager"; + +local timer = require "util.timer"; + +local function do_reset() + module:log("info", "Performing reset..."); + local hosts = {}; + for host in pairs(prosody.hosts) do + table.insert(hosts, host); + end + module:fire_event("server-resetting"); + for _, host in ipairs(hosts) do + hostmanager.deactivate(host); + timer.add_task(0, function () + hostmanager.activate(host); + module:log("info", "Reset complete"); + module:fire_event("server-reset"); + end); + end +end + +function module.add_host(host_module) + host_module:hook("resource-unbind", function () + if next(prosody.full_sessions) == nil then + timer.add_task(0, do_reset); + end + end); +end + +local console_env = module:shared("/*/admin_shell/env"); +console_env.debug_reset = { + reset = do_reset; +}; |