辣评架构演进：从微信机器人到 Web 平台（二）

// cmd/server/router/router.go
func SetupRouter() *gin.Engine {
    r := gin.Default()

    // 公开路由
    public := r.Group("/api")
    {
        public.POST("/auth/login", handlers.Login)
        public.POST("/auth/register", handlers.Register)
    }

    // 认证路由
    auth := r.Group("/api")
    auth.Use(middleware.JWTAuth())
    {
        auth.GET("/user/profile", handlers.GetProfile)
        auth.POST("/submissions", handlers.CreateSubmission)
        auth.POST("/comments", handlers.CreateComment)
    }

    // 管理员路由
    admin := r.Group("/api/admin")
    admin.Use(middleware.JWTAuth(), middleware.RequireRole("admin"))
    {
        admin.GET("/users", handlers.ListUsers)
        admin.GET("/statistics", handlers.GetStatistics)
        admin.POST("/competitions", handlers.CreateCompetition)
    }

    // 静态文件服务
    r.Static("/", "./frontend")

    return r
}

// cmd/server/middleware/auth.go
func JWTAuth() gin.HandlerFunc {
    return func(c *gin.Context) {
        token := c.GetHeader("Authorization")
        if token == "" {
            c.JSON(401, gin.H{"error": "未提供认证Token"})
            c.Abort()
            return
        }

        // 验证 Token
        claims, err := ValidateToken(token)
        if err != nil {
            c.JSON(401, gin.H{"error": "Token无效或已过期"})
            c.Abort()
            return
        }

        // 将用户信息存入上下文
        c.Set("userID", claims.UserID)
        c.Set("username", claims.Username)
        c.Next()
    }
}

// cmd/server/database/database.go
func Initialize() error {
    db, err := gorm.Open(sqlite.Open(dbPath), &gorm.Config{
        Logger: logger.Default.LogMode(logger.Info),
    })
    if err != nil {
        return err
    }

    // 设置连接池
    sqlDB, _ := db.DB()
    sqlDB.SetMaxIdleConns(10)
    sqlDB.SetMaxOpenConns(100)
    sqlDB.SetConnMaxLifetime(time.Hour)

    // 自动迁移
    db.AutoMigrate(
        &models.User{},
        &models.Submission{},
        &models.Comment{},
        &models.Competition{},
        &models.QualificationStatistics{},
        // ...
    )

    DB = db
    return nil
}