# HG changeset patch
# User Kim Alvefur <zash@zash.se>
# Date 1401818450 -7200
# Node ID 3f85aaca82822544cbbaee99cce0ffc8bbc84ab8
# Parent  091ee76745e8d02a1f0f57982dd2f47088a89277
misc/munin/prosody_.lua: Munin script for collecting statistics from mod_statistics

diff -r 091ee76745e8 -r 3f85aaca8282 misc/munin/prosody_.lua
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/misc/munin/prosody_.lua	Tue Jun 03 20:00:50 2014 +0200
@@ -0,0 +1,147 @@
+#!/usr/bin/env lua
+
+local print = print;
+local pairs = pairs;
+local socket = require"socket";
+
+local stats = {};
+
+stats.c2s = {
+	graph_title = "Prosody C2S Connections";
+	graph_vlabel = "users";
+	graph_category = "Prosody";
+	all_client_connections = {
+		label = "client connections";
+		_key = "total_c2s";
+	}
+}
+
+stats.s2s = {
+	graph_title = "Prosody S2S Connections";
+	graph_vlabel = "servers";
+	graph_category = "Prosody";
+	outgoing_connections = {
+		label = "outgoing connections";
+		_key = "total_s2sout";
+	};
+	incoming_connections = {
+		label = "incoming connections";
+		_key = "total_s2sin";
+	}
+}
+
+stats.mem = {
+	graph_title = "Prosody Memory Usage";
+	graph_vlabel = "Bytes";
+	graph_args = "--base 1024 -l 0";
+	graph_category = "Prosody"; --memory_unused
+	graph_order = "memory_total memory_rss memory_allocated memory_used memory_lua memory_returnable";
+
+	memory_allocated = { label = "Allocated", draw = "AREA"  };
+	memory_lua = { label = "Lua", draw = "AREA" };
+	memory_rss = { label = "RSS", draw = "AREA" };
+	memory_total = { label = "Total", draw = "AREA" };
+	-- memory_unused = { label = "Unused", draw = "AREA" };
+	memory_used = { label = "Used", draw = "AREA" };
+	memory_returnable = { label = "Returnable", draw = "AREA" };
+}
+
+stats.cpu = {
+	graph_title = "Prosody CPU Usage";
+	graph_category = "Prosody";
+	graph_args = "-l 0";
+	graph_vlabel = "CPU time used in milliseconds";
+
+	cpu_total = { label = "CPU"; type = "DERIVE"; min = 0; };
+}
+
+stats.auth = {
+	graph_title = "Prosody Authentications";
+	graph_category = "Prosody";
+	graph_args = "--base 1000";
+
+	c2s_auth = {
+		label = "Logins";
+		type = "DERIVE";
+		min = 0;
+	};
+	c2s_authfail = {
+		label = "Failed logins";
+		type = "DERIVE";
+		min = 0;
+	};
+}
+
+local function onerror(msg, err, exit)
+	io.stderr:write(msg, '\n');
+	if err then
+		io.stderr:write(err, '\n');
+	end
+	os.exit(exit or 1);
+end
+
+
+local function connect()
+	local conn, err = socket.connect(os.getenv"host" or "localhost", os.getenv"port" or 5782);
+	if not conn then onerror("Could not connect to prosody", err); end
+	conn:settimeout(1);
+	return conn;
+end
+
+local function get_config(item)
+	for k,v in pairs(item) do
+		if type(v) == "string" then
+			print(k .. " " .. v);
+		elseif type(v) == "table" then
+			for sk,v in pairs(v) do
+				if not sk:match("^_") then
+					print(k.."."..sk.." "..v);
+				end
+			end
+		end
+	end
+end
+
+local function get_stats(item)
+	local labels = {};
+	for key, val in pairs(item) do
+		if type(val) == "table" and val.label then
+			labels[val._key or key] = key;
+		end
+	end
+
+	local conn = connect();
+	local line, err = conn:receive("*l");
+	local stat, value, label;
+	while line and line ~= "" and next(labels) ~= nil do
+		stat, value = line:match('^STAT%s+"([^"]*)"%s*(%b())');
+		label = stat and labels[stat];
+		if label then
+			print(label..".value "..tonumber(value:sub(2,-2)));
+			labels[stat] = nil;
+		end
+		line, err = conn:receive("*l");
+	end
+	if err then onerror(err); end
+end
+
+local function main(stat, mode)
+	if mode == "suggest" then
+		for available_stat in pairs(stats) do
+			print(available_stat);
+		end
+	elseif mode == "config" then
+		return get_config(stats[stat]);
+	elseif stats[stat] then
+		return get_stats(stats[stat]);
+	end
+end
+
+if arg then return main(arg[0]:match("prosody_(%w*)"), ...); end
+
+return {
+	stats = stats,
+	get_stats = get_stats,
+	get_config = get_config,
+}
+