X-Git-Url: https://git.enpas.org/?a=blobdiff_plain;f=util%2Fevents.lua;h=3816f30b0d13cd660f609288835c6463feab4074;hb=a80f2d734aaf6cf54728a7d9e4a68335f91af8f9;hp=54816271bebf73d49885b26d8b5f3aedc686eadb;hpb=255e4034aa95afe6217d4224a267a703437b8ff1;p=prosody.git diff --git a/util/events.lua b/util/events.lua index 54816271..3816f30b 100644 --- a/util/events.lua +++ b/util/events.lua @@ -11,17 +11,16 @@ function new() local dispatchers = {}; local handlers = {}; local event_map = {}; - local function _rebuild_index() -- TODO optimize index rebuilding - for event, _handlers in pairs(event_map) do - local index = handlers[event]; - if index then - for i=#index,1,-1 do index[i] = nil; end - else index = {}; handlers[event] = index; end - for handler in pairs(_handlers) do - t_insert(index, handler); - end - t_sort(index, function(a, b) return _handlers[a] > _handlers[b]; end); + local function _rebuild_index(event) -- TODO optimize index rebuilding + local _handlers = event_map[event]; + local index = handlers[event]; + if index then + for i=#index,1,-1 do index[i] = nil; end + else index = {}; handlers[event] = index; end + for handler in pairs(_handlers) do + t_insert(index, handler); end + t_sort(index, function(a, b) return _handlers[a] > _handlers[b]; end); end; local function add_handler(event, handler, priority) local map = event_map[event]; @@ -31,13 +30,13 @@ function new() map = {[handler] = priority or 0}; event_map[event] = map; end - _rebuild_index(); + _rebuild_index(event); end; local function remove_handler(event, handler) local map = event_map[event]; if map then map[handler] = nil; - _rebuild_index(); + _rebuild_index(event); end end; local function add_plugin(plugin) @@ -53,9 +52,9 @@ function new() local function _create_dispatcher(event) -- FIXME duplicate code in fire_event local h = handlers[event]; if not h then h = {}; handlers[event] = h; end - local dispatcher = function(data) + local dispatcher = function(...) for _, handler in ipairs(h) do - local ret = handler(data); + local ret = handler(...); if ret ~= nil then return ret; end end end; @@ -65,11 +64,11 @@ function new() local function get_dispatcher(event) return dispatchers[event] or _create_dispatcher(event); end; - local function fire_event(event, data) -- FIXME duplicates dispatcher code + local function fire_event(event, ...) -- FIXME duplicates dispatcher code local h = handlers[event]; if h then for _, handler in ipairs(h) do - local ret = handler(data); + local ret = handler(...); if ret ~= nil then return ret; end end end