From 6b91dd8f3d443e6ff2bece9ff9997c0731a7ff69 Mon Sep 17 00:00:00 2001 From: ryan Date: Fri, 13 Mar 2026 12:25:51 +0800 Subject: [PATCH] =?UTF-8?q?[=E4=BF=AE=E5=A4=8D]=20GIN=E6=97=A5=E5=BF=97?= =?UTF-8?q?=E6=89=93=E5=8D=B0=E7=AD=89=E7=BA=A7=E7=BB=9F=E4=B8=80?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- README.md | 2 +- atsf_server/common/logger.go | 23 ++++++++++++++++++----- 2 files changed, 19 insertions(+), 6 deletions(-) diff --git a/README.md b/README.md index e4409559..2f9104f3 100644 --- a/README.md +++ b/README.md @@ -108,7 +108,6 @@ Origin services: atsflare: image: ghcr.io/rain-kl/atsflare:latest - container_name: atsflare restart: unless-stopped ports: - "3000:3000" @@ -116,6 +115,7 @@ services: SESSION_SECRET: replace-with-random-string SQLITE_PATH: /data/atsflare.db GIN_MODE: release + LOG_LEVEL: info PORT: "3000" volumes: - atsflare-data:/data diff --git a/atsf_server/common/logger.go b/atsf_server/common/logger.go index 5d326a22..09bf3fb7 100644 --- a/atsf_server/common/logger.go +++ b/atsf_server/common/logger.go @@ -22,6 +22,17 @@ const ( var currentLogLevel = logLevelInfo var currentLogLevelName = "info" +var commonLogWriter io.Writer = os.Stdout +var errorLogWriter io.Writer = os.Stderr + +func configureGinWriters() { + if shouldLog(logLevelDebug) { + gin.DefaultWriter = commonLogWriter + } else { + gin.DefaultWriter = io.Discard + } + gin.DefaultErrorWriter = errorLogWriter +} func SetLogLevel(level string) { normalized := strings.TrimSpace(strings.ToLower(level)) @@ -39,6 +50,7 @@ func SetLogLevel(level string) { currentLogLevel = logLevelInfo currentLogLevelName = "info" } + configureGinWriters() } func GetLogLevel() string { @@ -61,9 +73,10 @@ func SetupGinLog() { if err != nil { log.Fatal("failed to open log file") } - gin.DefaultWriter = io.MultiWriter(os.Stdout, commonFd) - gin.DefaultErrorWriter = io.MultiWriter(os.Stderr, errorFd) + commonLogWriter = io.MultiWriter(os.Stdout, commonFd) + errorLogWriter = io.MultiWriter(os.Stderr, errorFd) } + configureGinWriters() } func SysLog(s string) { @@ -71,7 +84,7 @@ func SysLog(s string) { return } t := time.Now() - _, _ = fmt.Fprintf(gin.DefaultWriter, "[SYS] %v | %s \n", t.Format("2006/01/02 - 15:04:05"), s) + _, _ = fmt.Fprintf(commonLogWriter, "[SYS] %v | %s \n", t.Format("2006/01/02 - 15:04:05"), s) } func SysError(s string) { @@ -79,11 +92,11 @@ func SysError(s string) { return } t := time.Now() - _, _ = fmt.Fprintf(gin.DefaultErrorWriter, "[SYS] %v | %s \n", t.Format("2006/01/02 - 15:04:05"), s) + _, _ = fmt.Fprintf(errorLogWriter, "[SYS] %v | %s \n", t.Format("2006/01/02 - 15:04:05"), s) } func FatalLog(v ...any) { t := time.Now() - _, _ = fmt.Fprintf(gin.DefaultErrorWriter, "[FATAL] %v | %v \n", t.Format("2006/01/02 - 15:04:05"), v) + _, _ = fmt.Fprintf(errorLogWriter, "[FATAL] %v | %v \n", t.Format("2006/01/02 - 15:04:05"), v) os.Exit(1) }