mirror of https://github.com/bjc/prosody
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
105 lines
2.6 KiB
105 lines
2.6 KiB
-- Prosody IM
|
|
-- Copyright (C) 2008-2010 Matthew Wild
|
|
-- Copyright (C) 2008-2010 Waqas Hussain
|
|
--
|
|
-- This project is MIT/X11 licensed. Please see the
|
|
-- COPYING file in the source package for more information.
|
|
--
|
|
|
|
local pcall = pcall;
|
|
|
|
local find = string.find;
|
|
local ipairs, pairs, setmetatable = ipairs, pairs, setmetatable;
|
|
|
|
module "logger"
|
|
|
|
local name_sinks, level_sinks = {}, {};
|
|
local name_patterns = {};
|
|
|
|
local make_logger;
|
|
|
|
function init(name)
|
|
local log_debug = make_logger(name, "debug");
|
|
local log_info = make_logger(name, "info");
|
|
local log_warn = make_logger(name, "warn");
|
|
local log_error = make_logger(name, "error");
|
|
|
|
--name = nil; -- While this line is not commented, will automatically fill in file/line number info
|
|
local namelen = #name;
|
|
return function (level, message, ...)
|
|
if level == "debug" then
|
|
return log_debug(message, ...);
|
|
elseif level == "info" then
|
|
return log_info(message, ...);
|
|
elseif level == "warn" then
|
|
return log_warn(message, ...);
|
|
elseif level == "error" then
|
|
return log_error(message, ...);
|
|
end
|
|
end
|
|
end
|
|
|
|
function make_logger(source_name, level)
|
|
local level_handlers = level_sinks[level];
|
|
if not level_handlers then
|
|
level_handlers = {};
|
|
level_sinks[level] = level_handlers;
|
|
end
|
|
|
|
local source_handlers = name_sinks[source_name];
|
|
|
|
local logger = function (message, ...)
|
|
if source_handlers then
|
|
for i = 1,#source_handlers do
|
|
if source_handlers[i](source_name, level, message, ...) == false then
|
|
return;
|
|
end
|
|
end
|
|
end
|
|
|
|
for i = 1,#level_handlers do
|
|
level_handlers[i](source_name, level, message, ...);
|
|
end
|
|
end
|
|
|
|
return logger;
|
|
end
|
|
|
|
function reset()
|
|
for k in pairs(name_sinks) do name_sinks[k] = nil; end
|
|
for level, handler_list in pairs(level_sinks) do
|
|
-- Clear all handlers for this level
|
|
for i = 1, #handler_list do
|
|
handler_list[i] = nil;
|
|
end
|
|
end
|
|
for k in pairs(name_patterns) do name_patterns[k] = nil; end
|
|
end
|
|
|
|
function add_level_sink(level, sink_function)
|
|
if not level_sinks[level] then
|
|
level_sinks[level] = { sink_function };
|
|
else
|
|
level_sinks[level][#level_sinks[level] + 1 ] = sink_function;
|
|
end
|
|
end
|
|
|
|
function add_name_sink(name, sink_function, exclusive)
|
|
if not name_sinks[name] then
|
|
name_sinks[name] = { sink_function };
|
|
else
|
|
name_sinks[name][#name_sinks[name] + 1] = sink_function;
|
|
end
|
|
end
|
|
|
|
function add_name_pattern_sink(name_pattern, sink_function, exclusive)
|
|
if not name_patterns[name_pattern] then
|
|
name_patterns[name_pattern] = { sink_function };
|
|
else
|
|
name_patterns[name_pattern][#name_patterns[name_pattern] + 1] = sink_function;
|
|
end
|
|
end
|
|
|
|
_M.new = make_logger;
|
|
|
|
return _M;
|
|
|