view mod_measure_memory/mod_measure_memory.lua @ 1647:8860405e2af6

mod_muc_ban_ip: Increase priority of hooks, fixes if eg mod_presence gets called first
author Kim Alvefur <zash@zash.se>
date Mon, 06 Apr 2015 22:21:23 +0200
parents 2c39af0fb93b
children 4d38b8c03dfe
line wrap: on
line source

module:set_global();

local measure = require"core.statsmanager".measure;
local have_pposix, pposix = pcall(require, "util.pposix");

local measures = {};
setmetatable(measures, {
	__index = function (t, k, m)
		m = measure("sizes", "memory."..k); t[k] = m; return m;
	end
});

module:hook("stats-update", function ()
	measures.lua(collectgarbage("count")*1024);
end);

if have_pposix and pposix.meminfo then
	module:hook_global("stats-update", function ()
		local m = measures;
		for k, v in pairs(pposix.meminfo()) do
			m[k](v);
		end
	end);
end

local statm = io.open("/proc/self/statm");
if statm then
	statm:close();
	local pagesize = module:get_option_number("memory_pagesize", 4096); -- getconf PAGESIZE

	module:hook("stats-update", function ()
		local statm, err = io.open("/proc/self/statm");
		if not statm then
			module:log("error", tostring(err));
			return;
		end
		-- virtual memory (caches, opened librarys, everything)
		measures.total(statm:read("*n") * pagesize);
		-- resident set size (actually used memory)
		measures.rss(statm:read("*n") * pagesize);
		statm:close();
	end);
end