perf(限流服务): 优化Redis查询以减少命令数量

使用zcount替代zremrangebyscore和zcard组合查询，减少一次Redis操作
2026-04-20 17:46:05 +08:00
parent b03300c804
commit 08b5c7bc8a
1 changed files with 2 additions and 3 deletions
--- a/api/app/services/api_key_service.py
+++ b/api/app/services/api_key_service.py
@@ -292,11 +292,10 @@ class RateLimiterService:
        key = f"rate_limit:qps:{api_key_id}"

        async with self.redis.pipeline() as pipe:
-            pipe.zremrangebyscore(key, 0, window_start)
-            pipe.zcard(key)
+            pipe.zcount(key, window_start, "+inf")
            results = await pipe.execute()

-        current = results[1]
+        current = results[0]

        if current >= limit:
            return False, {