rewrite from lager
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.

148 lines
5.2 KiB

  1. -include("rumCom.hrl").
  2. %% 应用名字
  3. -define(RumAppName, eRum).
  4. %% 三元表达式
  5. -define(IIF(Cond, Ret1, Ret2), (case Cond of true -> Ret1; _ -> Ret2 end)).
  6. %% 错误日志宏定义
  7. -define(ERR(Format), error_logger:error_msg(Format)).
  8. -define(ERR(Format, Args), error_logger:error_msg(Format, Args)).
  9. -define(Print(Args), io:format("IMY************~p~n", [Args])).
  10. %% 额外进程字典key
  11. -define(PdMdKey, pdRumMd).
  12. -define(RumTrackSink, '__trace_sink').
  13. %% 旋转日志超时时间
  14. -define(RumRotateTimeout, 100000).
  15. %% 部分默认配置值
  16. -define(RumDefTracer, lager_default_tracer).
  17. -define(RumErrLogSink, error_logger_lager_event).
  18. %% rumBkdConsole的选项
  19. -type rumConsoleOpt() ::
  20. {id, atom() | {atom(), atom()}} |
  21. {use_stderr, boolean()} |
  22. {group_leader, false | pid() | atom()} |
  23. {formatter, atom()} |
  24. {formatter_config, list()}.
  25. %% rumBkdFile的选项
  26. -type rumFileOpt() ::
  27. {file, binary()} |
  28. {level, rumAtomLevel() | atom()} |
  29. {size, non_neg_integer()} |
  30. {date, string()} |
  31. {count, non_neg_integer()} |
  32. {rotator, atom()} |
  33. {high_water_mark, non_neg_integer()} |
  34. {flush_queue, boolean()} |
  35. {flush_threshold, non_neg_integer()} |
  36. {sync_interval, non_neg_integer()} |
  37. {sync_size, non_neg_integer()} |
  38. {sync_on, rumAtomLevel()} |
  39. {check_interval, non_neg_integer()} |
  40. {formatter, atom()} |
  41. {formatter_config, term()}.
  42. %% BkdFile选项默认值
  43. -define(RumDefLogLevel, info).
  44. -define(RumDefRotateSize, 10485760). %% 10mb
  45. -define(RumDefRotateDate, "$D0"). %% midnight
  46. -define(RumDefRotateCnt, 5).
  47. -define(RumDefRotateMod, rumRotatorIns).
  48. -define(RumDefSyncLevel, error).
  49. -define(RumDefSyncInterval, 1000).
  50. -define(RumDefSyncSize, 1024 * 64). %% 64kb
  51. -define(RumDefCheckInterval, 1000).
  52. -define(RumDefCheckHWM, 1000). %% IMY-todo修正该默认值
  53. -define(RumDefFlushQueue, true).
  54. -define(RumDefFlushThreshold, 0). %% IMY-todo修正该默认值
  55. -define(RumDefFormatter, rumFormatter).
  56. -define(RumDefFormatterCfg, []).
  57. %% 默认日志文件选项
  58. -define(RumDefHandler,
  59. [
  60. {rumBkdConsole, [{level, info}]},
  61. {rumBkdFile, [{file, <<"./log/error.log">>}, {level, '>=error'}, {size, 10485760}, {date, "$D0"}, {count, 5}]},
  62. {rumBkdFile, [{file, <<"./log/console.log">>}, {level, '>=debug'}, {size, 10485760}, {date, "$D0"}, {count, 5}]}
  63. ]).
  64. -record(rumShaper, {
  65. id :: any()
  66. %% 每秒我们尝试传递多少消息
  67. , hwm = undefined :: 'undefined' | pos_integer()
  68. %% 这秒内我们收到了多少条消息
  69. , mps = 0 :: non_neg_integer()
  70. %% 当前秒
  71. , lastTime = rumTime:now() :: erlang:timestamp()
  72. %% 此秒内丢弃的消息数
  73. , dropped = 0 :: non_neg_integer()
  74. %% If true, flush notify messages from msg queue at overload
  75. %% 如果为true,则在过载时刷新来自消息队列的通知消息
  76. , flushQueue = true :: boolean()
  77. , flushThreshold = 0 :: integer()
  78. %% timer
  79. , timer = make_ref() :: reference()
  80. %% optional filter fun to avoid counting suppressed messages against HWM totals
  81. %% 可选的过滤器函数,以避免对HWM总数计算抑制消息
  82. , filter = fun(_) -> false end :: fun()
  83. }).
  84. -type rumShaper() :: #rumShaper{}.
  85. -type rumAtomLevel() :: none | debug | info | notice | warning | error | critical | alert | emergency.
  86. -type rumMaskLevel() :: 0..256.
  87. %% 日志等级列表
  88. -define(RumLevels, [debug, info, notice, warning, error, critical, alert, emergency, none]).
  89. %% 使用这些“功能”意味着,出于安全考虑,参数列表不会被截断
  90. -define(RumLevelsUnsafe, [{debug_unsafe, debug}, {info_unsafe, info}, {notice_unsafe, notice}, {warning_unsafe, warning}, {error_unsafe, error}, {critical_unsafe, critical}, {alert_unsafe, alert}, {emergency_unsafe, emergency}]).
  91. -define(RunShouldLog(Sink, Level), (rumUtil:levelToNum(Level) band element(1, rumConfig:get({Sink, loglevel}, {?none, []}))) /= 0).
  92. -define(RunShouldLog(Level), (rumUtil:levelToNum(Level) band element(1, rumConfig:get(loglevel, {?none, []}))) /= 0).
  93. -define(RunNotify(Level, Pid, Format, Args),
  94. gen_emm:info_notify(rumEvent, {mWriteLog, rumMsg:new(io_lib:format(Format, Args), Level, [{pid, Pid}, {line, ?LINE}, {file, ?FILE}, {module, ?MODULE}], [])})).
  95. %%仅供内部使用仅内部非阻塞日志记录调用,当我们仍在启动大型啤酒时尝试进行日志记录(通常为错误)时,会有一些特殊处理。
  96. -ifdef(TEST).
  97. -define(INT_LOG(Level, Format, Args),
  98. case ?RunShouldLog(Level) of
  99. true ->
  100. ?RunNotify(Level, self(), Format, Args);
  101. _ ->
  102. ok
  103. end).
  104. -else.
  105. -define(INT_LOG(Level, Format, Args),
  106. Self = self(),
  107. %%在生成中执行此操作,这样就不会导致从gen_event处理程序调用gen_event:which_handlers的死锁
  108. spawn(fun() ->
  109. case catch (gen_event:which_handlers(rumEvent)) of
  110. X when X == []; X == {'EXIT', noproc}; X == [rumBkdThrottle] ->
  111. %% there's no handlers yet or lager isn't running, try again
  112. %% in half a second.
  113. timer:sleep(500),
  114. ?RunNotify(Level, Self, Format, Args);
  115. _ ->
  116. case ?RunShouldLog(Level) of
  117. true ->
  118. ?RunNotify(Level, Self, Format, Args);
  119. _ ->
  120. ok
  121. end
  122. end
  123. end)).
  124. -endif.