完成基本的创建使用功能

pkg/logger/logger.go

@@ -0,0 +1,54 @@
+package logger
+
+import (
+	"io"
+	"log"
+	"os"
+	"path/filepath"
+	"time"
+
+	rotatelogs "github.com/lestrrat-go/file-rotatelogs"
+	"github.com/rifflock/lfshook"
+	"github.com/sirupsen/logrus"
+)
+
+func Init() {
+	// 添加日志切割输出
+	var hook = NewLfsHook("glogs", time.Hour*12, 6)
+	logrus.AddHook(hook)
+	// 忽略控制台打印
+	logrus.SetOutput(io.Discard)
+	// 展示日志行数
+	logrus.SetReportCaller(true)
+}
+
+// NewLfsHook 日志钩子(日志拦截，并重定向)
+func NewLfsHook(logName string, rotationTime time.Duration, leastDay uint) logrus.Hook {
+	_ = os.Mkdir(logName, os.ModeDir)
+	// 可设置按不同level创建不同的文件名，咱们把6中日志都写到同一个writer中
+	lfsHook := lfshook.NewHook(lfshook.WriterMap{
+		logrus.DebugLevel: NewWriter(filepath.Join(logName, "debug"), rotationTime, leastDay),
+		logrus.InfoLevel:  NewWriter(filepath.Join(logName, "info"), rotationTime, leastDay),
+		logrus.WarnLevel:  NewWriter(filepath.Join(logName, "warn"), rotationTime, leastDay),
+		logrus.ErrorLevel: NewWriter(filepath.Join(logName, "error"), rotationTime, leastDay),
+		logrus.FatalLevel: NewWriter(filepath.Join(logName, "fatal"), rotationTime, leastDay),
+		logrus.PanicLevel: NewWriter(filepath.Join(logName, "panic"), rotationTime, leastDay),
+	}, nil) //&logrus.JSONFormatter{TimestampFormat: "2006-01-02 15:04:05"}
+	return lfsHook
+}
+
+func NewWriter(logName string, rotationTime time.Duration, leastDay uint) io.Writer {
+	writer, err := rotatelogs.New(
+		// 1 日志文件名字
+		logName+".%Y-%m-%d_%H_%M", // _%S
+		// 2 日志周期(默认每86400秒/一天旋转一次)
+		rotatelogs.WithRotationTime(rotationTime),
+		// 3 清除历史 (WithMaxAge和WithRotationCount只能选其一)
+		//rotatelogs.WithMaxAge(time.Hour*24*7), //默认每7天清除下日志文件
+		rotatelogs.WithRotationCount(leastDay), //只保留最近的N个日志文件
+	)
+	if err != nil {
+		log.Panic(err)
+	}
+	return writer
+}