bc56d471a8
✨ feat(agent): 新增 Token 配额门禁中间件（Redis 快照 + 封禁键 + 7 天懒重置） - 🚪 在 `POST /api/v1/agent/chat` 挂载 `TokenQuotaGuard`，在请求进入业务逻辑前完成额度校验 - ⚡ 新增 Redis 配额快照与封禁键机制：超额用户命中封禁键后可快速拦截，降低重复查库带来的开销 - 🗃️ 新增用户配额 DAO 能力：按需读取 `token_limit`、`token_usage`、`last_reset_at`，并支持基于“到期条件更新”的懒重置 - 🔄 实现 7 天懒重置策略：用户访问时若检测到配额周期已到期，则重置 `token_usage` 并清理封禁状态 - 🚫 新增超额响应码 `40051`，用于标识 `token usage exceeds limit`
100 lines
4.6 KiB
Go
100 lines
4.6 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, userRepo *dao.UserDAO, 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.4.0.dev",
|
|
})
|
|
})
|
|
|
|
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", middleware.IdempotencyMiddleware(cache), handlers.TaskHandler.AddTask)
|
|
taskGroup.PUT("/complete", middleware.IdempotencyMiddleware(cache), handlers.TaskHandler.CompleteTask)
|
|
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", middleware.IdempotencyMiddleware(cache), handlers.CourseHandler.AddUserCourses)
|
|
}
|
|
taskClassGroup := apiGroup.Group("/task-class")
|
|
{
|
|
taskClassGroup.Use(middleware.JWTTokenAuth(cache), middleware.RateLimitMiddleware(limiter, 20, 1))
|
|
taskClassGroup.POST("/add", middleware.IdempotencyMiddleware(cache), handlers.TaskClassHandler.UserAddTaskClass)
|
|
taskClassGroup.GET("/list", handlers.TaskClassHandler.UserGetTaskClassInfos)
|
|
taskClassGroup.GET("/get", handlers.TaskClassHandler.UserGetCompleteTaskClass)
|
|
taskClassGroup.PUT("/update", middleware.IdempotencyMiddleware(cache), handlers.TaskClassHandler.UserUpdateTaskClass)
|
|
taskClassGroup.POST("/insert-into-schedule", middleware.IdempotencyMiddleware(cache), handlers.TaskClassHandler.UserAddTaskClassItemIntoSchedule)
|
|
taskClassGroup.DELETE("/delete-item", middleware.IdempotencyMiddleware(cache), handlers.TaskClassHandler.DeleteTaskClassItem)
|
|
taskClassGroup.DELETE("/delete-class", middleware.IdempotencyMiddleware(cache), handlers.TaskClassHandler.DeleteTaskClass)
|
|
taskClassGroup.PUT("/apply-batch-into-schedule", middleware.IdempotencyMiddleware(cache), handlers.TaskClassHandler.UserInsertBatchTaskClassItemsIntoSchedule)
|
|
}
|
|
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", middleware.IdempotencyMiddleware(cache), handlers.ScheduleHandler.DeleteScheduleEvent)
|
|
scheduleGroup.GET("/recent-completed", handlers.ScheduleHandler.GetUserRecentCompletedSchedules)
|
|
scheduleGroup.GET("/current", handlers.ScheduleHandler.GetUserOngoingSchedule)
|
|
scheduleGroup.DELETE("/undo-task-item", middleware.IdempotencyMiddleware(cache), handlers.ScheduleHandler.UserRevocateTaskItemFromSchedule)
|
|
scheduleGroup.GET("/smart-planning", handlers.ScheduleHandler.SmartPlanning)
|
|
}
|
|
agentGroup := apiGroup.Group("/agent")
|
|
{
|
|
agentGroup.Use(middleware.JWTTokenAuth(cache), middleware.RateLimitMiddleware(limiter, 20, 1))
|
|
agentGroup.POST("/chat", middleware.TokenQuotaGuard(cache, userRepo), handlers.AgentHandler.ChatAgent)
|
|
agentGroup.GET("/conversation-meta", handlers.AgentHandler.GetConversationMeta)
|
|
agentGroup.GET("/conversation-list", handlers.AgentHandler.GetConversationList)
|
|
}
|
|
}
|
|
// 初始化Gin引擎
|
|
log.Println("Routes setup completed")
|
|
return r
|
|
}
|