mod_sasl2_fast: Improve handling when SASL profile unexpectedly lacks CB
This fixes a traceback reported by riau, but likely does not solve the
underlying cause, whatever that is.
module:set_global();
local s_format = string.format;
local t_insert = table.insert;
local t_concat = table.concat;
local array = require"util.array";
local it = require"util.iterators";
local mt = require"util.multitable";
local meta = mt.new(); meta.data = module:shared"meta";
local data = mt.new(); data.data = module:shared"data";
local munin_listener = {};
local munin_commands = {};
local node_name = module:get_option_string("munin_node_name", "localhost");
local ignore_stats = module:get_option_set("munin_ignored_stats", { });
local function clean_fieldname(name)
return (name:gsub("[^A-Za-z0-9_]", "_"):gsub("^[^A-Za-z_]", "_%1"));
end
function munin_listener.onconnect(conn)
-- require"core.statsmanager".collect();
conn:write("# munin node at "..node_name.."\n");
end
function munin_listener.onincoming(conn, line)
line = line and line:match("^[^\r\n]+");
if type(line) ~= "string" then return end
-- module:log("debug", "incoming: %q", line);
local command = line:match"^%w+";
command = munin_commands[command];
if not command then
conn:write("# Unknown command.\n");
return;
end
local ok, err = pcall(command, conn, line);
if not ok then
module:log("error", "Error running %q: %s", line, err);
conn:close();
end
end
function munin_listener.ondisconnect() end
function munin_commands.cap(conn)
conn:write("cap\n");
end
function munin_commands.list(conn)
conn:write(array(it.keys(data.data)):concat(" ") .. "\n");
end
function munin_commands.config(conn, line)
-- TODO what exactly?
local stat = line:match("%s(%S+)");
if not stat then conn:write("# Unknown service\n.\n"); return end
for _, _, k, value in meta:iter(stat, "", nil) do
conn:write(s_format("%s %s\n", k, value));
end
for _, name, k, value in meta:iter(stat, nil, nil) do
if name ~= "" and not ignore_stats:contains(name) then
conn:write(s_format("%s.%s %s\n", name, k, value));
end
end
conn:write(".\n");
end
function munin_commands.fetch(conn, line)
local stat = line:match("%s(%S+)");
if not stat then conn:write("# Unknown service\n.\n"); return end
for _, name, value in data:iter(stat, nil) do
if not ignore_stats:contains(name) then
conn:write(s_format("%s.value %.12f\n", name, value));
end
end
conn:write(".\n");
end
function munin_commands.quit(conn)
conn:close();
end
module:hook("stats-updated", function (event)
local all_stats, this = event.stats_extra;
local host, sect, name, typ, key;
for stat, value in pairs(event.changed_stats) do
if not ignore_stats:contains(stat) then
this = all_stats[stat];
-- module:log("debug", "changed_stats[%q] = %s", stat, tostring(value));
host, sect, name, typ = stat:match("^/([^/]+)/([^/]+)/(.+):(%a+)$");
if host == nil then
sect, name, typ = stat:match("^([^.]+)%.(.+):(%a+)$");
elseif host == "*" then
host = nil;
end
if sect:find("^mod_measure_.") then
sect = sect:sub(13);
elseif sect:find("^mod_statistics_.") then
sect = sect:sub(16);
end
key = clean_fieldname(s_format("%s_%s_%s", host or "global", sect, typ));
if not meta:get(key) then
if host then
meta:set(key, "", "graph_title", s_format("%s %s on %s", sect, typ, host));
else
meta:set(key, "", "graph_title", s_format("Global %s %s", sect, typ));
end
meta:set(key, "", "graph_vlabel", this and this.units or typ);
meta:set(key, "", "graph_category", sect);
meta:set(key, name, "label", name);
elseif not meta:get(key, name, "label") then
meta:set(key, name, "label", name);
end
data:set(key, name, value);
end
end
end);
local function openmetrics_handler(event)
local registry = event.metric_registry
local host, sect, name, typ, key;
for family_name, metric_family in pairs(registry:get_metric_families()) do
if not ignore_stats:contains(family_name) then
-- module:log("debug", "changed_stats[%q] = %s", stat, tostring(value));
local host_key
if metric_family.label_keys[1] == "host" then
host_key = 1
end
if family_name:sub(1, 12) == "prosody_mod_" then
sect, name = family_name:match("^prosody_mod_([^/]+)/(.+)$")
else
sect, name = family_name:match("^([^_]+)_(.+)$")
end
name = clean_fieldname(name)
local metric_type = metric_family.type_
if metric_type == "gauge" or metric_type == "unknown" then
typ = "GAUGE"
else
typ = "DCOUNTER"
end
for labelset, metric in metric_family:iter_metrics() do
host = host_key and labelset[host_key] or "global"
local name_parts = {}
for i, label in ipairs(labelset) do
if i ~= host_key then
t_insert(name_parts, label)
end
end
local full_name = t_concat(name_parts, "_")
local display_name = #name_parts > 0 and full_name or name
key = clean_fieldname(s_format("%s_%s_%s", host or "global", sect, name));
local unit
local factor = 1
unit = metric_family.unit
if unit == "seconds" and typ == "DCOUNTER" then
factor = 100
unit = "%time"
elseif typ == "DCOUNTER" then
unit = unit .. "/s"
end
if not meta:get(key, "") then
meta:set(key, "", "graph_title", s_format(metric_family.description));
if unit ~= "" then
meta:set(key, "", "graph_vlabel", unit);
end
meta:set(key, "", "graph_category", sect);
end
if not meta:get(key, display_name) then
meta:set(key, display_name, "label", display_name);
meta:set(key, display_name, "type", typ)
end
for suffix, extra_labels, value in metric:iter_samples() do
if metric_type == "histogram" or metric_type == "summary" then
if suffix == "_sum" then
data:set(key, display_name, value * factor)
end
elseif suffix == "_total" or suffix == "" then
data:set(key, display_name, value * factor)
end
end
end
end
end
end
module:hook("openmetrics-updated", openmetrics_handler);
module:provides("net", {
listener = munin_listener;
default_mode = "*l";
default_port = 4949;
});