rewrite from lager
25개 이상의 토픽을 선택하실 수 없습니다. Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

676 lines
22 KiB

  1. -module(eRum1).
  2. -include("rumDef.hrl").
  3. -include("rumMsg.hrl").
  4. -include("rumCom.hrl").
  5. %% API
  6. -export([
  7. %% start stop
  8. start/0
  9. , stop/0
  10. %% log and log param
  11. , dispatch_log/13
  12. , do_log/16
  13. , do_log_unsafe/16
  14. , do_log_impl/16
  15. , safe_format/3
  16. , safe_format_chop/3
  17. , unsafe_format/2
  18. , md/0
  19. , md/1
  20. , getLogLevel/1
  21. , getLogLevel/2
  22. , setLogLevel/2
  23. , setLogLevel/3
  24. , setLogLevel/4
  25. , getLogLevels/1
  26. , updateLogevelCfg/1
  27. , setLogHwm/2
  28. , setLogHwm/3
  29. , setLogHwm/4
  30. , rotate_handler/1
  31. , rotate_handler/2
  32. , rotate_sink/1
  33. , rotate_all/0
  34. , posix_error/1
  35. %% stack
  36. , pr/2
  37. , pr/3
  38. , pr_stacktrace/1
  39. , pr_stacktrace/2
  40. %% trace
  41. , trace/2
  42. , trace/3
  43. , trace_file/2
  44. , trace_file/3
  45. , trace_file/4
  46. , trace_console/1
  47. , trace_console/2
  48. , install_trace/2
  49. , install_trace/3
  50. , remove_trace/1
  51. , trace_state/3
  52. , trace_func/3
  53. , list_all_sinks/0
  54. , clear_all_traces/0
  55. , clear_trace_by_destination/1
  56. , stop_trace/1
  57. , stop_trace/3
  58. , status/0
  59. ]).
  60. -record(trace_func_state_v1, {
  61. pid :: undefined | pid(),
  62. level :: rumAtomLevel(),
  63. count :: infinity | pos_integer(),
  64. format_string :: string(),
  65. timeout :: infinity | pos_integer(),
  66. started = os:timestamp() :: erlang:timestamp() %% use os:timestamp for compatability
  67. }).
  68. start() ->
  69. application:ensure_all_started(eRum).
  70. stop() ->
  71. application:stop(eRum).
  72. -spec dispatch_log(atom(), rumAtomLevel(), pid(), node(), atom(), atom(), atom(), integer(), list(), string(), list() | none, pos_integer(), safe | unsafe) -> ok | {error, lager_not_running} | {error, {sink_not_configured, atom()}}.
  73. %% this is the same check that the parse transform bakes into the module at compile time see rumTransform (lines 173-216)
  74. dispatch_log(Sink, Severity, Pid, Node, Module, Function, File, Line, Metadata, Format, Args, Size, Safety) ->
  75. SeverityAsInt = rumUtil:levelToNum(Severity),
  76. case {whereis(Sink), rumConfig:get({Sink, loglevel}, {?none, []})} of
  77. {undefined, _} -> {error, lager_not_running};
  78. {SinkPid, {Level, Traces}} when Safety =:= safe andalso ((Level band SeverityAsInt) /= 0 orelse Traces /= []) ->
  79. do_log(Severity, Pid, Node, Module, Function, File, Line, Metadata, Format, Args, Size, SeverityAsInt, Level, Traces, Sink, SinkPid);
  80. {SinkPid, {Level, Traces}} when Safety =:= unsafe andalso ((Level band SeverityAsInt) /= 0 orelse Traces /= []) ->
  81. do_log_unsafe(Severity, Pid, Node, Module, Function, File, Line, Metadata, Format, Args, Size, SeverityAsInt, Level, Traces, Sink, SinkPid);
  82. _ -> ok
  83. end.
  84. %% Should only be called externally from code generated from the parse transform
  85. do_log(Severity, Pid, Node, Module, Function, File, Line, Metadata, Format, Args, Size, SeverityAsInt, LevelThreshold, TraceFilters, Sink, SinkPid) ->
  86. FormatFun = fun() -> safe_format_chop(Format, Args, Size) end,
  87. do_log_impl(Severity, Pid, Node, Module, Function, File, Line, Metadata, Format, Args, SeverityAsInt, LevelThreshold, TraceFilters, Sink, SinkPid, FormatFun).
  88. %% Should only be called externally from code generated from the parse transform Specifically, it would be level ++ `_unsafe' as in `info_unsafe'.
  89. do_log_unsafe(Severity, Pid, Node, Module, Function, File, Line, Metadata, Format, Args, _Size, SeverityAsInt, LevelThreshold, TraceFilters, Sink, SinkPid) ->
  90. FormatFun = fun() -> unsafe_format(Format, Args) end,
  91. do_log_impl(Severity, Pid, Node, Module, Function, File, Line, Metadata, Format, Args, SeverityAsInt, LevelThreshold, TraceFilters, Sink, SinkPid, FormatFun).
  92. do_log_impl(Severity, Pid, Node, Module, Function, _File, Line, Metadata, Format, Args, SeverityAsInt, LevelThreshold, TraceFilters, Sink, SinkPid, FormatFun) ->
  93. {Destinations, TraceSinkPid} =
  94. case TraceFilters of
  95. [] ->
  96. {[], undefined};
  97. _ ->
  98. {rumUtil:check_traces(Metadata, SeverityAsInt, TraceFilters, []), whereis(?RumTrackSink)}
  99. end,
  100. case (LevelThreshold band SeverityAsInt) /= 0 orelse Destinations /= [] of
  101. true ->
  102. Msg =
  103. case Args of
  104. A when is_list(A) ->
  105. FormatFun();
  106. _ ->
  107. Format
  108. end,
  109. LagerMsg = ?newMsg(Severity, Pid, Node, Module, Function, Line, Metadata, Destinations, rumUtil:nowMs(), Msg),
  110. case rumConfig:get({Sink, async}, false) of
  111. true ->
  112. gen_emm:info_notify(SinkPid, {mWriteLog, LagerMsg});
  113. _ ->
  114. gen_emm:call_notify(SinkPid, {mWriteLog, LagerMsg})
  115. end,
  116. case TraceSinkPid /= undefined of
  117. true ->
  118. gen_emm:info_notify(TraceSinkPid, {mWriteLog, LagerMsg});
  119. false ->
  120. ok
  121. end;
  122. false ->
  123. ok
  124. end.
  125. %% @doc Get lager metadata for current process
  126. -spec md() -> [{atom(), any()}].
  127. md() ->
  128. case erlang:get(?PdMdKey) of
  129. undefined -> [];
  130. MD -> MD
  131. end.
  132. %% @doc Set lager metadata for current process.
  133. %% Will badarg if you don't supply a list of {key, value} tuples keyed by atoms.
  134. -spec md([{atom(), any()}, ...]) -> ok.
  135. md(NewMD) when is_list(NewMD) ->
  136. %% make sure its actually a real proplist
  137. case lists:all(
  138. fun({Key, _Value}) when is_atom(Key) -> true;
  139. (_) -> false
  140. end, NewMD) of
  141. true ->
  142. erlang:put(?PdMdKey, NewMD),
  143. ok;
  144. false ->
  145. erlang:error(badarg)
  146. end;
  147. md(_) ->
  148. erlang:error(badarg).
  149. validate_trace_filters(Filters, Level, Backend) ->
  150. Sink = proplists:get_value(sink, Filters, ?RumDefSink),
  151. {Sink,
  152. rumUtil:validate_trace({
  153. proplists:delete(sink, Filters),
  154. Level,
  155. Backend
  156. })
  157. }.
  158. %% @doc Set the loglevel for a particular backend.
  159. setLogLevel(Handler, Level) when is_atom(Level) ->
  160. setLogLevel(?RumDefSink, Handler, undefined, Level).
  161. %% @doc Set the loglevel for a particular backend that has multiple identifiers
  162. %% (eg. the file backend).
  163. setLogLevel(Handler, Ident, Level) when is_atom(Level) ->
  164. setLogLevel(?RumDefSink, Handler, Ident, Level).
  165. %% @doc Set the loglevel for a particular sink's backend that potentially has
  166. %% multiple identifiers. (Use `undefined' if it doesn't have any.)
  167. setLogLevel(Sink, Handler, Ident, Level) when is_atom(Level) ->
  168. HandlerArg =
  169. case Ident of
  170. undefined -> Handler;
  171. _ -> {Handler, Ident}
  172. end,
  173. Reply = gen_emm:call(Sink, HandlerArg, {mSetLogLevel, Level}, infinity),
  174. updateLogevelCfg(Sink),
  175. Reply.
  176. %% @doc Get the loglevel for a particular backend on the default sink. In the case that the backend has multiple identifiers, the lowest is returned.
  177. getLogLevel(Handler) ->
  178. getLogLevel(?RumDefSink, Handler).
  179. %% @doc Get the loglevel for a particular sink's backend. In the case that the backend
  180. %% has multiple identifiers, the lowest is returned.
  181. getLogLevel(Sink, Handler) ->
  182. case gen_emm:call(Sink, Handler, mGetLogLevel, infinity) of
  183. Mask when is_integer(Mask) ->
  184. case rumUtil:maskToLevels(Mask) of
  185. [] -> none;
  186. Levels -> hd(Levels)
  187. end;
  188. Y -> Y
  189. end.
  190. getLogLevels(Sink) ->
  191. [gen_emm:call(Sink, Handler, mGetLogLevel, infinity) || Handler <- gen_emm:which_epm(Sink)].
  192. %% @doc Set the loghwm for the default sink.
  193. setLogHwm(Handler, Hwm) when is_integer(Hwm) ->
  194. setLogHwm(?RumDefSink, Handler, Hwm).
  195. %% @doc Set the loghwm for a particular backend.
  196. setLogHwm(Sink, Handler, Hwm) when is_integer(Hwm) ->
  197. gen_emm:call(Sink, Handler, {mSetLogHwm, Hwm}, infinity).
  198. %% @doc Set the loghwm (log high water mark) for file backends with multiple identifiers
  199. setLogHwm(Sink, Handler, Ident, Hwm) when is_integer(Hwm) ->
  200. gen_emm:call(Sink, {Handler, Ident}, {mSetLogHwm, Hwm}, infinity).
  201. %% @doc Print stacktrace in human readable form
  202. pr_stacktrace(Stacktrace) ->
  203. Stacktrace1 =
  204. case rumUtil:get_env(reversePrettyStacktrace, true) of
  205. true ->
  206. lists:reverse(Stacktrace);
  207. _ ->
  208. Stacktrace
  209. end,
  210. pr_stacktrace_(Stacktrace1).
  211. pr_stacktrace_(Stacktrace) ->
  212. Indent = "\n ",
  213. lists:foldl(
  214. fun(Entry, Acc) ->
  215. Acc ++ Indent ++ binary_to_list(rumErrLoggerH:formatMfa(Entry))
  216. end,
  217. [],
  218. Stacktrace).
  219. pr_stacktrace(Stacktrace, {Class, Reason}) ->
  220. case rumUtil:get_env(reversePrettyStacktrace, true) of
  221. true ->
  222. lists:flatten(
  223. pr_stacktrace_(lists:reverse(Stacktrace)) ++ "\n" ++ io_lib:format("~s:~p", [Class, Reason]));
  224. _ ->
  225. lists:flatten(
  226. io_lib:format("~s:~p", [Class, Reason]) ++ pr_stacktrace_(Stacktrace))
  227. end.
  228. rotate_sink(Sink) ->
  229. Handlers = rumConfig:global_get(handlers),
  230. RotateHandlers = lists:filtermap(
  231. fun({Handler, _, S}) when S == Sink -> {true, {Handler, Sink}};
  232. (_) -> false
  233. end,
  234. Handlers),
  235. rotate_handlers(RotateHandlers).
  236. rotate_all() ->
  237. rotate_handlers(lists:map(fun({H, _, S}) -> {H, S} end,
  238. rumConfig:global_get(handlers))).
  239. rotate_handlers(Handlers) ->
  240. [rotate_handler(Handler, Sink) || {Handler, Sink} <- Handlers].
  241. rotate_handler(Handler) ->
  242. Handlers = rumConfig:global_get(handlers),
  243. case lists:keyfind(Handler, 1, Handlers) of
  244. {Handler, _, Sink} -> rotate_handler(Handler, Sink);
  245. false -> ok
  246. end.
  247. rotate_handler(Handler, Sink) ->
  248. gen_emm:call(Sink, Handler, mRotate, ?RumRotateTimeout).
  249. trace_file(File, Filter) ->
  250. trace_file(File, Filter, debug, []).
  251. trace_file(File, Filter, Level) when is_atom(Level) ->
  252. trace_file(File, Filter, Level, []);
  253. trace_file(File, Filter, Options) when is_list(Options) ->
  254. trace_file(File, Filter, debug, Options).
  255. trace_file(File, Filter, Level, Options) ->
  256. FileName = rumUtil:parsePath(File),
  257. case validate_trace_filters(Filter, Level, {rumBkdFile, FileName}) of
  258. {Sink, {ok, Trace}} ->
  259. Handlers = rumConfig:global_get(handlers, []),
  260. %% check if this file backend is already installed
  261. Res =
  262. case rumUtil:find_file(FileName, Handlers) of
  263. false ->
  264. %% install the handler
  265. LogFileConfig =
  266. lists:keystore(level, 1,
  267. lists:keystore(file, 1,
  268. Options,
  269. {file, FileName}),
  270. {level, none}),
  271. HandlerInfo =
  272. eRum_app:startHandler(Sink, {rumBkdFile, FileName}, LogFileConfig),
  273. rumConfig:global_set(handlers, [HandlerInfo | Handlers]),
  274. {ok, installed};
  275. {_Watcher, _Handler, Sink} ->
  276. {ok, exists};
  277. {_Watcher, _Handler, _OtherSink} ->
  278. {error, file_in_use}
  279. end,
  280. case Res of
  281. {ok, _} ->
  282. add_trace_to_loglevel_config(Trace, Sink),
  283. {ok, {{rumBkdFile, FileName}, Filter, Level}};
  284. {error, _} = E ->
  285. E
  286. end;
  287. {_Sink, Error} ->
  288. Error
  289. end.
  290. trace_console(Filter) ->
  291. trace_console(Filter, debug).
  292. trace_console(Filter, Level) ->
  293. trace(rumBkdConsole, Filter, Level).
  294. trace(Backend, Filter) ->
  295. trace(Backend, Filter, debug).
  296. trace({rumBkdFile, File}, Filter, Level) ->
  297. trace_file(File, Filter, Level);
  298. trace(Backend, Filter, Level) ->
  299. case validate_trace_filters(Filter, Level, Backend) of
  300. {Sink, {ok, Trace}} ->
  301. add_trace_to_loglevel_config(Trace, Sink),
  302. {ok, {Backend, Filter, Level}};
  303. {_Sink, Error} ->
  304. Error
  305. end.
  306. stop_trace(Backend, Filter, Level) ->
  307. case validate_trace_filters(Filter, Level, Backend) of
  308. {Sink, {ok, Trace}} ->
  309. stop_trace_int(Trace, Sink);
  310. {_Sink, Error} ->
  311. Error
  312. end.
  313. stop_trace({Backend, Filter, Level}) ->
  314. stop_trace(Backend, Filter, Level).
  315. %% Important: validate_trace_filters orders the arguments of
  316. %% trace tuples differently than the way outside callers have
  317. %% the trace tuple.
  318. %%
  319. %% That is to say, outside they are represented as
  320. %% `{Backend, Filter, Level}'
  321. %%
  322. %% and when they come back from validation, they're
  323. %% `{Filter, Level, Backend}'
  324. stop_trace_int({_Filter, _Level, Backend} = Trace, Sink) ->
  325. {Level, Traces} = rumConfig:get({Sink, loglevel}),
  326. NewTraces = lists:delete(Trace, Traces),
  327. _ = rumUtil:trace_filter([element(1, T) || T <- NewTraces]),
  328. %MinLevel = minimum_loglevel(get_loglevels() ++ get_trace_levels(NewTraces)),
  329. rumConfig:set({Sink, loglevel}, {Level, NewTraces}),
  330. case getLogLevel(Sink, Backend) of
  331. none ->
  332. %% check no other traces point here
  333. case lists:keyfind(Backend, 3, NewTraces) of
  334. false ->
  335. gen_emm:delEpm(Sink, Backend, []),
  336. rumConfig:global_set(handlers,
  337. lists:keydelete(Backend, 1,
  338. rumConfig:global_get(handlers)));
  339. _ ->
  340. ok
  341. end;
  342. _ ->
  343. ok
  344. end,
  345. ok.
  346. %% @doc installs a lager trace handler into the target process (using sys:install) at the specified level.
  347. -spec install_trace(pid(), rumAtomLevel()) -> ok.
  348. install_trace(Pid, Level) ->
  349. install_trace(Pid, Level, []).
  350. -spec install_trace(pid(), rumAtomLevel(), [{count, infinity | pos_integer()} | {format_string, string()} | {timeout, timeout()}]) -> ok.
  351. install_trace(Pid, Level, Options) ->
  352. sys:install(Pid, {fun ?MODULE:trace_func/3, trace_state(Pid, Level, Options)}).
  353. %% @doc remove a previously installed lager trace handler from the target process.
  354. -spec remove_trace(pid()) -> ok.
  355. remove_trace(Pid) ->
  356. sys:remove(Pid, fun ?MODULE:trace_func/3).
  357. list_all_sinks() ->
  358. sets:to_list(
  359. lists:foldl(fun({_Watcher, _Handler, Sink}, Set) ->
  360. sets:add_element(Sink, Set)
  361. end,
  362. sets:new(),
  363. rumConfig:global_get(handlers, []))).
  364. clear_traces_by_sink(Sinks) ->
  365. lists:foreach(fun(S) ->
  366. {Level, _Traces} =
  367. rumConfig:get({S, loglevel}),
  368. rumConfig:set({S, loglevel},
  369. {Level, []})
  370. end,
  371. Sinks).
  372. clear_trace_by_destination(ID) ->
  373. Sinks = lists:sort(list_all_sinks()),
  374. Traces = find_traces(Sinks),
  375. [stop_trace_int({Filter, Level, Destination}, Sink) || {Sink, {Filter, Level, Destination}} <- Traces, Destination == ID].
  376. clear_all_traces() ->
  377. Handlers = rumConfig:global_get(handlers, []),
  378. clear_traces_by_sink(list_all_sinks()),
  379. _ = rumUtil:trace_filter(none),
  380. rumConfig:global_set(handlers,
  381. lists:filter(
  382. fun({Handler, _Watcher, Sink}) ->
  383. case getLogLevel(Sink, Handler) of
  384. none ->
  385. gen_emm:delEpm(Sink, Handler, []),
  386. false;
  387. _ ->
  388. true
  389. end
  390. end, Handlers)).
  391. find_traces(Sinks) ->
  392. lists:foldl(fun(S, Acc) ->
  393. {_Level, Traces} = rumConfig:get({S, loglevel}),
  394. Acc ++ lists:map(fun(T) -> {S, T} end, Traces)
  395. end,
  396. [],
  397. Sinks).
  398. status() ->
  399. Handlers = rumConfig:global_get(handlers, []),
  400. Sinks = lists:sort(list_all_sinks()),
  401. Traces = find_traces(Sinks),
  402. TraceCount = case length(Traces) of
  403. 0 -> 1;
  404. N -> N
  405. end,
  406. Status = ["Lager status:\n",
  407. [begin
  408. Level = getLogLevel(Sink, Handler),
  409. get_sink_handler_status(Sink, Handler, Level)
  410. end || {Handler, _Watcher, Sink} <- lists:sort(fun({_, _, S1},
  411. {_, _, S2}) -> S1 =< S2 end,
  412. Handlers)],
  413. "Active Traces:\n",
  414. [begin
  415. LevelName =
  416. case rumUtil:maskToLevels(Level) of
  417. [] -> none;
  418. Levels -> hd(Levels)
  419. end,
  420. io_lib:format("Tracing messages matching ~p (sink ~s) at level ~p to ~p\n",
  421. [Filter, Sink, LevelName, Destination])
  422. end || {Sink, {Filter, Level, Destination}} <- Traces],
  423. [
  424. "Tracing Reductions:\n",
  425. case ?RumDefTracer:info('query') of
  426. {null, false} -> "";
  427. Query -> io_lib:format("~p~n", [Query])
  428. end
  429. ],
  430. [
  431. "Tracing Statistics:\n ",
  432. [begin
  433. [" ", atom_to_list(Table), ": ",
  434. integer_to_list(?RumDefTracer:info(Table) div TraceCount),
  435. "\n"]
  436. end || Table <- [input, output, filter]]
  437. ]],
  438. io:put_chars(Status).
  439. get_sink_handler_status(Sink, Handler, Level) ->
  440. case Handler of
  441. {rumBkdFile, File} ->
  442. io_lib:format("File ~ts (~s) at level ~p\n", [File, Sink, Level]);
  443. rumBkdConsole ->
  444. io_lib:format("Console (~s) at level ~p\n", [Sink, Level]);
  445. _ ->
  446. []
  447. end.
  448. %% @doc Try to convert an atom to a posix error, but fall back on printing the
  449. %% term if its not a valid posix error code.
  450. posix_error(Error) when is_atom(Error) ->
  451. case erl_posix_msg:message(Error) of
  452. "unknown POSIX error" -> atom_to_list(Error);
  453. Message -> Message
  454. end;
  455. posix_error(Error) ->
  456. safe_format_chop("~p", [Error], ?RumDefTruncation).
  457. %% @private
  458. add_trace_to_loglevel_config(Trace, Sink) ->
  459. {MinLevel, Traces} = rumConfig:get({Sink, loglevel}),
  460. case lists:member(Trace, Traces) of
  461. false ->
  462. NewTraces = [Trace | Traces],
  463. _ = rumUtil:trace_filter([element(1, T) || T <- NewTraces]),
  464. rumConfig:set({Sink, loglevel}, {MinLevel, [Trace | Traces]});
  465. _ ->
  466. ok
  467. end.
  468. %% @doc recalculate min log level
  469. updateLogevelCfg(error_logger) ->
  470. %% Not a sink under our control, part of the Erlang logging
  471. %% utility that error_logger_lager_h attaches to
  472. true;
  473. updateLogevelCfg(Sink) ->
  474. {_, Traces} = rumConfig:get({Sink, loglevel}, {ignore_me, []}),
  475. MinLog = minLogLevel(getLogLevels(Sink)),
  476. rumConfig:set({Sink, loglevel}, {MinLog, Traces}).
  477. minLogLevel(Levels) ->
  478. lists:foldl(
  479. fun(Mask, Acc) ->
  480. Mask bor Acc
  481. end, 0, Levels).
  482. %% @doc Print the format string `Fmt' with `Args' safely with a size
  483. %% limit of `Limit'. If the format string is invalid, or not enough
  484. %% arguments are supplied 'FORMAT ERROR' is printed with the offending
  485. %% arguments. The caller is NOT crashed.
  486. safe_format(Fmt, Args, Limit) ->
  487. safe_format(Fmt, Args, Limit, []).
  488. safe_format(Fmt, Args, Limit, Options) ->
  489. try rumTruncIo:format(Fmt, Args, Limit, Options)
  490. catch
  491. _:_ -> rumTruncIo:format("FORMAT ERROR: ~p ~p", [Fmt, Args], Limit)
  492. end.
  493. %% @private
  494. safe_format_chop(Fmt, Args, Limit) ->
  495. safe_format(Fmt, Args, Limit, [{chomp, true}]).
  496. %% @private Print the format string `Fmt' with `Args' without a size limit.
  497. %% This is unsafe because the output of this function is unbounded.
  498. %%
  499. %% Log messages with unbounded size will kill your application dead as
  500. %% OTP mechanisms stuggle to cope with them. So this function is
  501. %% intended <b>only</b> for messages which have a reasonable bounded
  502. %% size before they're formatted.
  503. %%
  504. %% If the format string is invalid or not enough arguments are
  505. %% supplied a 'FORMAT ERROR' message is printed instead with the
  506. %% offending arguments. The caller is NOT crashed.
  507. unsafe_format(Fmt, Args) ->
  508. try io_lib:format(Fmt, Args)
  509. catch
  510. _:_ -> io_lib:format("FORMAT ERROR: ~p ~p", [Fmt, Args])
  511. end.
  512. %% @doc Print a record or a list of records lager found during parse transform
  513. pr(Record, Module) when is_tuple(Record), is_atom(element(1, Record)) ->
  514. pr(Record, Module, []);
  515. pr(List, Module) when is_list(List) ->
  516. pr(List, Module, []);
  517. pr(Record, _) ->
  518. Record.
  519. %% @doc Print a record or a list of records lager found during parse transform
  520. pr(Record, Module, Options) when is_tuple(Record), is_atom(element(1, Record)), is_list(Options) ->
  521. try
  522. case is_record_known(Record, Module) of
  523. false ->
  524. Record;
  525. {RecordName, RecordFields} ->
  526. {'$lager_record', RecordName,
  527. zip(RecordFields, tl(tuple_to_list(Record)), Module, Options, [])}
  528. end
  529. catch
  530. error:undef ->
  531. Record
  532. end;
  533. pr([Head | Tail], Module, Options) when is_list(Options) ->
  534. [pr(Head, Module, Options) | pr(Tail, Module, Options)];
  535. pr(Record, _, _) ->
  536. Record.
  537. zip([FieldName | RecordFields], [FieldValue | Record], Module, Options, ToReturn) when is_list(FieldValue) ->
  538. zip(RecordFields, Record, Module, Options,
  539. [{FieldName, pr(FieldValue, Module, Options)} | ToReturn]);
  540. zip([FieldName | RecordFields], [FieldValue | Record], Module, Options, ToReturn) ->
  541. Compress = lists:member(compress, Options),
  542. case is_tuple(FieldValue) andalso
  543. tuple_size(FieldValue) > 0 andalso
  544. is_atom(element(1, FieldValue)) andalso
  545. is_record_known(FieldValue, Module) of
  546. false when Compress andalso FieldValue =:= undefined ->
  547. zip(RecordFields, Record, Module, Options, ToReturn);
  548. false ->
  549. zip(RecordFields, Record, Module, Options, [{FieldName, FieldValue} | ToReturn]);
  550. _Else ->
  551. F = {FieldName, pr(FieldValue, Module, Options)},
  552. zip(RecordFields, Record, Module, Options, [F | ToReturn])
  553. end;
  554. zip([], [], _Module, _Compress, ToReturn) ->
  555. lists:reverse(ToReturn).
  556. is_record_known(Record, Module) ->
  557. Name = element(1, Record),
  558. Attrs = Module:module_info(attributes),
  559. case lists:keyfind(lager_records, 1, Attrs) of
  560. false -> false;
  561. {lager_records, Records} ->
  562. case lists:keyfind(Name, 1, Records) of
  563. false -> false;
  564. {Name, RecordFields} ->
  565. case (tuple_size(Record) - 1) =:= length(RecordFields) of
  566. false -> false;
  567. true -> {Name, RecordFields}
  568. end
  569. end
  570. end.
  571. %% @private
  572. trace_func(#trace_func_state_v1{pid = Pid, level = Level, format_string = Fmt} = FuncState, Event, ProcState) ->
  573. _ = eRum:log(Level, Pid, Fmt, [Event, ProcState]),
  574. check_timeout(decrement_count(FuncState)).
  575. %% @private
  576. trace_state(Pid, Level, Options) ->
  577. #trace_func_state_v1{pid = Pid,
  578. level = Level,
  579. count = proplists:get_value(count, Options, infinity),
  580. timeout = proplists:get_value(timeout, Options, infinity),
  581. format_string = proplists:get_value(format_string, Options, "TRACE ~p ~p")}.
  582. decrement_count(#trace_func_state_v1{count = infinity} = FuncState) ->
  583. FuncState;
  584. decrement_count(#trace_func_state_v1{count = 1}) ->
  585. %% hit the counter limit
  586. done;
  587. decrement_count(#trace_func_state_v1{count = Count} = FuncState) ->
  588. FuncState#trace_func_state_v1{count = Count - 1}.
  589. check_timeout(#trace_func_state_v1{timeout = infinity} = FuncState) ->
  590. FuncState;
  591. check_timeout(#trace_func_state_v1{timeout = Timeout, started = Started} = FuncState) ->
  592. case (timer:now_diff(os:timestamp(), Started) / 1000) > Timeout of
  593. true ->
  594. done;
  595. false ->
  596. FuncState
  597. end.