Mercurial > prosody-modules
comparison mod_pubsub_stats/mod_pubsub_stats.lua @ 3068:380f92276e57
mod_pubsub_stats: Make the node used configurable
| author | Kim Alvefur <zash@zash.se> |
|---|---|
| date | Tue, 29 May 2018 05:34:05 +0200 |
| parents | b01ef74c9fc0 |
| children | 8a870e0319db |
comparison
equal
deleted
inserted
replaced
| 3067:b01ef74c9fc0 | 3068:380f92276e57 |
|---|---|
| 2 local dt = require "util.datetime"; | 2 local dt = require "util.datetime"; |
| 3 | 3 |
| 4 local pubsub = module:depends"pubsub"; | 4 local pubsub = module:depends"pubsub"; |
| 5 | 5 |
| 6 local actor = module.host .. "/modules/" .. module.name; | 6 local actor = module.host .. "/modules/" .. module.name; |
| 7 | |
| 8 local node = module:get_option_string(module.name .. "_node", "stats"); | |
| 7 | 9 |
| 8 local function publish_stats(stats, stats_extra) | 10 local function publish_stats(stats, stats_extra) |
| 9 local id = "current"; | 11 local id = "current"; |
| 10 local xitem = st.stanza("item", { id = id }) | 12 local xitem = st.stanza("item", { id = id }) |
| 11 :tag("query", { xmlns = "http://jabber.org/protocol/stats" }); | 13 :tag("query", { xmlns = "http://jabber.org/protocol/stats" }); |
| 14 local stat_extra = stats_extra[name]; | 16 local stat_extra = stats_extra[name]; |
| 15 local unit = stat_extra and stat_extra.units; | 17 local unit = stat_extra and stat_extra.units; |
| 16 xitem:tag("stat", { name = name, unit = unit, value = tostring(value) }):up(); | 18 xitem:tag("stat", { name = name, unit = unit, value = tostring(value) }):up(); |
| 17 end | 19 end |
| 18 | 20 |
| 19 local ok, err = pubsub.service:publish("stats", actor, id, xitem); | 21 local ok, err = pubsub.service:publish(node, actor, id, xitem); |
| 20 if not ok then | 22 if not ok then |
| 21 module:log("error", "Error publishing stats: %s", err); | 23 module:log("error", "Error publishing stats: %s", err); |
| 22 end | 24 end |
| 23 end | 25 end |
| 24 | 26 |
| 25 function module.load() | 27 function module.load() |
| 26 pubsub.service:create("stats", true); | 28 pubsub.service:create(node, true); |
| 27 pubsub.service:set_affiliation("stats", true, actor, "publisher"); | 29 pubsub.service:set_affiliation(node, true, actor, "publisher"); |
| 28 end | 30 end |
| 29 | 31 |
| 30 module:hook_global("stats-updated", function (event) | 32 module:hook_global("stats-updated", function (event) |
| 31 publish_stats(event.stats, event.stats_extra); | 33 publish_stats(event.stats, event.stats_extra); |
| 32 end); | 34 end); |
| 33 | 35 |
| 34 function module.unload() | 36 function module.unload() |
| 35 pubsub.service:delete("stats", true); | 37 pubsub.service:delete(node, true); |
| 36 end | 38 end |
