Mercurial > prosody-modules
view mod_measure_client_identities/mod_measure_client_identities.lua @ 5381:32a9817c7516
mod_firewall: Initialize compiled chunk just once for all handlers
This should fix a case where some stateful dependencies (such as throttles)
produce separate instances for every call to new_handler(), leading to
surprising behaviour (e.g. rules executed via JUMP CHAIN vs ::deliver would
have separate rate limits).
This also adds better error handling in case the compiled code fails to run
for some reason.
author | Matthew Wild <mwild1@gmail.com> |
---|---|
date | Fri, 28 Apr 2023 13:27:06 +0100 |
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)