log.go 4.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136
  1. package core
  2. import (
  3. "fmt"
  4. "github.com/gin-gonic/gin"
  5. rotatelogs "github.com/lestrrat-go/file-rotatelogs"
  6. oplogging "github.com/op/go-logging"
  7. "hongze/fms_api/config"
  8. "hongze/fms_api/global"
  9. "hongze/fms_api/utils"
  10. "io"
  11. "os"
  12. "strings"
  13. "time"
  14. )
  15. const (
  16. Module = "hongze_yb"
  17. )
  18. var (
  19. defaultFormatter = ` %{time:2006/01/02 - 15:04:05.000} %{longfile} %{color:bold}▶ [%{level:.6s}] %{message}%{color:reset}`
  20. )
  21. // 初始化日志
  22. func init() {
  23. logConfig := global.CONFIG.Log
  24. logger := oplogging.MustGetLogger(Module)
  25. var backends []oplogging.Backend
  26. //注册输出
  27. registerStdout(logConfig, &backends)
  28. //注册框架输出(日志文件)
  29. fileWriter := registerFile(logConfig, &backends, logConfig.LogDirPath, logConfig.LogSoftLink)
  30. if fileWriter != nil {
  31. if global.CONFIG.Serve.RunMode == "debug" {
  32. gin.DefaultWriter = io.MultiWriter(fileWriter, os.Stdout)
  33. } else {
  34. gin.DefaultWriter = io.MultiWriter(fileWriter)
  35. }
  36. }
  37. oplogging.SetBackend(backends...)
  38. global.LOG = logger
  39. //初始化mysql数据库日志
  40. initMysqlLog()
  41. }
  42. // initMysqlLog 初始化mysql数据库日志
  43. func initMysqlLog() {
  44. logConfig := global.CONFIG.Log
  45. var backends []oplogging.Backend
  46. //注册输出
  47. registerStdout(logConfig, &backends)
  48. //注册框架输出(日志文件)
  49. fileWriter := registerFile(logConfig, &backends, logConfig.BinlogDirPath, logConfig.BinlogSoftLink)
  50. global.MYSQL_LOG = fileWriter
  51. }
  52. func registerStdout(c config.Log, backends *[]oplogging.Backend) {
  53. if c.Stdout != "" {
  54. level, err := oplogging.LogLevel(c.Stdout)
  55. if err != nil {
  56. fmt.Println(err)
  57. }
  58. *backends = append(*backends, createBackend(os.Stdout, c, level))
  59. }
  60. }
  61. // registerFile 注册文件日志
  62. func registerFile(c config.Log, backends *[]oplogging.Backend, logDir, logSoftLink string) io.Writer {
  63. if c.FileStdout != "" {
  64. if ok, _ := utils.PathExists(logDir); !ok {
  65. // directory not exist
  66. fmt.Println("create log directory")
  67. _ = os.Mkdir(logDir, os.ModePerm)
  68. }
  69. //日志保留时间
  70. saveMaxTime := time.Duration(global.CONFIG.Log.SaveMaxDay) * 24 * time.Hour
  71. //日志切割时间(每隔多久切割一次)
  72. cuttingTime := time.Duration(global.CONFIG.Log.CuttingDay) * 24 * time.Hour
  73. //使用rotatelogs
  74. fileWriter, err := rotatelogs.New(
  75. logDir+string(os.PathSeparator)+"%Y-%m-%d-%H-%M.log",
  76. // 生成软链,指向最新日志文件
  77. rotatelogs.WithLinkName(logSoftLink),
  78. // 以下两个配置不能同时设置,一个是保留天数,一个是保留分拣份数
  79. rotatelogs.WithMaxAge(saveMaxTime), //日志保留多久,最小分钟为单位
  80. //rotatelogs.WithRotationCount(5), //number 默认7份 大于7份 或到了清理时间 开始清理
  81. // time period of log file switching
  82. rotatelogs.WithRotationTime(cuttingTime), //rotate 最小为1分钟轮询。默认60s 低于1分钟就按1分钟来
  83. )
  84. if err != nil {
  85. fmt.Println(err)
  86. }
  87. level, err := oplogging.LogLevel(c.FileStdout)
  88. if err != nil {
  89. fmt.Println(err)
  90. }
  91. *backends = append(*backends, createBackend(fileWriter, c, level))
  92. return fileWriter
  93. }
  94. return nil
  95. }
  96. // createBackend 创建一个新的日志记录器
  97. func createBackend(w io.Writer, c config.Log, level oplogging.Level) oplogging.Backend {
  98. backend := oplogging.NewLogBackend(w, c.Prefix, 0)
  99. stdoutWriter := false
  100. if w == os.Stdout {
  101. stdoutWriter = true
  102. }
  103. format := getLogFormatter(c, stdoutWriter)
  104. backendLeveled := oplogging.AddModuleLevel(oplogging.NewBackendFormatter(backend, format))
  105. backendLeveled.SetLevel(level, Module)
  106. return backendLeveled
  107. }
  108. func getLogFormatter(c config.Log, stdoutWriter bool) oplogging.Formatter {
  109. pattern := defaultFormatter
  110. if !stdoutWriter {
  111. // Color is only required for console output
  112. // Other writers don't need %{color} tag
  113. pattern = strings.Replace(pattern, "%{color:bold}", "", -1)
  114. pattern = strings.Replace(pattern, "%{color:reset}", "", -1)
  115. }
  116. if !c.LogFile {
  117. // Remove %{logfile} tag
  118. pattern = strings.Replace(pattern, "%{longfile}", "", -1)
  119. }
  120. return oplogging.MustStringFormatter(pattern)
  121. }