110 lines
3.1 KiB
Python
110 lines
3.1 KiB
Python
|
"""
|
|||
|
|
全局锁管理器模块
|
|||
|
|
用于管理需要冷却时间(CD)的操作锁
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
import asyncio
|
|||
|
|
import time
|
|||
|
|
from typing import Optional
|
|||
|
|
|
|||
|
|
|
|||
|
|
class RefreshLockManager:
|
|||
|
|
"""刷新操作锁管理器，确保一次只能执行一个刷新操作，且有30分钟的冷却时间"""
|
|||
|
|
|
|||
|
|
def __init__(self, cooldown_seconds: int = 1800): # 默认30分钟 = 1800秒
|
|||
|
|
"""
|
|||
|
|
初始化锁管理器
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
cooldown_seconds: 冷却时间（秒），默认为1800秒（30分钟）
|
|||
|
|
"""
|
|||
|
|
self._lock = asyncio.Lock()
|
|||
|
|
self._last_refresh_time: Optional[float] = None
|
|||
|
|
self._cooldown_seconds = cooldown_seconds
|
|||
|
|
self._is_refreshing = False
|
|||
|
|
|
|||
|
|
async def try_acquire(self) -> tuple[bool, Optional[float]]:
|
|||
|
|
"""
|
|||
|
|
尝试获取锁并检查冷却时间
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
tuple[bool, Optional[float]]:
|
|||
|
|
- bool: 是否成功获取锁（未在冷却期且未被占用）
|
|||
|
|
- Optional[float]: 如果在冷却期，返回剩余冷却时间（秒），否则为None
|
|||
|
|
"""
|
|||
|
|
# 检查是否有其他人正在刷新
|
|||
|
|
if self._is_refreshing:
|
|||
|
|
return False, None
|
|||
|
|
|
|||
|
|
# 检查冷却时间
|
|||
|
|
if self._last_refresh_time is not None:
|
|||
|
|
elapsed = time.time() - self._last_refresh_time
|
|||
|
|
if elapsed < self._cooldown_seconds:
|
|||
|
|
remaining_cooldown = self._cooldown_seconds - elapsed
|
|||
|
|
return False, remaining_cooldown
|
|||
|
|
|
|||
|
|
# 尝试获取锁（非阻塞）
|
|||
|
|
acquired = not self._lock.locked()
|
|||
|
|
if acquired:
|
|||
|
|
await self._lock.acquire()
|
|||
|
|
self._is_refreshing = True
|
|||
|
|
|
|||
|
|
return acquired, None
|
|||
|
|
|
|||
|
|
def release(self):
|
|||
|
|
"""
|
|||
|
|
释放锁并更新最后刷新时间
|
|||
|
|
"""
|
|||
|
|
self._last_refresh_time = time.time()
|
|||
|
|
self._is_refreshing = False
|
|||
|
|
if self._lock.locked():
|
|||
|
|
self._lock.release()
|
|||
|
|
|
|||
|
|
def get_remaining_cooldown(self) -> Optional[float]:
|
|||
|
|
"""
|
|||
|
|
获取剩余冷却时间
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
Optional[float]: 剩余冷却时间（秒），如果不在冷却期则返回None
|
|||
|
|
"""
|
|||
|
|
if self._last_refresh_time is None:
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
elapsed = time.time() - self._last_refresh_time
|
|||
|
|
if elapsed < self._cooldown_seconds:
|
|||
|
|
return self._cooldown_seconds - elapsed
|
|||
|
|
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
def is_in_cooldown(self) -> bool:
|
|||
|
|
"""
|
|||
|
|
检查是否在冷却期内
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
bool: 是否在冷却期内
|
|||
|
|
"""
|
|||
|
|
return self.get_remaining_cooldown() is not None
|
|||
|
|
|
|||
|
|
def is_refreshing(self) -> bool:
|
|||
|
|
"""
|
|||
|
|
检查是否正在刷新
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
bool: 是否正在刷新
|
|||
|
|
"""
|
|||
|
|
return self._is_refreshing
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 全局单例实例
|
|||
|
|
_refresh_lock_manager = RefreshLockManager(cooldown_seconds=1800) # 30分钟
|
|||
|
|
|
|||
|
|
|
|||
|
|
def get_refresh_lock_manager() -> RefreshLockManager:
|
|||
|
|
"""
|
|||
|
|
获取全局刷新锁管理器实例
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
RefreshLockManager: 全局锁管理器实例
|
|||
|
|
"""
|
|||
|
|
return _refresh_lock_manager
|