rewrite from lager
Вы не можете выбрать более 25 тем Темы должны начинаться с буквы или цифры, могут содержать дефисы(-) и должны содержать не более 35 символов.

151 строка
5.6 KiB

4 лет назад
4 лет назад
4 лет назад
4 лет назад
4 лет назад
  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. %% 对于特定的接收器事件队列刷新,请使用改选项
  35. {flush_queue, boolean()} |
  36. %% 对于接收器 如果flush_queue为true,则可以设置消息队列长度阈值,在该阈值处将开始丢弃消息。默认阈值为0,
  37. %% 这意味着如果flush_queue为true,则超过高水位标记时将丢弃消息,而不管消息队列的长度如何。:
  38. {flush_threshold, non_neg_integer()} |
  39. {sync_interval, non_neg_integer()} |
  40. {sync_size, non_neg_integer()} |
  41. {sync_on, rumAtomLevel()} |
  42. {check_interval, non_neg_integer()} |
  43. {formatter, atom()} |
  44. {formatter_config, term()}.
  45. %% BkdFile选项默认值
  46. -define(RumDefLogLevel, info).
  47. -define(RumDefRotateSize, 10485760). %% 10mb
  48. -define(RumDefRotateDate, "$D0"). %% midnight
  49. -define(RumDefRotateCnt, 5).
  50. -define(RumDefRotateMod, rumRotatorIns).
  51. -define(RumDefSyncLevel, error).
  52. -define(RumDefSyncInterval, 1000).
  53. -define(RumDefSyncSize, 1024 * 64). %% 64kb
  54. -define(RumDefCheckInterval, 1000).
  55. -define(RumDefCheckHWM, 1000). %% IMY-todo修正该默认值
  56. -define(RumDefFlushQueue, true).
  57. -define(RumDefFlushThreshold, 0). %% IMY-todo修正该默认值
  58. -define(RumDefFormatter, rumFormatter).
  59. -define(RumDefFormatterCfg, []).
  60. %% 默认日志文件选项
  61. -define(RumDefHandler,
  62. [
  63. {rumBkdConsole, [{level, info}]},
  64. {rumBkdFile, [{file, <<"./log/error.log">>}, {level, '>=error'}, {size, 10485760}, {date, "$D0"}, {count, 5}]},
  65. {rumBkdFile, [{file, <<"./log/console.log">>}, {level, '>=debug'}, {size, 10485760}, {date, "$D0"}, {count, 5}]}
  66. ]).
  67. -record(rumShaper, {
  68. id :: any()
  69. %% 每秒我们尝试传递多少消息
  70. , hwm = undefined :: 'undefined' | pos_integer()
  71. %% 这秒内我们收到了多少条消息
  72. , mps = 0 :: non_neg_integer()
  73. %% 当前秒
  74. , lastTime = rumTime:now() :: erlang:timestamp()
  75. %% 此秒内丢弃的消息数
  76. , dropped = 0 :: non_neg_integer()
  77. %% If true, flush notify messages from msg queue at overload
  78. %% 如果为true,则在过载时刷新来自消息队列的通知消息
  79. , flushQueue = true :: boolean()
  80. , flushThreshold = 0 :: integer()
  81. %% timer
  82. , timer = make_ref() :: reference()
  83. %% optional filter fun to avoid counting suppressed messages against HWM totals
  84. %% 可选的过滤器函数,以避免对HWM总数计算抑制消息
  85. , filter = fun(_) -> false end :: fun()
  86. }).
  87. -type rumShaper() :: #rumShaper{}.
  88. -type rumAtomLevel() :: none | debug | info | notice | warning | error | critical | alert | emergency.
  89. -type rumMaskLevel() :: 0..256.
  90. %% 日志等级列表
  91. -define(RumLevels, [debug, info, notice, warning, error, critical, alert, emergency, none]).
  92. %% 使用这些“功能”意味着,出于安全考虑,参数列表不会被截断
  93. -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}]).
  94. -define(RunShouldLog(Sink, Level), (rumUtil:levelToNum(Level) band element(1, rumConfig:get({Sink, loglevel}, {?none, []}))) /= 0).
  95. -define(RunShouldLog(Level), (rumUtil:levelToNum(Level) band element(1, rumConfig:get(loglevel, {?none, []}))) /= 0).
  96. -define(RunNotify(Level, Pid, Format, Args),
  97. gen_emm:info_notify(rumEvent, {mWriteLog, rumMsg:new(io_lib:format(Format, Args), Level, [{pid, Pid}, {line, ?LINE}, {file, ?FILE}, {module, ?MODULE}], [])})).
  98. %%仅供内部使用仅内部非阻塞日志记录调用,当我们仍在启动大型啤酒时尝试进行日志记录(通常为错误)时,会有一些特殊处理。
  99. -ifdef(TEST).
  100. -define(INT_LOG(Level, Format, Args),
  101. case ?RunShouldLog(Level) of
  102. true ->
  103. ?RunNotify(Level, self(), Format, Args);
  104. _ ->
  105. ok
  106. end).
  107. -else.
  108. -define(INT_LOG(Level, Format, Args),
  109. Self = self(),
  110. %%在生成中执行此操作,这样就不会导致从gen_event处理程序调用gen_event:which_handlers的死锁
  111. spawn(fun() ->
  112. case catch (gen_event:which_handlers(rumEvent)) of
  113. X when X == []; X == {'EXIT', noproc}; X == [rumBkdThrottle] ->
  114. %% there's no handlers yet or lager isn't running, try again
  115. %% in half a second.
  116. timer:sleep(500),
  117. ?RunNotify(Level, Self, Format, Args);
  118. _ ->
  119. case ?RunShouldLog(Level) of
  120. true ->
  121. ?RunNotify(Level, Self, Format, Args);
  122. _ ->
  123. ok
  124. end
  125. end
  126. end)).
  127. -endif.