jiazhizhong/higress
1
0
Fork 0
mirror of https://github.com/alibaba/higress.git synced 2026-05-30 23:57:28 +08:00
Code Issues Packages Projects Releases Wiki Activity

improve the logic for constructing redis key (#1933)

Browse Source
This commit is contained in:
rinfx
2025-03-21 14:02:59 +08:00
committed by GitHub
parent 4223b2d666
commit 2640c76760
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
plugins/wasm-go/extensions/ai-token-ratelimit/main.go
View File

@@ -39,7 +39,7 @@ func main() {
} }
const ( const (
ClusterRateLimitFormat string = "higress-token-ratelimit:%s:%s:%s:%s" ClusterRateLimitFormat string = "higress-token-ratelimit:%s:%s:%d:%d:%s:%s" // ruleName, limitType, timewindow, windowsize, key, val
RequestPhaseFixedWindowScript string = ` RequestPhaseFixedWindowScript string = `
local ttl = redis.call('ttl', KEYS[1]) local ttl = redis.call('ttl', KEYS[1])
if ttl < 0 then if ttl < 0 then
@@ -103,7 +103,7 @@ func onHttpRequestHeaders(ctx wrapper.HttpContext, config ClusterKeyRateLimitCon
} }
// 构建redis限流key和参数 // 构建redis限流key和参数
limitKey := fmt.Sprintf(ClusterRateLimitFormat, config.ruleName, ruleItem.limitType, ruleItem.key, val) limitKey := fmt.Sprintf(ClusterRateLimitFormat, config.ruleName, ruleItem.limitType, configItem.timeWindow, configItem.count, ruleItem.key, val)
keys := []interface{}{limitKey} keys := []interface{}{limitKey}
args := []interface{}{configItem.count, configItem.timeWindow} args := []interface{}{configItem.count, configItem.timeWindow}