Software / code / prosody
Annotate
util/pubsub.lua @ 1519:2c9a650ff1b7
util.pubsub: Fix undefined global accesses
| author | Matthew Wild <mwild1@gmail.com> |
|---|---|
| date | Fri, 10 Jul 2009 02:28:24 +0100 |
| parent | 849:5049b4512df0 |
| child | 1522:569d58d21612 |
| rev | line source |
|---|---|
| 849 | 1 |
| 2 local ipairs, pairs, setmetatable, type = | |
| 3 ipairs, pairs, setmetatable, type; | |
| 4 | |
| 5 module "pubsub" | |
| 6 | |
| 7 local pubsub_node_mt = { __index = _M }; | |
| 8 | |
| 9 function new_node(name) | |
| 10 return setmetatable({ name = name, subscribers = {} }, pubsub_node_mt); | |
| 11 end | |
| 12 | |
| 13 function set_subscribers(node, subscribers_list, list_type) | |
| 14 local subscribers = node.subscribers; | |
| 15 | |
| 16 if list_type == "array" then | |
| 17 for _, jid in ipairs(subscribers_list) do | |
| 18 if not subscribers[jid] then | |
| 19 node:add_subscriber(jid); | |
| 20 end | |
| 21 end | |
| 22 elseif (not list_type) or list_type == "set" then | |
| 23 for jid in pairs(subscribers_list) do | |
| 24 if type(jid) == "string" then | |
| 25 node:add_subscriber(jid); | |
| 26 end | |
| 27 end | |
| 28 end | |
| 29 end | |
| 30 | |
| 31 function get_subscribers(node) | |
| 32 return node.subscribers; | |
| 33 end | |
| 34 | |
| 35 function publish(node, item, dispatcher, data) | |
| 36 local subscribers = node.subscribers; | |
| 37 for i = 1,#subscribers do | |
| 38 item.attr.to = subscribers[i]; | |
| 39 dispatcher(data, item); | |
| 40 end | |
| 41 end | |
| 42 | |
| 43 function add_subscriber(node, jid) | |
| 44 local subscribers = node.subscribers; | |
| 45 if not subscribers[jid] then | |
| 46 local space = #subscribers; | |
| 47 subscribers[space] = jid; | |
| 48 subscribers[jid] = space; | |
| 49 end | |
| 50 end | |
| 51 | |
|
1519
2c9a650ff1b7
util.pubsub: Fix undefined global accesses
Matthew Wild <mwild1@gmail.com>
parents:
849
diff
changeset
|
52 function remove_subscriber(node, jid) |
| 849 | 53 local subscribers = node.subscribers; |
| 54 if subscribers[jid] then | |
| 55 subscribers[subscribers[jid]] = nil; | |
| 56 subscribers[jid] = nil; | |
| 57 end | |
| 58 end | |
| 59 | |
| 60 return _M; |
