From 0a910917e7a4c846444d4e4bdb1fe212d5f6145e Mon Sep 17 00:00:00 2001 From: Kim Alvefur Date: Sat, 17 Aug 2024 23:13:28 +0200 Subject: mod_csi: Count how many sessions are using CSI and their state --- plugins/mod_csi.lua | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) 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); -- cgit v1.2.3