Mercurial > prosody-modules
view mod_measure_client_identities/mod_measure_client_identities.lua @ 4542:fb4a50bf60f1
mod_prometheus: Invoke stats collection if in 'manual' mode
Since 10d13e0554f9 a special value for statistics_interval "manual"
exists, where a module is expected to invoke processing in connection to
collection of stats. This makes internal collection and exporting to
Prometheus happens at the same time with no chance of timers getting out
of sync.
author | Kim Alvefur <zash@zash.se> |
---|---|
date | Tue, 13 Apr 2021 23:53:53 +0200 |
parents | fdbf7c2aed7b |
children |
line wrap: on
line source
module:set_global(); local measure = require"core.statsmanager".measure; local counters = { unknown = measure("amount", "client_identities.unknown"), }; module:hook("stats-update", function () local buckets = { unknown = 0, }; for _, session in pairs(prosody.full_sessions) do if session.caps_cache ~= nil then local node_string = session.caps_cache.attr.node; local node = node_string:match("([^#]+)"); if buckets[node] == nil then buckets[node] = 0; end buckets[node] = buckets[node] + 1; else buckets.unknown = buckets.unknown + 1; end end local visited = {}; for bucket, count in pairs(buckets) do if counters[bucket] == nil then counters[bucket] = measure("amount", "client_identities."..bucket); end counters[bucket](count); visited[bucket] = true; end for bucket, counter in pairs(counters) do if not visited[bucket] then counter(0); end end end)