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.

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