util/pubsub.lua

   1 module("pubsub", package.seeall);
   2
   3 local service = {};
   4 local service_mt = { __index = service };
   5
   6 function new(cb)
   7         return setmetatable({ cb = cb or {}, nodes = {} }, service_mt);
   8 end
   9
  10 function service:add_subscription(node, actor, jid)
  11         local node_obj = self.nodes[node];
  12         if not node_obj then
  13                 return false, "item-not-found";
  14         end
  15         node_obj.subscribers[jid] = true;
  16         return true;
  17 end
  18
  19 function service:remove_subscription(node, actor, jid)
  20         self.nodes[node].subscribers[jid] = nil;
  21         return true;
  22 end
  23
  24 function service:get_subscription(node, actor, jid)
  25         local node_obj = self.nodes[node];
  26         if node_obj then
  27                 return node_obj.subscribers[jid];
  28         end
  29 end
  30
  31 function service:publish(node, actor, id, item)
  32         local node_obj = self.nodes[node];
  33         if not node_obj then
  34                 node_obj = { name = node, subscribers = {}, config = {}, data = {} };
  35                 self.nodes[node] = node_obj;
  36         end
  37         node_obj.data[id] = item;
  38         self.cb.broadcaster(node, node_obj.subscribers, item);
  39         return true;
  40 end
  41
  42 function service:get(node, actor, id)
  43         local node_obj = self.nodes[node];
  44         if node_obj then
  45                 if id then
  46                         return { node_obj.data[id] };
  47                 else
  48                         return node_obj.data;
  49                 end
  50         end
  51 end
  52
  53 return _M;