From 0a09dc2feeb01358f200eb47872e49b3a69765e6 Mon Sep 17 00:00:00 2001
From: duanhf2012 <6549168@qq.com>
Date: Wed, 11 Dec 2024 17:14:29 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96=E6=97=A5=E5=BF=97?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 log/log.go | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/log/log.go b/log/log.go
index 710c080..da0848f 100644
--- a/log/log.go
+++ b/log/log.go
@@ -23,6 +23,7 @@ type Logger struct {
 	Encoder       zapcore.Encoder
 	LogConfig     *lumberjack.Logger
 	SugaredLogger *zap.SugaredLogger
+	WriteSyncer   zapcore.WriteSyncer
 }
 
 func SetLogger(logger *Logger) {
@@ -104,10 +105,13 @@ func (logger *Logger) Init() {
 		core := zapcore.NewCore(logger.Encoder, zapcore.AddSync(os.Stdout), logger.LogLevel)
 		coreList = append(coreList, core)
 	}
-	
+
 	if logger.LogPath != "" {
-		writeSyncer := zapcore.AddSync(logger.LogConfig)
-		core := zapcore.NewCore(logger.Encoder, writeSyncer, logger.LogLevel)
+		if logger.WriteSyncer == nil {
+			logger.WriteSyncer = zapcore.AddSync(logger.LogConfig)
+		}
+
+		core := zapcore.NewCore(logger.Encoder, logger.WriteSyncer, logger.LogLevel)
 		coreList = append(coreList, core)
 	}