Mercurial > prosody-modules
view mod_measure_storage/mod_measure_storage.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 | 04ae5b45e6c7 |
children |
line wrap: on
line source
module:set_global() local function return_args_after_calling(f, ...) f(); return ... end local function time_method(module, store_name, store_type, method_name, method_function) local opt_use_tags = module:get_option_boolean("measure_storage_tagged_metric", false); local metric_name, metric_tags; if opt_use_tags then metric_name, metric_tags = "storage_operation", ("store_name:%s,store_type:%s,store_operation:%s"):format(store_name, store_type, method_name); else metric_name = store_name.."_"..store_type.."_"..method_name; end local measure_operation_started = module:measure(metric_name, "times", metric_tags); return function (...) module:log("debug", "Measuring storage operation %s (%s)", metric_name, metric_tags or "no tags"); local measure_operation_complete = measure_operation_started(); return return_args_after_calling(measure_operation_complete, method_function(...)); end; end local function wrap_store(module, store_name, store_type, store) local new_store = setmetatable({}, { __index = function (t, method_name) local original_method = store[method_name]; if type(original_method) ~= "function" then if original_method then rawset(t, method_name, original_method); end return original_method; end local timed_method = time_method(module, store_name, store_type, method_name, original_method); rawset(t, method_name, timed_method); return timed_method; end; }); return new_store; end local function hook_event(module) module:hook("store-opened", function(event) event.store = wrap_store(module, event.store_name, event.store_type or "keyval", event.store); end); end function module.load() hook_event(module); end function module.add_host(module) hook_event(module); end