From 4ffd66fad936850b9f188c847158c8d2d4a66527 Mon Sep 17 00:00:00 2001 From: Mark Allen Date: Thu, 18 May 2017 18:48:54 -0500 Subject: [PATCH] WIP --- src/lager_console_backend.erl | 107 ++++++++++++++++++++++++++-------- 1 file changed, 84 insertions(+), 23 deletions(-) diff --git a/src/lager_console_backend.erl b/src/lager_console_backend.erl index e922ec3..734eb1c 100644 --- a/src/lager_console_backend.erl +++ b/src/lager_console_backend.erl @@ -14,8 +14,17 @@ %% specific language governing permissions and limitations %% under the License. -%% @doc Console backend for lager. Configured with a single option, the loglevel -%% desired. +%% @doc Console backend for lager. +%% Configuration is a proplist with the following keys: +%% -module(lager_console_backend). @@ -25,6 +34,7 @@ code_change/3]). -record(state, {level :: {'mask', integer()}, + out = user :: user | standard_error, formatter :: atom(), format_config :: any(), colors=[] :: list()}). @@ -34,17 +44,33 @@ -compile([{parse_transform, lager_transform}]). -endif. + -include("lager.hrl"). + -define(TERSE_FORMAT,[time, " ", color, "[", severity,"] ", message]). +-define(DEFAULT_FORMAT_CONFIG, ?TERSE_FORMAT ++ [eol()]). +-define(FORMAT_CONFIG_OFF, [{eol, eol()}]). + +-ifdef(TEST). +-define(DEPRECATION(_Msg), ok). +-else. +-define(DEPRECATION(Msg), + io:format(user, "WARNING: This is a deprecated console configuration. Please use \"~p\" instead.", [Msg])). +-endif. %% @private init([Level]) when is_atom(Level) -> - init(Level); -init([Level, true]) -> % for backwards compatibility - init([Level,{lager_default_formatter,[{eol, eol()}]}]); -init([Level,false]) -> % for backwards compatibility - init([Level,{lager_default_formatter,?TERSE_FORMAT ++ [eol()]}]); -init([Level,{Formatter,FormatterConfig}]) when is_atom(Formatter) -> + ?DEPRECATION([{level, Level}]), + init([{level, Level}]); +init([Level, true]) when is_atom(Level) -> % for backwards compatibility + ?DEPRECATION([{level, Level}, {formatter_config, [{eol, "\\r\\n\\"}]}]), + init([{level, Level}, {formatter_config, ?FORMAT_CONFIG_OFF}]); +init([Level,false]) when is_atom(Level) -> % for backwards compatibility + ?DEPRECATION([{level, Level}]), + init([{level, Level}]); + +init(Options) when is_list(Options) -> + true = validate_options(Options), Colors = case application:get_env(lager, colored) of {ok, true} -> {ok, LagerColors} = application:get_env(lager, colors), @@ -52,6 +78,7 @@ init([Level,{Formatter,FormatterConfig}]) when is_atom(Formatter) -> _ -> [] end, + Level = get_option(level, Options, undefined), try {is_new_style_console_available(), lager_util:config_to_mask(Level)} of {false, _} -> Msg = "Lager's console backend is incompatible with the 'old' shell, not enabling it", @@ -65,23 +92,57 @@ init([Level,{Formatter,FormatterConfig}]) when is_atom(Formatter) -> io:format("WARNING: " ++ Msg ++ "~n"), ?INT_LOG(warning, Msg, []), {error, {fatal, old_shell}}; - {true, Levels} -> - {ok, #state{level=Levels, + {true, L} -> + [UseErr, Formatter, Config] = [ get_option(K, Options, Default) || {K, Default} <- [ + {use_stderr, false}, + {formatter, lager_default_formatter}, + {formatter_config, ?DEFAULT_FORMAT_CONFIG} + ] + ], + Out = case UseErr of + false -> user; + true -> standard_error + end, + {ok, #state{level=L, + out=Out, formatter=Formatter, - format_config=FormatterConfig, + format_config=Config, colors=Colors}} catch _:_ -> {error, {fatal, bad_log_level}} end; -init(Level) -> - init([Level,{lager_default_formatter,?TERSE_FORMAT ++ [eol()]}]). +init(Level) when is_atom(Level) -> + ?DEPRECATION([{level, Level}]), + init([{level, Level}]); +init(Other) -> + {error, {fatal, {bad_console_config, Other}}}. + +validate_options([]) -> true; +validate_options([{level, L}|T]) when is_atom(L) -> + validate_options(T); +validate_options([{use_stderr, true}|T]) -> + validate_options(T); +validate_options([{use_stderr, false}|T]) -> + validate_options(T); +validate_options([{formatter, M}|T]) when is_atom(M) -> + validate_options(T); +validate_options([{formatter_config, C}|T]) when is_list(C) -> + validate_options(T); +validate_options([H|_]) -> + throw({error, {fatal, {bad_console_config, H}}}). + +get_option(K, Options, Default) -> + case lists:keyfind(K, 1, Options) of + {K, V} -> V; + false -> Default + end. %% @private handle_call(get_loglevel, #state{level=Level} = State) -> {ok, Level, State}; handle_call({set_loglevel, Level}, State) -> - try lager_util:config_to_mask(Level) of + try lager_util:config_to_mask(Level) of Levels -> {ok, ok, State#state{level=Levels}} catch @@ -93,10 +154,10 @@ handle_call(_Request, State) -> %% @private handle_event({log, Message}, - #state{level=L,formatter=Formatter,format_config=FormatConfig,colors=Colors} = State) -> + #state{level=L,out=Out,formatter=Formatter,format_config=FormatConfig,colors=Colors} = State) -> case lager_util:is_loggable(Message, L, ?MODULE) of true -> - io:put_chars(user, Formatter:format(Message,FormatConfig,Colors)), + io:put_chars(Out, Formatter:format(Message,FormatConfig,Colors)), {ok, State}; false -> {ok, State} @@ -186,7 +247,7 @@ console_log_test_() -> unregister(user), register(user, Pid), erlang:group_leader(Pid, whereis(lager_event)), - gen_event:add_handler(lager_event, lager_console_backend, info), + gen_event:add_handler(lager_event, lager_console_backend, [{level, info}]), lager_config:set({lager_event, loglevel}, {element(2, lager_util:config_to_mask(info)), []}), lager:log(info, self(), "Test message"), receive @@ -227,7 +288,7 @@ console_log_test_() -> register(user, Pid), erlang:group_leader(Pid, whereis(lager_event)), gen_event:add_handler(lager_event, lager_console_backend, - [info, {lager_default_formatter, [date,"#",time,"#",severity,"#",node,"#",pid,"#", + [{level, info}, {formatter, lager_default_formatter}, {formatter_config, [date,"#",time,"#",severity,"#",node,"#",pid,"#", module,"#",function,"#",file,"#",line,"#",message,"\r\n"]}]), lager_config:set({lager_event, loglevel}, {?INFO, []}), lager:info("Test message"), @@ -250,7 +311,7 @@ console_log_test_() -> Pid = spawn(F(self())), unregister(user), register(user, Pid), - gen_event:add_handler(lager_event, lager_console_backend, info), + gen_event:add_handler(lager_event, lager_console_backend, [{level, info}]), erlang:group_leader(Pid, whereis(lager_event)), lager_config:set({lager_event, loglevel}, {element(2, lager_util:config_to_mask(info)), []}), lager:debug("Test message"), @@ -280,7 +341,7 @@ console_log_test_() -> Pid = spawn(F(self())), unregister(user), register(user, Pid), - gen_event:add_handler(lager_event, lager_console_backend, info), + gen_event:add_handler(lager_event, lager_console_backend, [{level, info}]), lager_config:set({lager_event, loglevel}, {element(2, lager_util:config_to_mask(info)), []}), erlang:group_leader(Pid, whereis(lager_event)), lager:debug("Test message"), @@ -319,7 +380,7 @@ console_log_test_() -> Pid = spawn(F(self())), unregister(user), register(user, Pid), - gen_event:add_handler(lager_event, lager_console_backend, info), + gen_event:add_handler(lager_event, lager_console_backend, [{level, info}]), lager_config:set({lager_event, loglevel}, {element(2, lager_util:config_to_mask(info)), []}), lager:set_loglevel(lager_console_backend, '!=info'), erlang:group_leader(Pid, whereis(lager_event)), @@ -350,7 +411,7 @@ console_log_test_() -> Pid = spawn(F(self())), unregister(user), register(user, Pid), - gen_event:add_handler(lager_event, lager_console_backend, info), + gen_event:add_handler(lager_event, lager_console_backend, [{level, info}]), lager_config:set({lager_event, loglevel}, {element(2, lager_util:config_to_mask(info)), []}), lager:set_loglevel(lager_console_backend, '=debug'), erlang:group_leader(Pid, whereis(lager_event)), @@ -384,7 +445,7 @@ set_loglevel_test_() -> fun() -> error_logger:tty(false), application:load(lager), - application:set_env(lager, handlers, [{lager_console_backend, info}]), + application:set_env(lager, handlers, [{lager_console_backend, [{level, info}]), application:set_env(lager, error_logger_redirect, false), lager:start() end,