d5f0b8da63
feat: 🚦 新增基于 Redis 令牌桶的限流中间件 - 使用 Redis 实现令牌桶算法进行限流 🪣 - 覆盖除登录、注册、刷新 token 以外的所有接口 🔒 fix: 🐛 修复任务块添加到日程接口可修改已安排任务时间的问题 - 禁止通过该接口直接修改已安排任务块的时间 - 修正不合理的业务逻辑，保证数据一致性 ✅
85 lines
3.0 KiB
Go
85 lines
3.0 KiB
Go
// Package routers 路由配置
|
|
// 定义所有HTTP路由和路由组
|
|
package routers
|
|
|
|
import (
|
|
"log"
|
|
|
|
"github.com/LoveLosita/smartflow/backend/api"
|
|
"github.com/LoveLosita/smartflow/backend/dao"
|
|
"github.com/LoveLosita/smartflow/backend/middleware"
|
|
"github.com/LoveLosita/smartflow/backend/pkg"
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/spf13/viper"
|
|
)
|
|
|
|
// StartEngine 注册路由
|
|
func StartEngine(r *gin.Engine) {
|
|
// 从配置中获取端口
|
|
port := viper.GetString("server.port")
|
|
if port == "" {
|
|
port = "8080" // 默认端口
|
|
}
|
|
|
|
// 启动服务器
|
|
log.Printf("Server starting on port %s...", port)
|
|
if err := r.Run(":" + port); err != nil {
|
|
log.Fatalf("Failed to start server: %v", err)
|
|
}
|
|
}
|
|
|
|
func RegisterRouters(handlers *api.ApiHandlers, cache *dao.CacheDAO, limiter *pkg.RateLimiter) *gin.Engine {
|
|
// 初始化Gin引擎
|
|
r := gin.Default()
|
|
// 在这里注册所有的路由和路由组
|
|
apiGroup := r.Group("/api/v1")
|
|
{
|
|
// 健康检查路由
|
|
apiGroup.GET("/health", func(c *gin.Context) {
|
|
c.JSON(200, gin.H{
|
|
"status": "ok",
|
|
"version": "0.2.0.dev.260210",
|
|
})
|
|
})
|
|
|
|
userGroup := apiGroup.Group("/user")
|
|
{
|
|
userGroup.POST("/register", handlers.UserHandler.UserRegister)
|
|
userGroup.POST("/login", handlers.UserHandler.UserLogin)
|
|
userGroup.POST("/refresh-token", handlers.UserHandler.RefreshTokenHandler)
|
|
userGroup.POST("/logout", middleware.JWTTokenAuth(cache), middleware.RateLimitMiddleware(limiter, 20, 1), handlers.UserHandler.UserLogout)
|
|
}
|
|
taskGroup := apiGroup.Group("/task")
|
|
{
|
|
taskGroup.Use(middleware.JWTTokenAuth(cache), middleware.RateLimitMiddleware(limiter, 20, 1))
|
|
taskGroup.POST("/create", handlers.TaskHandler.AddTask)
|
|
taskGroup.GET("/get", handlers.TaskHandler.GetUserTasks)
|
|
}
|
|
courseGroup := apiGroup.Group("/course")
|
|
{
|
|
courseGroup.Use(middleware.JWTTokenAuth(cache), middleware.RateLimitMiddleware(limiter, 20, 1))
|
|
courseGroup.POST("/validate", handlers.CourseHandler.CheckUserCourse)
|
|
courseGroup.POST("/import", handlers.CourseHandler.AddUserCourses)
|
|
}
|
|
taskClassGroup := apiGroup.Group("/task-class")
|
|
{
|
|
taskClassGroup.Use(middleware.JWTTokenAuth(cache), middleware.RateLimitMiddleware(limiter, 20, 1))
|
|
taskClassGroup.POST("/add", handlers.TaskClassHandler.UserAddTaskClass)
|
|
taskClassGroup.GET("/list", handlers.TaskClassHandler.UserGetTaskClassInfos)
|
|
taskClassGroup.GET("/get", handlers.TaskClassHandler.UserGetCompleteTaskClass)
|
|
taskClassGroup.PUT("/update", handlers.TaskClassHandler.UserUpdateTaskClass)
|
|
taskClassGroup.POST("/insert-into-schedule", handlers.TaskClassHandler.UserAddTaskClassItemIntoSchedule)
|
|
}
|
|
scheduleGroup := apiGroup.Group("/schedule")
|
|
{
|
|
scheduleGroup.Use(middleware.JWTTokenAuth(cache), middleware.RateLimitMiddleware(limiter, 20, 1))
|
|
scheduleGroup.GET("/today", handlers.ScheduleHandler.GetUserTodaySchedule)
|
|
scheduleGroup.GET("/week", handlers.ScheduleHandler.GetUserWeeklySchedule)
|
|
scheduleGroup.DELETE("/delete", handlers.ScheduleHandler.DeleteScheduleEvent)
|
|
}
|
|
}
|
|
// 初始化Gin引擎
|
|
log.Println("Routes setup completed")
|
|
return r
|
|
}
|