浏览代码

添加日志分割

yjp 2 年之前
父节点
当前提交
8f2bb1d09a
共有 1 个文件被更改,包括 18 次插入26 次删除
  1. 18 26
      fslog_instance.go

+ 18 - 26
fslog_instance.go

@@ -7,7 +7,6 @@ import (
 	"gopkg.in/natefinch/lumberjack.v2"
 	"os"
 	"path/filepath"
-	"sync"
 )
 
 const (
@@ -35,35 +34,32 @@ func chooseLoggerLever() zapcore.Level {
 }
 
 var logger *zap.Logger
-var initLoggerOnce sync.Once
 
 const logDir = "logs"
 
-func lazyInitialize() {
-	initLoggerOnce.Do(func() {
-		if !pathExists(logDir) {
-			err := os.MkdirAll(logDir, os.ModeDir|os.ModePerm)
-			if err != nil {
-				panic(err)
-			}
+func init() {
+	if !pathExists(logDir) {
+		err := os.MkdirAll(logDir, os.ModeDir|os.ModePerm)
+		if err != nil {
+			panic(err)
 		}
+	}
 
-		logFilePath := filepath.Join(logDir, "log")
-		lumberJackLogger := &lumberjack.Logger{
-			Filename: logFilePath,
-			MaxAge:   10,
-		}
+	logFilePath := filepath.Join(logDir, "log")
+	lumberJackLogger := &lumberjack.Logger{
+		Filename: logFilePath,
+		MaxAge:   10,
+	}
 
-		developmentEncoderConfig := zap.NewDevelopmentEncoderConfig()
+	developmentEncoderConfig := zap.NewDevelopmentEncoderConfig()
 
-		encoder := zapcore.NewConsoleEncoder(developmentEncoderConfig)
-		core := zapcore.NewTee(
-			zapcore.NewCore(encoder, zapcore.AddSync(lumberJackLogger), chooseLoggerLever()),
-			zapcore.NewCore(encoder, zapcore.AddSync(os.Stdout), chooseLoggerLever()),
-		)
+	encoder := zapcore.NewConsoleEncoder(developmentEncoderConfig)
+	core := zapcore.NewTee(
+		zapcore.NewCore(encoder, zapcore.AddSync(lumberJackLogger), chooseLoggerLever()),
+		zapcore.NewCore(encoder, zapcore.AddSync(os.Stdout), chooseLoggerLever()),
+	)
 
-		logger = zap.New(core, zap.AddCaller(), zap.AddCallerSkip(1))
-	})
+	logger = zap.New(core, zap.AddCaller(), zap.AddCallerSkip(1))
 }
 
 func GetLogger() *zap.Logger {
@@ -71,23 +67,19 @@ func GetLogger() *zap.Logger {
 }
 
 func Info(message string) {
-	lazyInitialize()
 	logger.Info(message)
 }
 
 func Error(err error) {
-	lazyInitialize()
 	message := fmt.Sprintf("\n%+v\n", err)
 	logger.Error(message)
 }
 
 func Debug(message string) {
-	lazyInitialize()
 	logger.Debug(message)
 }
 
 func Warn(message string) {
-	lazyInitialize()
 	logger.Warn(message)
 }