Kris
05abecc02b
该提交实现了支持企业微信、微信公众号、个人微信桥接三种模式的完整微信渠道适配器，包含以下核心模块： 1. 基础认证与配置相关：auth_adapter、config_reload、setup_contract等 2. 消息处理与格式转换：format、attachment_adapter、outbound_adapter等 3. 多模式客户端支持：wecom/mp子模块，包含加解密、消息收发能力 4. 辅助能力：限速器、防抖、会话绑定、事件映射、模板渲染等 5. 扩展能力：二维码登录、消息读取、特权用户、心跳监控等 实现了完整的微信生态对接能力，支持消息收发、事件处理、API调用限流、配置热重载等功能。
56 lines
1.4 KiB
Python
56 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:
|
|
tokens_before = self._tokens
|
|
self._try_acquire_sync()
|
|
return self._tokens < tokens_before
|
|
|
|
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) -> None:
|
|
|
|
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
|
|
|
|
@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
|