xevion/linkpulse
1
0
Fork 0
mirror of https://github.com/Xevion/linkpulse.git synced 2025-12-06 03:15:35 -06:00
Code Issues Packages Projects Releases Wiki Activity

Switch limits to asynchronous storage/strategy

Browse Source
This commit is contained in:
Xevion
2024-11-09 23:27:27 -06:00
parent fd90873f7b
commit 10919d0333
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
backend/linkpulse/dependencies.py
View File

@@ -1,8 +1,8 @@
import os import os
import structlog import structlog
from fastapi import HTTPException, Request, Response, status from fastapi import HTTPException, Request, Response, status
from limits.strategies import MovingWindowRateLimiter from limits.aio.strategies import MovingWindowRateLimiter
from limits.storage import MemoryStorage from limits.aio.storage import MemoryStorage
from limits import parse from limits import parse
storage = MemoryStorage() storage = MemoryStorage()
@@ -31,7 +31,7 @@ class RateLimiter:
# The reason for this is so tests don't compete with each other for rate limiting # The reason for this is so tests don't compete with each other for rate limiting
key += "." + os.environ["PYTEST_CURRENT_TEST"] key += "." + os.environ["PYTEST_CURRENT_TEST"]
if not strategy.hit(self.limit, key): if not await strategy.hit(self.limit, key):
logger.warning("Rate limit exceeded", key=key) logger.warning("Rate limit exceeded", key=key)
raise HTTPException( raise HTTPException(
status_code=status.HTTP_429_TOO_MANY_REQUESTS, status_code=status.HTTP_429_TOO_MANY_REQUESTS,