新增zap日志库的支持,并提供Recovery的中间件

server/core/zap.go

@@ -0,0 +1,67 @@
+package core
+
+import (
+	"fmt"
+	"gin-vue-admin/global"
+	"gin-vue-admin/utils"
+	zaprotatelogs "github.com/lestrrat-go/file-rotatelogs"
+	"go.uber.org/zap"
+	"go.uber.org/zap/zapcore"
+	"os"
+	"time"
+)
+
+const (
+	zapLogDir      = "log"
+	zapLogSoftLink = "latest_log"
+	zapModule      = "gin-vue-admin"
+)
+
+func init() {
+	if global.GVA_CONFIG.Zap.File == "" {
+		global.GVA_CONFIG.Zap.File = "DEBUG"
+	}
+	if ok, _ := utils.PathExists(zapLogDir); !ok {
+		// directory not exist
+		fmt.Println("create log directory")
+		_ = os.Mkdir(zapLogDir, os.ModePerm)
+	}
+	var l = new(zapcore.Level)
+	writeSyncer, err := getWriteSyncer()
+	if err != nil {
+		fmt.Printf("Get Write Syncer Failed err:%v", err.Error())
+		return
+	}
+	encoder := getEncoderConfig()
+	if err := l.UnmarshalText([]byte(global.GVA_CONFIG.Zap.Level)); err != nil {
+		fmt.Printf("Unmarshal Level Failed err:%v", err.Error())
+		return
+	}
+	core := zapcore.NewCore(encoder, writeSyncer, l)
+	global.GVA_ZAP = zap.New(core, zap.AddCaller())
+}
+
+// getWriteSyncer zap logger中加入file-rotatelogs
+func getWriteSyncer() (zapcore.WriteSyncer, error) {
+	fileWriter, err := zaprotatelogs.New(
+		zapLogDir + string(os.PathSeparator) + "%Y-%m-%d-%H-%M.log",
+		zaprotatelogs.WithLinkName(zapLogSoftLink),
+		zaprotatelogs.WithMaxAge(7*24*time.Hour),
+		zaprotatelogs.WithRotationTime(24*time.Hour),
+	)
+	return zapcore.AddSync(fileWriter), err
+}
+
+// getEncoderConfig 获取zapcore.Encoder
+func getEncoderConfig() zapcore.Encoder {
+	encoderConfig := zap.NewProductionEncoderConfig()
+	encoderConfig.EncodeTime = zapcore.ISO8601TimeEncoder
+	encoderConfig.TimeKey = "time"
+	encoderConfig.EncodeLevel = zapcore.CapitalLevelEncoder
+	encoderConfig.EncodeDuration = zapcore.SecondsDurationEncoder
+	encoderConfig.EncodeCaller = zapcore.ShortCallerEncoder
+	return zapcore.NewConsoleEncoder(encoderConfig)
+}
+
+
+