From 9eafd3e6ca3b215fad799fdfab9dc47cd77d5b2f Mon Sep 17 00:00:00 2001
From: RockYang <yangjian102621@gmail.com>
Date: Mon, 31 Jul 2023 08:34:11 +0800
Subject: [PATCH] feat: add env var to set log level

---
 api/handler/admin/config_handler.go |  2 +-
 api/handler/chat_handler.go         |  5 +----
 api/logger/logger.go                | 16 +++++++++++++++-
 3 files changed, 17 insertions(+), 6 deletions(-)

diff --git a/api/handler/admin/config_handler.go b/api/handler/admin/config_handler.go
index 3e36e56d..6cac766b 100644
--- a/api/handler/admin/config_handler.go
+++ b/api/handler/admin/config_handler.go
@@ -57,7 +57,7 @@ func (h *ConfigHandler) Update(c *gin.Context) {
 			resp.ERROR(c, "Failed to update config cache: "+err.Error())
 			return
 		}
-		logger.Infof("Update AppServer's config successfully: %v", config.Config)
+		logger.Debugf("Update AppServer's config successfully: %v", config.Config)
 	}
 
 	resp.SUCCESS(c, config)
diff --git a/api/handler/chat_handler.go b/api/handler/chat_handler.go
index bd839d2c..84f7d185 100644
--- a/api/handler/chat_handler.go
+++ b/api/handler/chat_handler.go
@@ -205,9 +205,7 @@ func (h *ChatHandler) sendMessage(ctx context.Context, session types.ChatSession
 			}
 		}
 
-		if h.App.Debug { // 调试打印聊天上下文
-			logger.Info("聊天上下文：", chatCtx)
-		}
+		logger.Debugf("聊天上下文：%+v", chatCtx)
 	}
 	reqMgs := make([]interface{}, 0)
 	for _, m := range chatCtx {
@@ -312,7 +310,6 @@ func (h *ChatHandler) sendMessage(ctx context.Context, session types.ChatSession
 				logger.Info(functionName)
 				logger.Info(arguments)
 				f := h.App.Functions[functionName]
-				// TODO 调用函数完成任务
 				data, err := f.Invoke(arguments)
 				if err != nil {
 					msg := "调用函数出错：" + err.Error()
diff --git a/api/logger/logger.go b/api/logger/logger.go
index ceb0586b..d744708d 100644
--- a/api/logger/logger.go
+++ b/api/logger/logger.go
@@ -5,6 +5,7 @@ import (
 	"go.uber.org/zap/zapcore"
 	"gopkg.in/natefinch/lumberjack.v2"
 	"os"
+	"strings"
 )
 
 var logger *zap.Logger
@@ -15,7 +16,7 @@ func GetLogger() *zap.SugaredLogger {
 		return sugarLogger
 	}
 
-	logLevel := zap.NewAtomicLevelAt(zapcore.InfoLevel)
+	logLevel := zap.NewAtomicLevelAt(getLogLevel(os.Getenv("LOG_LEVEL")))
 	encoder := getEncoder()
 	writerSyncer := getLogWriter()
 	fileCore := zapcore.NewCore(encoder, writerSyncer, logLevel)
@@ -58,3 +59,16 @@ func getLogWriter() zapcore.WriteSyncer {
 	}
 	return zapcore.AddSync(lumberJackLogger)
 }
+
+func getLogLevel(level string) zapcore.Level {
+	switch strings.ToUpper(level) {
+	case "DEBUG":
+		return zapcore.DebugLevel
+	case "WARN":
+		return zapcore.WarnLevel
+	case "ERROR":
+		return zapcore.ErrorLevel
+	default:
+		return zapcore.InfoLevel
+	}
+}