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.

165 line
6.2 KiB

4 年之前
4 年之前
4 年之前
4 年之前
4 年之前
4 年之前
4 年之前
4 年之前
4 年之前
4 年之前
4 年之前
4 年之前
4 年之前
4 年之前
4 年之前
4 年之前
4 年之前
4 年之前
4 年之前
4 年之前
4 年之前
4 年之前
  1. %% 三元表达式
  2. -define(IIF(Cond, Ret1, Ret2), (case Cond of true -> Ret1; _ -> Ret2 end)).
  3. %% 应用名字
  4. -define(RumAppName, eRum).
  5. %% 部分默认配置值
  6. -define(RumDefTruncation, 4096).
  7. -define(RumDefTracer, lager_default_tracer).
  8. -define(RumDefSink, rumEvent).
  9. -define(RumErrLogSink, error_logger_lager_event).
  10. %% 进程字典Key
  11. -define(RumPdLogNames, rumPdLogNames).
  12. -define(RumBackendThrottle, rumBackendThrottle).
  13. %% 默认日志文件选项
  14. -define(DEFAULT_HANDLER_CONF,
  15. [
  16. {lager_console_backend, [{level, info}]},
  17. {lager_file_backend, [{file, "error.log"}, {level, error}, {size, 10485760}, {date, "$D0"}, {count, 5}]},
  18. {lager_file_backend, [{file, "console.log"}, {level, info}, {size, 10485760}, {date, "$D0"}, {count, 5}]}
  19. ]).
  20. %% 获取元数据的宏定义
  21. -define(RumMetadata(Extras), [
  22. {severity, info},
  23. {pid, self()},
  24. {node, node()},
  25. {module, ?MODULE},
  26. {function, ?FUNCTION_NAME},
  27. {function_arity, ?FUNCTION_ARITY},
  28. {file, ?FILE},
  29. {line, ?LINE} | Extras
  30. ]).
  31. -type rumLevel() :: none | debug | info | notice | warning | error | critical | alert | emergency.
  32. -type rumLevelNum() :: 0..256.
  33. %% 日志等级列表
  34. -define(RumLevels, [debug, info, notice, warning, error, critical, alert, emergency, none]).
  35. %% 使用这些“功能”意味着,出于安全考虑,参数列表不会被截断
  36. -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}]).
  37. %% 日志等级定义
  38. -define(DEBUG, 128).
  39. -define(INFO, 64).
  40. -define(NOTICE, 32).
  41. -define(WARNING, 16).
  42. -define(ERROR, 8).
  43. -define(CRITICAL, 4).
  44. -define(ALERT, 2).
  45. -define(EMERGENCY, 1).
  46. -define(LOG_NONE, 0).
  47. -define(RunShouldLog(Sink, Level), (rumUtil:levelToNum(Level) band element(1, rumConfig:get({Sink, loglevel}, {?LOG_NONE, []}))) /= 0).
  48. -define(RunShouldLog(Level), (rumUtil:levelToNum(Level) band element(1, rumConfig:get(loglevel, {?LOG_NONE, []}))) /= 0).
  49. -define(RunNotify(Level, Pid, Format, Args),
  50. gen_event:notify(rumEvent, {log, rumMsg:new(io_lib:format(Format, Args), Level, [{pid, Pid}, {line, ?LINE}, {file, ?FILE}, {module, ?MODULE}], [])})).
  51. %%仅供内部使用仅内部非阻塞日志记录调用,当我们仍在启动大型啤酒时尝试进行日志记录(通常为错误)时,会有一些特殊处理。
  52. -ifdef(TEST).
  53. -define(INT_LOG(Level, Format, Args),
  54. case ?RunShouldLog(Level) of
  55. true ->
  56. ?RunNotify(Level, self(), Format, Args);
  57. _ ->
  58. ok
  59. end).
  60. -else.
  61. -define(INT_LOG(Level, Format, Args),
  62. Self = self(),
  63. %% do this in a spawn so we don't cause a deadlock calling gen_event:which_handlers
  64. %% from a gen_event handler
  65. spawn(fun() ->
  66. case catch (gen_event:which_handlers(rumEvent)) of
  67. X when X == []; X == {'EXIT', noproc}; X == [lager_backend_throttle] ->
  68. %% there's no handlers yet or lager isn't running, try again
  69. %% in half a second.
  70. timer:sleep(500),
  71. ?RunNotify(Level, Self, Format, Args);
  72. _ ->
  73. case ?RunShouldLog(Level) of
  74. true ->
  75. ?RunNotify(Level, Self, Format, Args);
  76. _ ->
  77. ok
  78. end
  79. end
  80. end)).
  81. -endif.
  82. -define(rumLog(Severity, Format, Args, Safety),
  83. ?rumLog(?RumDefSink, Severity, ?RumMetadata(eRum:md()), Format, Args, ?RumDefTruncation, Safety)).
  84. -define(rumLog(Severity, Metadata, Format, Args, Safety),
  85. ?rumLog(?RumDefSink, Severity, ?RumMetadata(Metadata ++ eRum:md()), Format, Args, ?RumDefTruncation, Safety)).
  86. -define(rumLog(Sink, Severity, Metadata, Format, Args, Size, Safety),
  87. _ = eRum:dispatch_log(Sink, Severity, Metadata, Format, Args, Size, Safety)).
  88. -define(rumDebug(Format, Args), ?rumLog(debug, Format, Args, safe)).
  89. -define(rumDebug(Metadata, Format, Args), ?rumLog(debug, Metadata, Format, Args, safe)).
  90. -define(rumInfo(Format, Args), ?rumLog(info, Format, Args, safe)).
  91. -define(rumInfo(Metadata, Format, Args), ?rumLog(info, Metadata, Format, Args, safe)).
  92. -define(rumNotice(Format, Args), ?rumLog(notice, Format, Args, safe)).
  93. -define(rumNotice(Metadata, Format, Args), ?rumLog(notice, Metadata, Format, Args, safe)).
  94. -define(rumWarning(Format, Args), ?rumLog(warning, Format, Args, safe)).
  95. -define(rumWarning(Metadata, Format, Args), ?rumLog(warning, Metadata, Format, Args, safe)).
  96. -define(rumError(Format, Args), ?rumLog(error, Format, Args, safe)).
  97. -define(rumError(Metadata, Format, Args), ?rumLog(error, Metadata, Format, Args, safe)).
  98. -define(rumCritical(Format, Args), ?rumLog(critical, Format, Args, safe)).
  99. -define(rumCritical(Metadata, Format, Args), ?rumLog(critical, Metadata, Format, Args, safe)).
  100. -define(rumAlert(Format, Args), ?rumLog(alert, Format, Args, safe)).
  101. -define(rumAlert(Metadata, Format, Args), ?rumLog(alert, Metadata, Format, Args, safe)).
  102. -define(rumEmergency(Format, Args), ?rumLog(emergency, Format, Args, safe)).
  103. -define(rumEmergency(Metadata, Format, Args), ?rumLog(emergency, Metadata, Format, Args, safe)).
  104. -define(rumNone(Format, Args), ?rumLog(none, Format, Args, safe)).
  105. -define(rumNone(Metadata, Format, Args), ?rumLog(none, Metadata, Format, Args, safe)).
  106. -record(rumShaper, {
  107. id :: any()
  108. %% 每秒我们尝试传递多少消息
  109. , hwm = undefined :: 'undefined' | pos_integer()
  110. %% 这秒内我们收到了多少条消息
  111. , mps = 0 :: non_neg_integer()
  112. %% 当前秒
  113. , lastTime = os:timestamp() :: erlang:timestamp()
  114. %% 此秒内丢弃的消息数
  115. , dropped = 0 :: non_neg_integer()
  116. %% If true, flush notify messages from msg queue at overload
  117. %% 如果为true,则在过载时刷新来自消息队列的通知消息
  118. , flushQueue = true :: boolean()
  119. , flushThreshold = 0 :: integer()
  120. %% timer
  121. , timer = make_ref() :: reference()
  122. %% optional filter fun to avoid counting suppressed messages against HWM totals
  123. %% 可选的过滤器函数,以避免对HWM总数计算抑制消息
  124. , filter = fun(_) -> false end :: fun()
  125. }).
  126. -record(rumMsg, {
  127. destinations :: list(),
  128. metadata :: [tuple()],
  129. severity :: rumLevel(),
  130. datetime :: binary(),
  131. timestamp :: non_neg_integer(),
  132. message :: list()
  133. }).
  134. -type rumShaper() :: #rumShaper{}.