From be7107c0ab7c74824c1f52d668efa87f4ff1fbee Mon Sep 17 00:00:00 2001 From: Matthew Wild Date: Tue, 28 Apr 2015 13:52:07 +0100 Subject: moduleapi: Experimental API for modules to measure the rate+duration of events --- core/moduleapi.lua | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) (limited to 'core/moduleapi.lua') diff --git a/core/moduleapi.lua b/core/moduleapi.lua index e32c116a..71e19e01 100644 --- a/core/moduleapi.lua +++ b/core/moduleapi.lua @@ -375,6 +375,25 @@ function api:measure(name, type) return measure(type, "/"..self.host.."/mod_"..self.name.."/"..name); end +function api:measure_object_event(events_object, event_name, stat_name) + local m = self:measure(stat_name or event_name, "duration"); + local function handler(handlers, event_name, event_data) + local finished = m(); + local ret = handlers(event_name, event_data); + finished(); + return ret; + end + return self:hook_object_event(events_object, event_name, handler); +end + +function api:measure_event(event_name, stat_name) + return self:hook_object_event((hosts[self.host] or prosody).events.wrappers, event_name, handler); +end + +function api:measure_global_event(event_name, stat_name) + return self:hook_object_event(prosody.events.wrappers, event_name, handler); +end + function api.init(mm) modulemanager = mm; return api; -- cgit v1.2.3