feat(架构): 完善基础架构设计

internal/shared/middleware/ratelimit.go

@@ -2,11 +2,11 @@ package middleware
 
 import (
 	"fmt"
-	"net/http"
 	"sync"
 	"time"
 
 	"tyapi-server/internal/config"
+	"tyapi-server/internal/shared/interfaces"
 
 	"github.com/gin-gonic/gin"
 	"golang.org/x/time/rate"
@@ -15,14 +15,16 @@ import (
 // RateLimitMiddleware 限流中间件
 type RateLimitMiddleware struct {
 	config   *config.Config
+	response interfaces.ResponseBuilder
 	limiters map[string]*rate.Limiter
 	mutex    sync.RWMutex
 }
 
 // NewRateLimitMiddleware 创建限流中间件
-func NewRateLimitMiddleware(cfg *config.Config) *RateLimitMiddleware {
+func NewRateLimitMiddleware(cfg *config.Config, response interfaces.ResponseBuilder) *RateLimitMiddleware {
 	return &RateLimitMiddleware{
 		config:   cfg,
+		response: response,
 		limiters: make(map[string]*rate.Limiter),
 	}
 }
@@ -48,15 +50,13 @@ func (m *RateLimitMiddleware) Handle() gin.HandlerFunc {
 
 		// 检查是否允许请求
 		if !limiter.Allow() {
+			// 添加限流头部信息
 			c.Header("X-RateLimit-Limit", fmt.Sprintf("%d", m.config.RateLimit.Requests))
 			c.Header("X-RateLimit-Window", m.config.RateLimit.Window.String())
 			c.Header("Retry-After", "60")
 
-			c.JSON(http.StatusTooManyRequests, gin.H{
-				"success": false,
-				"message": "Rate limit exceeded",
-				"error":   "Too many requests",
-			})
+			// 使用统一的响应格式
+			m.response.TooManyRequests(c, "请求过于频繁，请稍后再试")
 			c.Abort()
 			return
 		}