From 7bd99bb163496fb4c2167ae09ab2e13bf1922914 Mon Sep 17 00:00:00 2001
From: kercylan98 <kercylan@gmail.com>
Date: Tue, 30 May 2023 10:37:16 +0800
Subject: [PATCH] =?UTF-8?q?=E6=97=A5=E5=BF=97=E6=A8=A1=E5=BC=8F=E8=AE=BE?=
 =?UTF-8?q?=E7=BD=AE=E6=97=A0=E6=95=88=E4=BF=AE=E5=A4=8D=E3=80=81=E5=A2=9E?=
 =?UTF-8?q?=E5=8A=A0=E8=BE=93=E5=87=BA=E7=9B=AE=E5=BD=95=E8=AE=BE=E7=BD=AE?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 utils/log/log.go | 46 ++++++++++++++++++++++++++++++++++------------
 1 file changed, 34 insertions(+), 12 deletions(-)

diff --git a/utils/log/log.go b/utils/log/log.go
index 4e5260d..78128cb 100644
--- a/utils/log/log.go
+++ b/utils/log/log.go
@@ -16,13 +16,19 @@ var (
 	prod   bool
 )
 
+var (
+	logPath string
+)
+
 const (
-	logPath = "./logs"
 	logTime = 7
 )
 
 func init() {
 	logger = newLogger()
+	if prod && len(logPath) == 0 {
+		Warn("Logger", zap.String("Tip", "in production mode, if the log file output directory is not set, only the console will be output"))
+	}
 }
 
 func newLogger() *zap.Logger {
@@ -47,7 +53,7 @@ func newLogger() *zap.Logger {
 	debugLevel := zap.LevelEnablerFunc(func(lvl zapcore.Level) bool {
 		return lvl <= zapcore.FatalLevel
 	})
-	warnLevel := zap.LevelEnablerFunc(func(lvl zapcore.Level) bool {
+	errorLevel := zap.LevelEnablerFunc(func(lvl zapcore.Level) bool {
 		return lvl >= zapcore.ErrorLevel
 	})
 
@@ -58,23 +64,27 @@ func newLogger() *zap.Logger {
 			zapcore.NewCore(encoder, zapcore.AddSync(os.Stdout), debugLevel),
 		)
 	} else {
-		infoWriter := getWriter(fmt.Sprintf("%s/info.log", logPath), logTime)
-		warnWriter := getWriter(fmt.Sprintf("%s/error.log", logPath), logTime)
-		cores = zapcore.NewTee(
-			zapcore.NewCore(encoder, zapcore.AddSync(infoWriter), infoLevel),
-			zapcore.NewCore(encoder, zapcore.AddSync(warnWriter), warnLevel),
-		)
+		if len(logPath) == 0 {
+			cores = zapcore.NewTee(
+				zapcore.NewCore(encoder, zapcore.AddSync(os.Stdout), debugLevel),
+			)
+		} else {
+			infoWriter := getWriter(fmt.Sprintf("%s/info.log", logPath), logTime)
+			errorWriter := getWriter(fmt.Sprintf("%s/error.log", logPath), logTime)
+			cores = zapcore.NewTee(
+				zapcore.NewCore(encoder, zapcore.AddSync(infoWriter), infoLevel),
+				zapcore.NewCore(encoder, zapcore.AddSync(errorWriter), errorLevel),
+				zapcore.NewCore(encoder, zapcore.AddSync(os.Stdout), errorLevel),
+			)
+		}
 	}
 
 	return zap.New(cores, zap.AddCaller(), zap.AddCallerSkip(1))
 }
 
 func getWriter(filename string, times int32) io.Writer {
-	// 生成rotatelogs的Logger 实际生成的文件名 demo.log.YYmmddHH
-	// demo.log是指向最新日志的链接
-	// //保存7天内的日志，每天分割一次日志
 	hook, err := rotatelogs.New(
-		filename+".%Y%m%d", // 没有使用go风格反人类的format格式
+		filename+".%Y%m%d",
 		rotatelogs.WithLinkName(filename),
 		rotatelogs.WithMaxAge(time.Hour*24*7),
 		rotatelogs.WithRotationTime(time.Hour*time.Duration(times)),
@@ -116,4 +126,16 @@ func ErrorWithStack(msg, stack string, fields ...zap.Field) {
 
 func SetProd() {
 	prod = true
+	if logger != nil {
+		_ = logger.Sync()
+		logger = newLogger()
+	}
+}
+
+func SetLogDir(dir string) {
+	logPath = dir
+	if logger != nil {
+		_ = logger.Sync()
+		logger = newLogger()
+	}
 }