136 lines
2.9 KiB
Go
136 lines
2.9 KiB
Go
package main
|
|
|
|
import (
|
|
"go_blog/config"
|
|
"go_blog/models"
|
|
"go_blog/routers"
|
|
"log/slog"
|
|
"os"
|
|
"time"
|
|
|
|
"go_blog/themes"
|
|
|
|
"gorm.io/gorm"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
// @title 个人博客系统API
|
|
// @version 1.0
|
|
// @description 基于Go语言的可定制主题博客系统
|
|
const templatePath = "./templates/*"
|
|
|
|
func main() {
|
|
|
|
//读取配置文件
|
|
// 1. 初始化配置
|
|
conf, err := config.LoadConfig("config.yml")
|
|
if err != nil {
|
|
slog.Error("配置加载失败", "error", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
// 2. 初始化日志系统
|
|
// if err := logger.Initialize(cfg.Log); err != nil {
|
|
// slog.Error("日志初始化失败", "error", err)
|
|
// os.Exit(1)
|
|
// }
|
|
// defer logger.Flush()
|
|
logger := slog.New(slog.NewJSONHandler(os.Stdout, nil))
|
|
slog.SetDefault(logger)
|
|
|
|
// 3. 初始化数据库
|
|
models.InitDatabase(conf)
|
|
sqlDB, _ := models.DB.DB()
|
|
defer func() {
|
|
if err := sqlDB.Close(); err != nil {
|
|
slog.Error("数据库关闭异常", "error", err)
|
|
}
|
|
}()
|
|
// 4. 初始化主题系统
|
|
themeManager := themes.NewManager("web\\themes")
|
|
if err := themeManager.LoadTheme(conf.Theme.Current); err != nil {
|
|
slog.Error("主题系统初始化失败", "error", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
// 5. 创建Gin实例
|
|
// 根据环境设置Gin模式
|
|
if conf.Env == config.EnvProduction {
|
|
gin.SetMode(gin.ReleaseMode)
|
|
}
|
|
router := gin.Default()
|
|
router.LoadHTMLGlob(templatePath)
|
|
themeManager.RegisterStaticRoutes(router)
|
|
|
|
// 6. 注册中间件
|
|
router.Use(
|
|
loggerMiddleware(),
|
|
gin.Logger(),
|
|
gin.Recovery(),
|
|
databaseMiddleware(models.DB),
|
|
//configMiddleware(cfg),
|
|
themeMiddleware(themeManager),
|
|
)
|
|
|
|
// 7. 设置模板路径
|
|
//router.HTMLRender = createTemplateRenderer(cfg.Theme.Current)
|
|
|
|
// 8. 注册路由
|
|
routers.RegisterRoutes(router)
|
|
// 9. 启动服务
|
|
router.Run(":" + conf.Server.Port)
|
|
}
|
|
|
|
// 自定义模板渲染器
|
|
// func createTemplateRenderer(currentTheme string) gin.HTMLRender {
|
|
// return gin.HTMLTemplates{
|
|
// Root: "web/templates",
|
|
// Extension: ".html",
|
|
// FuncMap: template.FuncMap{
|
|
// "safe": func(s string) template.HTML { return template.HTML(s) },
|
|
// },
|
|
// Directory: "web/themes/" + currentTheme,
|
|
// }
|
|
// }
|
|
|
|
// 数据库中间件
|
|
func databaseMiddleware(db *gorm.DB) gin.HandlerFunc {
|
|
return func(c *gin.Context) {
|
|
c.Set("DB", db)
|
|
c.Next()
|
|
}
|
|
}
|
|
|
|
// 配置上下文中间件
|
|
func configMiddleware(cfg *config.Config) gin.HandlerFunc {
|
|
return func(c *gin.Context) {
|
|
c.Set("Config", cfg)
|
|
c.Next()
|
|
}
|
|
}
|
|
|
|
// 主题管理器中间件
|
|
func themeMiddleware(manager *themes.ThemeManager) gin.HandlerFunc {
|
|
return func(c *gin.Context) {
|
|
c.Set("ThemeManager", manager)
|
|
c.Next()
|
|
}
|
|
}
|
|
|
|
func loggerMiddleware() gin.HandlerFunc {
|
|
return func(c *gin.Context) {
|
|
start := time.Now()
|
|
|
|
c.Next()
|
|
|
|
duration := time.Since(start)
|
|
slog.Info("请求处理完成",
|
|
"status", c.Writer.Status(),
|
|
"method", c.Request.Method,
|
|
"path", c.Request.URL.Path,
|
|
"duration", duration,
|
|
)
|
|
}
|
|
}
|