diff mod_pubsub_stats/mod_pubsub_stats.lua @ 3067:b01ef74c9fc0

mod_pubsub_stats: Simple module that publishes stats in XEP-0039 format
author Kim Alvefur <zash@zash.se>
date Tue, 29 May 2018 05:31:56 +0200
parents
children 380f92276e57
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/mod_pubsub_stats/mod_pubsub_stats.lua	Tue May 29 05:31:56 2018 +0200
@@ -0,0 +1,36 @@
+local st = require "util.stanza";
+local dt = require "util.datetime";
+
+local pubsub = module:depends"pubsub";
+
+local actor = module.host .. "/modules/" .. module.name;
+
+local function publish_stats(stats, stats_extra)
+	local id = "current";
+	local xitem = st.stanza("item", { id = id })
+		:tag("query", { xmlns = "http://jabber.org/protocol/stats" });
+
+	for name, value in pairs(stats) do
+		local stat_extra = stats_extra[name];
+		local unit = stat_extra and stat_extra.units;
+		xitem:tag("stat", { name = name, unit = unit, value = tostring(value) }):up();
+	end
+
+	local ok, err = pubsub.service:publish("stats", actor, id, xitem);
+	if not ok then
+		module:log("error", "Error publishing stats: %s", err);
+	end
+end
+
+function module.load()
+	pubsub.service:create("stats", true);
+	pubsub.service:set_affiliation("stats", true, actor, "publisher");
+end
+
+module:hook_global("stats-updated", function (event)
+	publish_stats(event.stats, event.stats_extra);
+end);
+
+function module.unload()
+	pubsub.service:delete("stats", true);
+end