Kris
1f78c44b03
这是一个批量整理提交，包含以下主要改动： 1. 删除多处冗余的空行和未使用的导入 2. 修复文件末尾缺少换行符的问题 3. 调整部分模块的导入顺序与代码排版 4. 修复部分配置默认值与策略逻辑 5. 新增多个功能模块与辅助工具 6. 完善异常处理与日志记录 7. 修复速率限制、消息缓存、权限校验等逻辑bug 8. 废弃部分旧有API与配置项并添加警告提示
55 lines
1.4 KiB
Python
55 lines
1.4 KiB
Python
from __future__ import annotations
|
|
|
|
import asyncio
|
|
import time
|
|
|
|
|
|
class TokenBucketRateLimiter:
|
|
def __init__(self, rate: int, per: float = 60.0):
|
|
self._rate = rate
|
|
self._per = per
|
|
self._tokens = float(rate)
|
|
self._last_refill = time.monotonic()
|
|
self._lock = asyncio.Lock()
|
|
|
|
async def acquire(self) -> bool:
|
|
async with self._lock:
|
|
return self._try_acquire()
|
|
|
|
def acquire_sync(self) -> bool:
|
|
return self._try_acquire_sync()
|
|
|
|
def _try_acquire(self) -> bool:
|
|
now = time.monotonic()
|
|
elapsed = now - self._last_refill
|
|
self._tokens = min(self._rate, self._tokens + elapsed * (self._rate / self._per))
|
|
self._last_refill = now
|
|
|
|
if self._tokens >= 1.0:
|
|
self._tokens -= 1.0
|
|
return True
|
|
return False
|
|
|
|
def _try_acquire_sync(self) -> bool:
|
|
now = time.monotonic()
|
|
elapsed = now - self._last_refill
|
|
self._tokens = min(self._rate, self._tokens + elapsed * (self._rate / self._per))
|
|
self._last_refill = now
|
|
|
|
if self._tokens >= 1.0:
|
|
self._tokens -= 1.0
|
|
return True
|
|
return False
|
|
|
|
@property
|
|
def available_tokens(self) -> float:
|
|
return self._tokens
|
|
|
|
@property
|
|
def rate(self) -> int:
|
|
return self._rate
|
|
|
|
@property
|
|
def period(self) -> float:
|
|
return self._per
|