From 4b754a0164621911f1139dc06a095115230afb92 Mon Sep 17 00:00:00 2001
From: cc <cc@chaochao.email>
Date: Sat, 20 Jul 2024 07:56:58 +0000
Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0=E4=BA=86=E6=97=A5=E5=BF=97?=
 =?UTF-8?q?=E8=AE=B0=E5=BD=95=E7=BA=A7=E5=88=AB=E9=85=8D=E7=BD=AE?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 log/zaplog/config.go | 11 +++++++++++
 log/zaplog/zaplog.go |  2 +-
 2 files changed, 12 insertions(+), 1 deletion(-)

diff --git a/log/zaplog/config.go b/log/zaplog/config.go
index b4e202c..6c7896e 100644
--- a/log/zaplog/config.go
+++ b/log/zaplog/config.go
@@ -1,5 +1,15 @@
 package zaplog
 
+const (
+	Debug int8 = iota - 1
+	Info
+	Warn
+	Error
+	DPanic
+	Panic
+	Fatal
+)
+
 type LogFileConfig struct {
 	Filename    string // 日志文件存放目录及名称，如果文件夹不存在会自动创建
 	FileSize_MB int    // 文件大小限制,单位MB
@@ -9,6 +19,7 @@ type LogFileConfig struct {
 
 type ZapConfig struct {
 	JsonFormat bool          // 是否使用json格式,默认false
+	Mode       int8          // 日志启用级别
 	DebugLog   LogFileConfig // Debug日志配置
 	InfoLog    LogFileConfig // Info日志配置
 	WarnLog    LogFileConfig // Warn日志配置
diff --git a/log/zaplog/zaplog.go b/log/zaplog/zaplog.go
index d8639cf..73dec36 100644
--- a/log/zaplog/zaplog.go
+++ b/log/zaplog/zaplog.go
@@ -129,5 +129,5 @@ func New(config ...ZapConfig) *zap.SugaredLogger {
 	HighFileCore := zapcore.NewCore(encoder, zapcore.NewMultiWriteSyncer(HighFileWriteSyncer, zapcore.AddSync(os.Stdout)), HighPriority)
 	coreArr = append(coreArr, HighFileCore)
 
-	return zap.New(zapcore.NewTee(coreArr...)).Sugar()
+	return zap.New(zapcore.NewTee(coreArr...), zap.IncreaseLevel(zapcore.Level(cfg.Mode))).Sugar()
 }