2 local stats = require "util.statistics".new();
3 local config = require "core.configmanager";
4 local log = require "util.logger".init("stats");
5 local timer = require "util.timer";
6 local fire_event = prosody.events.fire_event;
8 local stats_config = config.get("*", "statistics_interval");
9 local stats_interval = tonumber(stats_config);
10 if stats_config and not stats_interval then
11 log("error", "Invalid 'statistics_interval' setting, statistics will be disabled");
14 local measure, collect;
15 local latest_stats = {};
16 local changed_stats = {};
17 local stats_extra = {};
19 if stats_interval then
20 log("debug", "Statistics collection is enabled every %d seconds", stats_interval);
21 function measure(type, name)
22 local f = assert(stats[type], "unknown stat type: "..type);
26 local mark_collection_start = measure("times", "stats.collection");
27 local mark_processing_start = measure("times", "stats.processing");
30 local mark_collection_done = mark_collection_start();
31 fire_event("stats-update");
32 changed_stats, stats_extra = {}, {};
33 for stat_name, getter in pairs(stats.get_stats()) do
34 local type, value, extra = getter();
35 local old_value = latest_stats[stat_name];
36 latest_stats[stat_name] = value;
37 if value ~= old_value then
38 changed_stats[stat_name] = value;
41 stats_extra[stat_name] = extra;
44 mark_collection_done();
45 local mark_processing_done = mark_processing_start();
46 fire_event("stats-updated", { stats = latest_stats, changed_stats = changed_stats, stats_extra = stats_extra });
47 mark_processing_done();
48 return stats_interval;
51 timer.add_task(stats_interval, collect);
52 prosody.events.add_handler("server-started", function () collect() end, -1);
54 log("debug", "Statistics collection is disabled");
66 get_stats = function ()
67 return latest_stats, changed_stats, stats_extra;