Mercurial > prosody-modules
diff mod_prometheus/mod_prometheus.lua @ 3138:9817e45a79e6
mod_prometheus: Fetch statistics at every request.
author | Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> |
---|---|
date | Fri, 22 Jun 2018 15:19:55 +0200 |
parents | 99ac6dda9878 |
children | ccbfe7df02dc |
line wrap: on
line diff
--- a/mod_prometheus/mod_prometheus.lua Fri Jun 22 11:12:34 2018 +0200 +++ b/mod_prometheus/mod_prometheus.lua Fri Jun 22 15:19:55 2018 +0200 @@ -12,8 +12,7 @@ local t_insert = table.insert; local t_concat = table.concat; local socket = require "socket"; - -local data = {}; +local get_stats = require "core.statsmanager".get_stats; local function escape(text) return text:gsub("\\", "\\\\"):gsub("\"", "\\\""):gsub("\n", "\\n"); @@ -61,15 +60,12 @@ return escape_name(metric)..repr_labels(labels).." "..value.." "..timestamp.."\n"; end -module:hook("stats-updated", function (event) - local all_stats, this = event.stats_extra; +local function parse_stats() local timestamp = tostring(get_timestamp()); - local host, sect, name, typ; - data = {}; - for stat, value in pairs(event.stats) do - this = all_stats[stat]; + local data = {}; + for stat, value in pairs(get_stats()) do -- module:log("debug", "changed_stats[%q] = %s", stat, tostring(value)); - host, sect, name, typ = stat:match("^/([^/]+)/([^/]+)/(.+):(%a+)$"); + local host, sect, name, typ = stat:match("^/([^/]+)/([^/]+)/(.+):(%a+)$"); if host == nil then sect, name, typ = stat:match("^([^.]+)%.(.+):(%a+)$"); elseif host == "*" then @@ -97,14 +93,15 @@ end t_insert(data[key], field); end -end); + return data; +end local function get_metrics(event) local response = event.response; - response.headers.content_type = "text/plain; version=0.4.4"; + response.headers.content_type = "text/plain; version=0.0.4"; local answer = {}; - for key, fields in pairs(data) do + for key, fields in pairs(parse_stats()) do t_insert(answer, repr_help(key, "TODO: add a description here.")); t_insert(answer, repr_type(key, fields[1].typ)); for _, field in pairs(fields) do