From 8941f625c87178be5cc279a9fc2b03855c5e3bd5 Mon Sep 17 00:00:00 2001
From: yyc12345 <yyc12321@outlook.com>
Date: Fri, 17 Jul 2026 12:12:14 +0800
Subject: [PATCH] fix: fix logger style err and gin warning

---
 backend/logger/logger.go | 53 ++++++++++++++++++++++++++++++++++------
 backend/server/server.go | 15 ++++++++++++
 2 files changed, 60 insertions(+), 8 deletions(-)

diff --git a/backend/logger/logger.go b/backend/logger/logger.go
index c3985a1..831909d 100644
--- a/backend/logger/logger.go
+++ b/backend/logger/logger.go
@@ -1,6 +1,8 @@
 package logger
 
 import (
+	"context"
+	"io"
 	"log/slog"
 	"os"
 )
@@ -30,13 +32,48 @@ func toSlogLevel(level LoggerLevel) slog.Level {
 	}
 }
 
-// New creates a *slog.Logger that writes to os.Stderr via slog's default
-// TextHandler, with its verbosity driven by the given LoggerLevel. This
-// replaces the legacy module-level singleton LOGGER; callers own the returned
-// logger and thread it through to sub-modules.
+// New creates a *slog.Logger that writes to os.Stderr in the bracketed format
+// "[<time>] [<LEVEL>] <message> [key=value ...]". Its verbosity is driven by the
+// given LoggerLevel. This replaces the legacy module-level singleton LOGGER;
+// callers own the returned logger and thread it through to sub-modules.
 func New(level LoggerLevel) *slog.Logger {
-	handler := slog.NewTextHandler(os.Stderr, &slog.HandlerOptions{
-		Level: toSlogLevel(level),
-	})
-	return slog.New(handler)
+	return slog.New(&bracketHandler{w: os.Stderr, level: toSlogLevel(level)})
 }
+
+// bracketHandler is a custom slog.Handler emitting the bracketed format
+// "[<time>] [<LEVEL>] <message>" followed by any attributes as "key=value",
+// matching the legacy "[LEVEL] message" style with an added timestamp.
+type bracketHandler struct {
+	w     io.Writer
+	level slog.Level
+}
+
+func (h *bracketHandler) Enabled(_ context.Context, l slog.Level) bool {
+	return l >= h.level
+}
+
+func (h *bracketHandler) Handle(_ context.Context, r slog.Record) error {
+	var b []byte
+	b = append(b, '[')
+	b = r.Time.AppendFormat(b, "2006-01-02T15:04:05.000-07:00")
+	b = append(b, "] ["...)
+	b = append(b, r.Level.String()...)
+	b = append(b, "] "...)
+	b = append(b, r.Message...)
+	r.Attrs(func(a slog.Attr) bool {
+		b = append(b, ' ')
+		b = append(b, a.Key...)
+		b = append(b, '=')
+		b = append(b, a.Value.String()...)
+		return true
+	})
+	b = append(b, '\n')
+	_, err := h.w.Write(b)
+	return err
+}
+
+// WithAttrs and WithGroup are no-ops: this project does not derive sub-loggers
+// via Logger.With, so they simply return the receiver.
+func (h *bracketHandler) WithAttrs(_ []slog.Attr) slog.Handler { return h }
+
+func (h *bracketHandler) WithGroup(_ string) slog.Handler { return h }
diff --git a/backend/server/server.go b/backend/server/server.go
index d8e872d..6a42223 100644
--- a/backend/server/server.go
+++ b/backend/server/server.go
@@ -77,7 +77,22 @@ func fetchClientInfo(c *gin.Context) (ua, ip string) {
 // Run sets up the Gin engine with all routes bound to the handler and starts
 // listening on the configured web port.
 func Run(handler *Handler) error {
+	if handler.Cfg.Others.Debug {
+		gin.SetMode(gin.DebugMode)
+	} else {
+		gin.SetMode(gin.ReleaseMode)
+	}
+
 	r := gin.Default()
+
+	err := r.SetTrustedProxies([]string{
+		"127.0.0.1",        // IPv4 本地回环
+		"::1",              // IPv6 本地回环
+	})
+	if err != nil {
+		return err
+	}
+
 	registerRoutes(r, handler)
 	return r.Run(fmt.Sprintf(":%d", handler.Cfg.Web.Port))
 }