优化router、main等文件。

main.go

@@ -1,15 +1,24 @@
 package main
 
 import (
+	"context"
+	"errors"
 	"log/slog"
+	"net/http"
 	"os"
+	"os/signal"
+	"strconv"
+	"syscall"
+	"time"
 
 	"goblog/config"
 	"goblog/database"
 	"goblog/models"
 	"goblog/routers"
+	"goblog/utils"
 
 	"github.com/gin-gonic/gin"
+	"gorm.io/gorm"
 )
 
 func main() {
@@ -19,10 +28,8 @@ func main() {
 	// 设置 Gin 模式
 	gin.SetMode(cfg.Server.Mode)
 
-	// 初始化日志系统
-	logger := slog.New(slog.NewTextHandler(os.Stdout, &slog.HandlerOptions{
-		Level: slog.LevelInfo,
-	}))
+	// 初始化日志系统（使用 Server.LogLevel）
+	logger := newLogger(cfg.Server.LogLevel)
 	slog.SetDefault(logger)
 
 	slog.Info("启动博客系统",
@@ -30,6 +37,8 @@ func main() {
 		"theme", cfg.App.Theme,
 		"db_driver", cfg.Database.Driver,
 		"port", cfg.Server.Port,
+		"mode", cfg.Server.Mode,
+		"log_level", cfg.Server.LogLevel,
 	)
 
 	// 初始化数据库
@@ -37,31 +46,105 @@ func main() {
 		slog.Error("数据库初始化失败", "error", err)
 		os.Exit(1)
 	}
+	// defer database.Close() // 如果提供
 
-	// 从数据库读取主题设置，如果不存在则使用配置文件中的值
-	theme := cfg.App.Theme
-	var option models.Option
-	if err := database.DB.Where("name = ? AND user_id = ?", "theme", 0).First(&option).Error; err == nil {
-		theme = option.Value
-		slog.Info("从数据库加载主题", "theme", theme)
-	} else {
-		slog.Info("使用配置文件中的主题", "theme", theme)
+	// 从数据库加载主题（若不存在则使用配置值）
+	theme, err := loadTheme(cfg.App.Theme)
+	if err != nil {
+		slog.Error("加载主题失败", "error", err)
+		os.Exit(1)
 	}
 
-	// 创建路由引擎
-	r := gin.Default()
+	// 创建路由引擎（使用自定义 slog 中间件）
+	r := gin.New()
+	r.Use(gin.Recovery())
+	r.Use(slogMiddleware(logger))
 
-	// 配置模板渲染器
-	routers.SetupTemplates(r, theme)
+	// 配置模板渲染器（现在 LoadTemplates 返回 error）
+	htmlRenderer, err := utils.LoadTemplates(theme)
+	if err != nil {
+		slog.Error("加载模板失败", "error", err, "theme", theme)
+		os.Exit(1)
+	}
+	utils.SetGlobalRenderer(htmlRenderer)
+	r.HTMLRender = htmlRenderer
 
 	// 配置所有路由
 	routers.SetupRoutes(r)
 
-	// 启动服务器
-	port := routers.GetServerPort(cfg.Server.Port)
-	slog.Info("服务器启动成功", "port", port, "mode", cfg.Server.Mode)
-	if err := r.Run(port); err != nil {
-		slog.Error("服务器启动失败", "error", err)
-		os.Exit(1)
+	// 启动服务器（优雅关闭）
+	runServer(r, cfg.Server.Port)
+}
+
+// newLogger 根据配置字符串创建 slog.Logger
+func newLogger(level string) *slog.Logger {
+	var lvl slog.Level
+	if err := lvl.UnmarshalText([]byte(level)); err != nil {
+		lvl = slog.LevelInfo // 默认 info
+	}
+	opts := &slog.HandlerOptions{Level: lvl}
+	return slog.New(slog.NewTextHandler(os.Stdout, opts))
+}
+
+// loadTheme 从数据库读取系统主题，若不存在则使用 fallback
+func loadTheme(fallback string) (string, error) {
+	var option models.Option
+	err := database.DB.Where("name = ? AND user_id = ?", "theme", 0).First(&option).Error
+	if err == nil {
+		slog.Info("从数据库加载主题", "theme", option.Value)
+		return option.Value, nil
+	}
+	if errors.Is(err, gorm.ErrRecordNotFound) {
+		slog.Info("使用配置文件中的主题", "theme", fallback)
+		return fallback, nil
+	}
+	return "", err // 真实错误必须返回
+}
+
+// slogMiddleware 自定义请求日志中间件
+func slogMiddleware(logger *slog.Logger) gin.HandlerFunc {
+	return func(c *gin.Context) {
+		start := time.Now()
+		path := c.Request.URL.Path
+
+		c.Next()
+
+		logger.Info("HTTP请求",
+			"method", c.Request.Method,
+			"path", path,
+			"status", c.Writer.Status(),
+			"duration", time.Since(start),
+			"client_ip", c.ClientIP(),
+		)
+	}
+}
+
+// runServer 启动 HTTP 服务并支持优雅关闭
+func runServer(r *gin.Engine, port int) {
+	addr := ":" + strconv.Itoa(port)
+	srv := &http.Server{
+		Addr:    addr,
+		Handler: r,
+	}
+
+	go func() {
+		slog.Info("服务器启动成功", "addr", addr)
+		if err := srv.ListenAndServe(); err != nil && !errors.Is(err, http.ErrServerClosed) {
+			slog.Error("服务器启动失败", "error", err)
+			os.Exit(1)
+		}
+	}()
+
+	quit := make(chan os.Signal, 1)
+	signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM)
+	<-quit
+	slog.Info("正在关闭服务器...")
+
+	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
+	defer cancel()
+	if err := srv.Shutdown(ctx); err != nil {
+		slog.Error("服务器强制关闭", "error", err)
+	} else {
+		slog.Info("服务器已优雅关闭")
 	}
 }