logger.go 7.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273
  1. package log
  2. import (
  3. "fmt"
  4. "runtime/debug"
  5. )
  6. type Logger interface {
  7. Info(args ...interface{})
  8. Infof(format string, args ...interface{})
  9. Infow(msg string, keysAndValues ...interface{})
  10. Warn(args ...interface{})
  11. Warnf(format string, args ...interface{})
  12. Warnw(msg string, keysAndValues ...interface{})
  13. Error(args ...interface{})
  14. Errorf(format string, args ...interface{})
  15. Errorw(msg string, keysAndValues ...interface{})
  16. Debug(args ...interface{})
  17. Debugf(format string, args ...interface{})
  18. Debugw(msg string, keysAndValues ...interface{})
  19. Panic(args ...interface{})
  20. Panicf(format string, args ...interface{})
  21. Panicw(msg string, keysAndValues ...interface{})
  22. Fatal(args ...interface{})
  23. Fatalf(format string, args ...interface{})
  24. Fatalw(msg string, keysAndValues ...interface{})
  25. }
  26. // CustomLogger 包装了一个 Logger 实例
  27. type CustomLogger struct {
  28. logger Logger
  29. Close func()
  30. }
  31. var (
  32. logger *CustomLogger
  33. )
  34. func InitCustomLogger(appName string, isDisplayToConsole bool) *CustomLogger {
  35. filePath := fmt.Sprintf("../logs/%s", appName)
  36. log, w := InitLogger(filePath, isDisplayToConsole)
  37. close := func() {
  38. log.Sync()
  39. close(w.entryChan)
  40. w.wg.Wait()
  41. }
  42. l := new(CustomLogger)
  43. l.logger = log
  44. l.Close = close
  45. logger = l
  46. return l
  47. }
  48. func L() *CustomLogger {
  49. return logger
  50. }
  51. // Info 记录一条信息级别的日志
  52. func (c *CustomLogger) Info(args ...interface{}) {
  53. c.logger.Info(args...)
  54. }
  55. // Infof 格式化记录一条信息级别的日志
  56. func (c *CustomLogger) Infof(format string, args ...interface{}) {
  57. c.logger.Infof(format, args...)
  58. }
  59. // Infoln 记录一条信息级别的日志,并在末尾添加换行符
  60. func (c *CustomLogger) Infow(msg string, keysAndValues ...interface{}) {
  61. c.logger.Infow(msg, keysAndValues...)
  62. }
  63. // Warn 记录一条警告级别的日志
  64. func (c *CustomLogger) Warn(args ...interface{}) {
  65. c.logger.Warn(args...)
  66. }
  67. // Warnf 格式化记录一条警告级别的日志
  68. func (c *CustomLogger) Warnf(format string, args ...interface{}) {
  69. c.logger.Warnf(format, args...)
  70. }
  71. // Warnln 记录一条警告级别的日志,并在末尾添加换行符
  72. func (c *CustomLogger) Warnw(msg string, keysAndValues ...interface{}) {
  73. c.logger.Warnw(msg, keysAndValues...)
  74. }
  75. // Error 记录一条错误级别的日志
  76. func (c *CustomLogger) Error(args ...interface{}) {
  77. c.logger.Error(args...)
  78. }
  79. // Errorf 格式化记录一条错误级别的日志
  80. func (c *CustomLogger) Errorf(format string, args ...interface{}) {
  81. c.logger.Errorf(format, args...)
  82. }
  83. // Errorln 记录一条错误级别的日志,并在末尾添加换行符
  84. func (c *CustomLogger) Errorw(msg string, keysAndValues ...interface{}) {
  85. c.logger.Errorw(msg, keysAndValues...)
  86. }
  87. // Debug 记录一条调试级别的日志
  88. func (c *CustomLogger) Debug(args ...interface{}) {
  89. c.logger.Debug(args...)
  90. }
  91. // Debugf 格式化记录一条调试级别的日志
  92. func (c *CustomLogger) Debugf(format string, args ...interface{}) {
  93. c.logger.Debugf(format, args...)
  94. }
  95. // Debugln 记录一条调试级别的日志,并在末尾添加换行符
  96. func (c *CustomLogger) Debugw(msg string, keysAndValues ...interface{}) {
  97. c.logger.Debugw(msg, keysAndValues...)
  98. }
  99. // Panic 记录一条致命级别的日志并触发 panic
  100. func (c *CustomLogger) Panic(args ...interface{}) {
  101. c.logger.Error(args...)
  102. c.logger.Panic(string(debug.Stack()))
  103. }
  104. // Panicf 格式化记录一条致命级别的日志并触发 panic
  105. func (c *CustomLogger) Panicf(format string, args ...interface{}) {
  106. c.logger.Errorf(format, args...)
  107. c.logger.Panic(string(debug.Stack()))
  108. }
  109. // Panicln 记录一条致命级别的日志并触发 panic,并在末尾添加换行符
  110. func (c *CustomLogger) Panicw(msg string, keysAndValues ...interface{}) {
  111. c.logger.Errorw(msg, keysAndValues...)
  112. c.logger.Panic(string(debug.Stack()))
  113. }
  114. // Fatal 记录一条致命级别的日志并调用 os.Exit(1)
  115. func (c *CustomLogger) Fatal(args ...interface{}) {
  116. c.logger.Error(args...)
  117. c.logger.Fatal(string(debug.Stack()))
  118. }
  119. // Fatalf 格式化记录一条致命级别的日志并调用 os.Exit(1)
  120. func (c *CustomLogger) Fatalf(format string, args ...interface{}) {
  121. c.logger.Errorf(format, args...)
  122. c.logger.Fatal(string(debug.Stack()))
  123. }
  124. // Fatalln 记录一条致命级别的日志并调用 os.Exit(1),并在末尾添加换行符
  125. func (c *CustomLogger) Fatalw(msg string, keysAndValues ...interface{}) {
  126. c.logger.Errorw(msg, keysAndValues...)
  127. c.logger.Fatal(string(debug.Stack()))
  128. }
  129. // ////////////////
  130. func Info(args ...interface{}) {
  131. logger.Info(args...)
  132. }
  133. func Infoln(args ...interface{}) {
  134. logger.Info(args...)
  135. }
  136. // Infof 格式化记录一条信息级别的日志
  137. func Infof(format string, args ...interface{}) {
  138. logger.Infof(format, args...)
  139. }
  140. // Infow 记录一条信息级别的日志,并在末尾添加换行符
  141. func Infow(msg string, keysAndValues ...interface{}) {
  142. logger.Infow(msg, keysAndValues...)
  143. }
  144. // Warn 记录一条警告级别的日志
  145. func Warn(args ...interface{}) {
  146. logger.Warn(args...)
  147. }
  148. func Warnln(args ...interface{}) {
  149. logger.Warn(args...)
  150. }
  151. // Warnf 格式化记录一条警告级别的日志
  152. func Warnf(format string, args ...interface{}) {
  153. logger.Warnf(format, args...)
  154. }
  155. // Warnln 记录一条警告级别的日志,并在末尾添加换行符
  156. func Warnw(msg string, keysAndValues ...interface{}) {
  157. logger.Warnw(msg, keysAndValues...)
  158. }
  159. // Error 记录一条错误级别的日志
  160. func Error(args ...interface{}) {
  161. logger.Error(args...)
  162. }
  163. func Errorln(args ...interface{}) {
  164. logger.Error(args...)
  165. }
  166. // Errorf 格式化记录一条错误级别的日志
  167. func Errorf(format string, args ...interface{}) {
  168. logger.Errorf(format, args...)
  169. }
  170. // Errorln 记录一条错误级别的日志,并在末尾添加换行符
  171. func Errorw(msg string, keysAndValues ...interface{}) {
  172. logger.Errorw(msg, keysAndValues...)
  173. }
  174. // Debug 记录一条调试级别的日志
  175. func Debug(args ...interface{}) {
  176. logger.Debug(args...)
  177. }
  178. func Debugln(args ...interface{}) {
  179. logger.Debug(args...)
  180. }
  181. // Debugf 格式化记录一条调试级别的日志
  182. func Debugf(format string, args ...interface{}) {
  183. logger.Debugf(format, args...)
  184. }
  185. // Debugln 记录一条调试级别的日志,并在末尾添加换行符
  186. func Debugw(msg string, keysAndValues ...interface{}) {
  187. logger.Debugw(msg, keysAndValues...)
  188. }
  189. // Panic 记录一条致命级别的日志并触发 panic
  190. func Panic(args ...interface{}) {
  191. logger.Panic(args...)
  192. }
  193. func Panicln(args ...interface{}) {
  194. logger.Panic(args...)
  195. }
  196. // Panicf 格式化记录一条致命级别的日志并触发 panic
  197. func Panicf(format string, args ...interface{}) {
  198. logger.Panicf(format, args...)
  199. }
  200. // Panicln 记录一条致命级别的日志并触发 panic,并在末尾添加换行符
  201. func Panicw(msg string, keysAndValues ...interface{}) {
  202. logger.Panicw(msg, keysAndValues...)
  203. }
  204. // Fatal 记录一条致命级别的日志并调用 os.Exit(1)
  205. func Fatal(args ...interface{}) {
  206. logger.Fatal(args...)
  207. }
  208. func Fatalln(args ...interface{}) {
  209. logger.Fatal(args...)
  210. }
  211. // Fatalf 格式化记录一条致命级别的日志并调用 os.Exit(1)
  212. func Fatalf(format string, args ...interface{}) {
  213. logger.Fatalf(format, args...)
  214. }
  215. // Fatalln 记录一条致命级别的日志并调用 os.Exit(1),并在末尾添加换行符
  216. func Fatalw(msg string, keysAndValues ...interface{}) {
  217. logger.Fatalw(msg, keysAndValues...)
  218. }