XEngineers/CamTalk
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: 添加限流触发日志

Browse Source
This commit is contained in:
hhs
2026-06-21 22:16:38 +08:00
parent 065673fae2
commit 8a43f4406a
1 changed files with 9 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
backend/internal/ratelimit/middleware.go
View File

@@ -5,6 +5,8 @@ import (
"net/http" "net/http"
"github.com/gin-gonic/gin" "github.com/gin-gonic/gin"
"github.com/hhs/camtalk/internal/trace"
) )
// Middleware 返回 Gin 中间件，按 key 维度限流。 // Middleware 返回 Gin 中间件，按 key 维度限流。
@@ -26,6 +28,13 @@ func Middleware(limiter Limiter, keyFunc func(*gin.Context) string) gin.HandlerF
allowed, retryAfter := limiter.Allow(c.Request.Context(), key) allowed, retryAfter := limiter.Allow(c.Request.Context(), key)
if !allowed { if !allowed {
log := trace.FromContext(c.Request.Context())
log.Warnw("rate limited",
"client_ip", c.ClientIP(),
"path", c.Request.URL.Path,
"limit_key", key,
"retry_after_sec", int(retryAfter.Seconds()+0.5))
// 设置 Retry-After header（秒） // 设置 Retry-After header（秒）
c.Header("Retry-After", fmt.Sprintf("%d", int(retryAfter.Seconds()+0.5))) c.Header("Retry-After", fmt.Sprintf("%d", int(retryAfter.Seconds()+0.5)))