aboutsummaryrefslogtreecommitdiffstats
path: root/plugins/mod_csi.lua
diff options
context:
space:
mode:
Diffstat (limited to 'plugins/mod_csi.lua')
-rw-r--r--plugins/mod_csi.lua20
1 files changed, 20 insertions, 0 deletions
diff --git a/plugins/mod_csi.lua b/plugins/mod_csi.lua
index 82efd831..63e0154f 100644
--- a/plugins/mod_csi.lua
+++ b/plugins/mod_csi.lua
@@ -1,8 +1,10 @@
+local statsmanager = require "prosody.core.statsmanager";
local st = require "prosody.util.stanza";
local xmlns_csi = "urn:xmpp:csi:0";
local csi_feature = st.stanza("csi", { xmlns = xmlns_csi });
local change = module:metric("counter", "changes", "events", "CSI state changes", {"csi_state"});
+local count = module:metric("gauge", "state", "sessions", "", { "state" });
module:hook("stream-features", function (event)
if event.origin.username then
@@ -23,3 +25,21 @@ end
module:hook("stanza/"..xmlns_csi..":active", refire_event("csi-client-active"));
module:hook("stanza/"..xmlns_csi..":inactive", refire_event("csi-client-inactive"));
+
+module:hook_global("stats-update", function()
+ local sessions = prosody.hosts[module.host].sessions;
+ if not sessions then return end
+ statsmanager.cork();
+ -- Can't do :clear() on host-scoped measures?
+ count:with_labels("active"):set(0);
+ count:with_labels("inactive"):set(0);
+ count:with_labels("flushing"):set(0);
+ for user, user_session in pairs(sessions) do
+ for resource, session in pairs(user_session.sessions) do
+ if session.state == "inactive" or session.state == "active" or session.state == "flushing" then
+ count:with_labels(session.state):add(1);
+ end
+ end
+ end
+ statsmanager.uncork();
+end);