refactor(main): 重构主应用结构以提高模块化

- 将main.go中的初始化逻辑提取到新的app包中
- 创建bootstrap包来处理配置加载和数据库初始化
- 添加middleware包来管理CORS中间件
- 创建router包来处理路由配置
- 简化main函数,只保留应用启动逻辑
- 使用依赖注入模式组织服务和处理器
This commit is contained in:
2026-02-26 16:19:13 +08:00
parent 2230b5ec9e
commit cf0685d102
6 changed files with 121 additions and 64 deletions

View File

@@ -1,78 +1,19 @@
package main
import (
"fmt"
"log"
"memora-api/internal/config"
"memora-api/internal/handler"
"memora-api/internal/model"
"memora-api/internal/service"
"github.com/gin-gonic/gin"
"gorm.io/driver/mysql"
"gorm.io/gorm"
"memora-api/internal/app"
)
func main() {
// 加载配置
cfg, err := config.LoadConfig("config.yaml")
application, err := app.New("config.yaml")
if err != nil {
log.Fatalf("加载配置失败: %v", err)
log.Fatal(err)
}
// 连接数据库
db, err := gorm.Open(mysql.Open(cfg.Database.DSN()), &gorm.Config{})
if err != nil {
log.Fatalf("连接数据库失败: %v", err)
}
// 自动迁移表
if err := db.AutoMigrate(&model.Word{}, &model.MemoryRecord{}); err != nil {
log.Fatalf("数据库迁移失败: %v", err)
}
// 初始化服务
wordService := service.NewWordService(db)
wordHandler := handler.NewWordHandler(wordService)
// 启动 Gin
r := gin.Default()
// CORS 中间件
r.Use(func(c *gin.Context) {
c.Writer.Header().Set("Access-Control-Allow-Origin", "*")
c.Writer.Header().Set("Access-Control-Allow-Methods", "GET, POST, PUT, DELETE, OPTIONS")
c.Writer.Header().Set("Access-Control-Allow-Headers", "Content-Type, Authorization")
if c.Request.Method == "OPTIONS" {
c.AbortWithStatus(204)
return
}
c.Next()
})
// 路由
api := r.Group("/api")
{
// 记忆模式
api.POST("/words", wordHandler.AddWord)
api.GET("/words", wordHandler.GetWords)
// 复习模式
api.GET("/review", wordHandler.GetReviewWords)
api.POST("/review", wordHandler.SubmitReview)
// 统计
api.GET("/stats", wordHandler.GetStatistics)
// 音频
api.GET("/audio", wordHandler.GetAudio)
}
// 启动服务器
addr := fmt.Sprintf("%s:%d", cfg.Server.Host, cfg.Server.Port)
log.Printf("服务器启动在: %s", addr)
if err := r.Run(addr); err != nil {
log.Printf("服务器启动在: %s", application.Addr())
if err := application.Run(); err != nil {
log.Fatalf("启动服务器失败: %v", err)
}
}