Mercurial > prosody-modules
diff mod_cleanup_http/mod_cleanup_http.lua @ 610:50dc20e96a78
mod_cleanup_http: "spring cleans" module, aka unload global http modules without hassle (open ports cleanup doesn't work with server_event)
author | Marco Cirillo <maranda@lightwitch.org> |
---|---|
date | Sun, 12 Feb 2012 18:04:51 +0000 |
parents | |
children | d87a9e1e6d30 |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/mod_cleanup_http/mod_cleanup_http.lua Sun Feb 12 18:04:51 2012 +0000 @@ -0,0 +1,39 @@ +-- Auto-cleanup BOSH stuff module is added globally. + +module:set_global() + +local http_modules = module:get_option("cleanup_http_modules", {}) +if type(http_modules) ~= "table" then module:log("error", "cleanup_http_modules needs to be a module.") ; return false end + +local function cleanup(data) + if data.module == "cleanup_http" then -- it's us getting unloaded destroy handler. + prosody.events.remove_handler("module-unloaded", cleanup) + elseif http_modules[data.module] then + local ports = http_modules[data.module] + + module:log("debug", "Cleaning up http handlers and ports as module %s is being unloaded.", data.module) + for _, options in ipairs(ports) do + if options.port then + httpserver.new.http_servers[options.port].handlers[options.path or "register_account"] = nil + end + end + + -- if there are no handlers left clean and close the socket, doesn't work with server_event + local event = require "core.configmanager".get("*", "core", "use_libevent") + + if not event then + for _, options in ipairs(ports) do + if options.port and not next(httpserver.new.http_servers[options.port].handlers) then + httpserver.new.http_servers[options.port] = nil + if options.interface then + for _, value in ipairs(options.interface) do + if server.getserver(value, options.port) then server.removeserver(value, options.port) end + end + else if server.getserver("*", options.port) then server.removeserver("*", options.port) end end + end + end + end + end +end + +prosody.events.add_handler("module-unloaded", cleanup)