fix: fix missing response header issue in fixed window strategy

dancing-ui · dancing-ui · commit 11896b803e79 · 2025-09-03T12:22:54.000+08:00
diff --git a/core/llm_token_ratelimit/ratelimit_checker.go b/core/llm_token_ratelimit/ratelimit_checker.go
@@ -82,6 +82,9 @@ func (c *FixedWindowChecker) checkLimitKey(ctx *Context, rule *MatchedRule) bool
 		)
 		return false
 	}
+	defer func() {
+		ctx.Set(KeyResponseHeaders, responseHeader)
+	}()
 	responseHeader.Set(KeyRequestID, ctx.Get(KeyRequestID).(string))
 	if remaining < 0 {
 		responseHeader.Set(ResponseHeaderRemainingTokens, fmt.Sprintf("%d", remaining))

Original file line number	Diff line number	Diff line change
`@@ -82,6 +82,9 @@ func (c FixedWindowChecker) checkLimitKey(ctx Context, rule *MatchedRule) bool`
`82`	`82`	`)`
`83`	`83`	`return false`
`84`	`84`	`}`
	`85`	`+ defer func() {`
	`86`	`+ ctx.Set(KeyResponseHeaders, responseHeader)`
	`87`	`+ }()`
`85`	`88`	`responseHeader.Set(KeyRequestID, ctx.Get(KeyRequestID).(string))`
`86`	`89`	`if remaining < 0 {`
`87`	`90`	`responseHeader.Set(ResponseHeaderRemainingTokens, fmt.Sprintf("%d", remaining))`