feat:初版

apps/api/app/services/rate_limit.py

@@ -0,0 +1,33 @@
+import time
+import redis.asyncio as redis
+from ..core import get_settings
+
+settings = get_settings()
+
+
+class RateLimitService:
+    def __init__(self):
+        self.redis: redis.Redis | None = None
+
+    async def connect(self):
+        self.redis = redis.from_url(settings.redis_url)
+
+    async def disconnect(self):
+        if self.redis:
+            await self.redis.close()
+
+    async def is_allowed(self, key: str, limit: int | None = None) -> bool:
+        if not self.redis:
+            return True
+        limit = limit or settings.rate_limit_per_minute
+        now = int(time.time())
+        window_key = f"rl:{key}:{now // 60}"
+
+        count = await self.redis.incr(window_key)
+        if count == 1:
+            await self.redis.expire(window_key, 60)
+
+        return count <= limit
+
+
+rate_limit_service = RateLimitService()