team/tyapi-server
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
ZhangRongHong
2025-08-10 15:19:10 +08:00
parent bb291f6847
commit ebc02b11e8
1 changed files with 5 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
internal/shared/middleware/daily_rate_limit.go
View File

@@ -3,7 +3,6 @@ package middleware
import ( import (
"context" "context"
"fmt" "fmt"
"net/http"
"strconv" "strconv"
"strings" "strings"
"time" "time"
@@ -91,7 +90,7 @@ func (m *DailyRateLimitMiddleware) GetPriority() int {
} }
// Handle 返回中间件处理函数 // Handle 返回中间件处理函数
func (c *DailyRateLimitMiddleware) Handle() gin.HandlerFunc { func (m *DailyRateLimitMiddleware) Handle() gin.HandlerFunc {
return func(c *gin.Context) { return func(c *gin.Context) {
ctx := c.Request.Context() ctx := c.Request.Context()
@@ -151,7 +150,7 @@ func (c *DailyRateLimitMiddleware) Handle() gin.HandlerFunc {
zap.String("request_id", c.GetString("request_id")), zap.String("request_id", c.GetString("request_id")),
zap.Error(err)) zap.Error(err))
// 隐藏限制信息，返回通用错误 // 隐藏限制信息，返回通用错误
m.response.InternalServerError(c, "系统繁忙，请稍后再试") m.response.InternalError(c, "系统繁忙，请稍后再试")
c.Abort() c.Abort()
return return
} }
@@ -163,7 +162,7 @@ func (c *DailyRateLimitMiddleware) Handle() gin.HandlerFunc {
zap.String("request_id", c.GetString("request_id")), zap.String("request_id", c.GetString("request_id")),
zap.Error(err)) zap.Error(err))
// 隐藏限制信息，返回通用错误 // 隐藏限制信息，返回通用错误
m.response.InternalServerError(c, "系统繁忙，请稍后再试") m.response.InternalError(c, "系统繁忙，请稍后再试")
c.Abort() c.Abort()
return return
} }
@@ -171,7 +170,7 @@ func (c *DailyRateLimitMiddleware) Handle() gin.HandlerFunc {
// 7. 增加计数 // 7. 增加计数
m.incrementCounters(ctx, clientIP) m.incrementCounters(ctx, clientIP)
// 7. 添加隐藏的响应头（仅用于内部监控） // 8. 添加隐藏的响应头（仅用于内部监控）
m.addHiddenHeaders(c, clientIP) m.addHiddenHeaders(c, clientIP)
c.Next() c.Next()
@@ -414,7 +413,7 @@ func (m *DailyRateLimitMiddleware) getCounter(ctx context.Context, key string) (
return 0, fmt.Errorf("解析计数失败: %w", err) return 0, fmt.Errorf("解析计数失败: %w", err)
} }
return count return count, nil
} }
// incrementCounter 增加计数器 // incrementCounter 增加计数器