|
|
@@ -1,56 +1,73 @@
|
|
|
package utils
|
|
|
|
|
|
import (
|
|
|
- "encoding/json"
|
|
|
- "github.com/beego/beego/v2/adapter/logs"
|
|
|
- "os"
|
|
|
+ "github.com/sirupsen/logrus"
|
|
|
+ "gopkg.in/natefinch/lumberjack.v2"
|
|
|
+ "path"
|
|
|
)
|
|
|
|
|
|
-var FileLog *logs.BeeLogger
|
|
|
-var Binlog *logs.BeeLogger
|
|
|
+var (
|
|
|
+ DefaultLogPath = "./rdlucklog"
|
|
|
+ DefaultLogFile = "eta_task.log"
|
|
|
+ DefaultBinlogPath = "./binlog"
|
|
|
+ DefaultBinlogFile = "binlog.log"
|
|
|
+)
|
|
|
+
|
|
|
+var FileLog = logrus.New()
|
|
|
+var Binlog = logrus.New()
|
|
|
|
|
|
func init() {
|
|
|
- FileLog = logs.NewLogger(1000000)
|
|
|
- FileLog.SetLogger(logs.AdapterFile, `{"filename":"./rdlucklog/eta_task.log"}`)
|
|
|
- //初始化binlog日志
|
|
|
+ logPath := LogPath
|
|
|
+ if logPath == "" {
|
|
|
+ logPath = DefaultLogPath
|
|
|
+ }
|
|
|
+ logFile := LogFile
|
|
|
+ if logFile == "" {
|
|
|
+ logFile = DefaultLogFile
|
|
|
+ }
|
|
|
+ // 打开文件
|
|
|
+ logFileName := path.Join(logPath, logFile)
|
|
|
+ // 使用滚动压缩方式记录日志
|
|
|
+ rolling(FileLog, logFileName)
|
|
|
+ //rolling(bLogFileName)
|
|
|
+ // 设置日志输出JSON格式
|
|
|
+ FileLog.SetFormatter(&logrus.JSONFormatter{})
|
|
|
+ //LogInstance.SetFormatter(&logrus.TextFormatter{})
|
|
|
+ // 设置日志记录级别
|
|
|
+ //FileLog.SetLevel(logrus.DebugLevel)
|
|
|
+
|
|
|
+ //FileLog.Info("abc")
|
|
|
initBinlog()
|
|
|
}
|
|
|
|
|
|
func initBinlog() {
|
|
|
//binlog日志
|
|
|
- binLogDir := `./binlog`
|
|
|
- os.MkdirAll(binLogDir, os.ModePerm)
|
|
|
- Binlog = logs.NewLogger(1000000)
|
|
|
- logConfig := getDefaultLogConfig()
|
|
|
- logConfig.FileName = "./binlog/binlog.log"
|
|
|
- logConfig.MaxLines = 10000000
|
|
|
- logConfig.Rotate = true
|
|
|
- b, _ := json.Marshal(logConfig)
|
|
|
- Binlog.SetLogger(logs.AdapterFile, string(b))
|
|
|
- Binlog.EnableFuncCallDepth(true)
|
|
|
-}
|
|
|
-
|
|
|
-type logConfig struct {
|
|
|
- FileName string `json:"filename" description:"保存的文件名"`
|
|
|
- MaxLines int `json:"maxlines" description:"每个文件保存的最大行数，默认值 1000000"`
|
|
|
- MaxSize int `json:"maxsize" description:"每个文件保存的最大尺寸，默认值是 1 << 28, //256 MB"`
|
|
|
- Daily bool `json:"daily" description:"是否按照每天 logrotate，默认是 true"`
|
|
|
- MaxDays int `json:"maxdays" description:"文件最多保存多少天，默认保存 7 天"`
|
|
|
- Rotate bool `json:"rotate" description:"是否开启 logrotate，默认是 true"`
|
|
|
- Level int `json:"level" description:"日志保存的时候的级别，默认是 Trace 级别"`
|
|
|
- Color bool `json:"color" description:"日志是否输出颜色"`
|
|
|
- //Perm string `json:"perm" description:"日志文件权限"`
|
|
|
+ binlogPath := BinLogPath
|
|
|
+ if binlogPath == "" {
|
|
|
+ binlogPath = DefaultBinlogPath
|
|
|
+ }
|
|
|
+ binlogFile := BinLogFile
|
|
|
+ if binlogFile == "" {
|
|
|
+ binlogFile = DefaultBinlogFile
|
|
|
+ }
|
|
|
+ // 使用滚动压缩方式记录日志
|
|
|
+ logFileName := path.Join(binlogPath, binlogFile)
|
|
|
+ rolling(Binlog, logFileName)
|
|
|
+ //rolling(bLogFileName)
|
|
|
+ // 设置日志输出JSON格式
|
|
|
+ jsonFormat := new(logrus.JSONFormatter)
|
|
|
+ jsonFormat.DisableHTMLEscape = true
|
|
|
+ Binlog.SetFormatter(jsonFormat)
|
|
|
}
|
|
|
|
|
|
-func getDefaultLogConfig() logConfig {
|
|
|
- return logConfig{
|
|
|
- FileName: "",
|
|
|
- MaxLines: 0,
|
|
|
- MaxSize: 1 << 28,
|
|
|
- Daily: true,
|
|
|
- MaxDays: 31, //我就是喜欢31天，咋滴，不喜欢你就自己改-_-!
|
|
|
- Rotate: true,
|
|
|
- Level: logs.LevelTrace,
|
|
|
- //Perm: "",
|
|
|
- }
|
|
|
+// 日志滚动设置
|
|
|
+func rolling(fLog *logrus.Logger, logFile string) {
|
|
|
+ // 设置输出
|
|
|
+ fLog.SetOutput(&lumberjack.Logger{
|
|
|
+ Filename: logFile, //日志文件位置
|
|
|
+ MaxSize: 500, // 单文件最大容量,单位是MB
|
|
|
+ MaxBackups: 3, // 最大保留过期文件个数
|
|
|
+ MaxAge: 7, // 保留过期文件的最大时间间隔,单位是天
|
|
|
+ Compress: true, // 是否需要压缩滚动日志, 使用的 gzip 压缩
|
|
|
+ })
|
|
|
}
|
