Mercurial > prosody-modules
view mod_reload_modules/mod_reload_modules.lua @ 5536:96dec7681af8
mod_firewall: Update user marks to store instantly via map store
The original approach was to keep marks in memory only, and persist them at
shutdown. That saves I/O, at the cost of potentially losing marks on an
unclean shutdown.
This change persists marks instantly, which may have some performance overhead
but should be more "correct".
It also splits the marking/unmarking into an event which may be watched or
even fired by other modules.
author | Matthew Wild <mwild1@gmail.com> |
---|---|
date | Thu, 08 Jun 2023 16:20:42 +0100 |
parents | cc14bfec209b |
children |
line wrap: on
line source
local array, it, set = require "util.array", require "util.iterators", require "util.set"; local mm = require "core.modulemanager"; function reload_all() local modules = module:get_option_set("reload_modules", {}); if not modules then module:log("warn", "No modules listed in the config to reload - set reload_modules to a list"); return; end local configured_modules = module:get_option_inherited_set("modules_enabled", {}); local component_module = module:get_option_string("component_module"); if component_module then -- Ensure awareness of the component module so that it is not unloaded configured_modules:add(component_module); end -- ignore removed hosts if not prosody.hosts[module.host] then module:log("warn", "Ignoring host %s: host was removed...", module.host); return; end local loaded_modules = set.new(array.collect(it.keys(prosody.hosts[module.host].modules))); local need_to_load = set.intersection(configured_modules - loaded_modules, modules); local need_to_unload = set.intersection(loaded_modules - configured_modules, modules); for module_name in need_to_load do module:log("debug", "Loading %s", module_name); mm.load(module.host, module_name); end for module_name in need_to_unload do module:log("debug", "Unloading %s", module_name); mm.unload(module.host, module_name); end modules:exclude(need_to_load+need_to_unload) for module_name in set.intersection(modules,configured_modules) do module:log("debug", "Reloading %s", module_name); mm.reload(module.host, module_name); end local global_modules = module:get_option_set("reload_global_modules", {}); for module_name in global_modules do module:log("debug", "Global reload of mod_%s", module_name); mm.reload("*", module_name); end end if module.hook_global then module:hook_global("config-reloaded", reload_all); else -- COMPAT w/pre-0.9 function module.load() prosody.events.add_handler("config-reloaded", reload_all); end function module.unload() prosody.events.remove_handler("config-reloaded", reload_all); end end