mirrors/prosody
1
0
Fork 0
mirror of https://github.com/bjc/prosody.git synced 2025-04-03 21:27:38 +03:00
Code Issues Projects Releases Packages Wiki Activity Actions

mod_csi: Count how many sessions are using CSI and their state

Browse source
This commit is contained in:
Kim Alvefur 2024-08-17 23:13:28 +02:00
parent 4ca4a6af7d
commit 0a910917e7
1 changed files with 20 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

20
plugins/mod_csi.lua
View file

@ -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);