mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2026-10-05 11:49:59 +08:00
性能优化 (#2126)
* 性能优化 * 代码改进 * 优化浏览器代际切换逻辑 * 统一缓存与生命周期 * 添加aiomysql依赖 * 优化插件路径处理逻辑,简化条件判断;在虚拟环境包管理器中添加编码和错误处理参数以增强稳定性 * 🚨 auto fix by pre-commit hooks * 优化Windows下的关闭逻辑 * 代码优化 * bugfix:修复配置重载问题 * bugfix:修复插件加载启动竞态问题 * 收敛事件入口和权限上下文 * 优化 Windows launcher 关闭重启兜底 --------- Co-authored-by: HibiKier <775757368@qq.com> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
This commit is contained in:
co-authored by
HibiKier
pre-commit-ci[bot]
parent
24c316cd2c
commit
5d92ccd3b0
@@ -1,25 +1,77 @@
|
||||
from collections import OrderedDict
|
||||
import time
|
||||
from typing import ClassVar
|
||||
|
||||
|
||||
class MessageManager:
|
||||
data: ClassVar[dict[str, list[str]]] = {}
|
||||
_MAX_USERS: ClassVar[int] = 4096
|
||||
_MAX_MESSAGES_PER_USER: ClassVar[int] = 200
|
||||
_TRIM_MESSAGES_TO: ClassVar[int] = 100
|
||||
_USER_TTL_SECONDS: ClassVar[float] = 6 * 60 * 60
|
||||
data: ClassVar[OrderedDict[str, tuple[float, list[str]]]] = OrderedDict()
|
||||
|
||||
@classmethod
|
||||
def _prune(cls, now: float | None = None) -> None:
|
||||
now = time.monotonic() if now is None else now
|
||||
stale_before = now - cls._USER_TTL_SECONDS
|
||||
stale_uids = [
|
||||
uid for uid, (last_seen, _) in cls.data.items() if last_seen <= stale_before
|
||||
]
|
||||
for uid in stale_uids:
|
||||
cls.data.pop(uid, None)
|
||||
while len(cls.data) > cls._MAX_USERS:
|
||||
cls.data.popitem(last=False)
|
||||
|
||||
@classmethod
|
||||
def _touch(cls, uid: str, messages: list[str], now: float | None = None) -> None:
|
||||
now = time.monotonic() if now is None else now
|
||||
cls.data[uid] = (now, messages)
|
||||
cls.data.move_to_end(uid)
|
||||
|
||||
@classmethod
|
||||
def add(cls, uid: str, msg_id: str):
|
||||
if uid not in cls.data:
|
||||
cls.data[uid] = []
|
||||
cls.data[uid].append(msg_id)
|
||||
now = time.monotonic()
|
||||
cls._prune(now)
|
||||
_, messages = cls.data.get(uid, (now, []))
|
||||
messages.append(msg_id)
|
||||
cls._touch(uid, messages, now)
|
||||
cls.remove_check(uid)
|
||||
cls._prune(now)
|
||||
|
||||
@classmethod
|
||||
def check(cls, uid: str, msg_id: str) -> bool:
|
||||
return msg_id in cls.data.get(uid, [])
|
||||
now = time.monotonic()
|
||||
cls._prune(now)
|
||||
entry = cls.data.get(uid)
|
||||
if entry is None:
|
||||
return False
|
||||
_, messages = entry
|
||||
cls._touch(uid, messages, now)
|
||||
return msg_id in messages
|
||||
|
||||
@classmethod
|
||||
def remove_check(cls, uid: str):
|
||||
if len(cls.data[uid]) > 200:
|
||||
cls.data[uid] = cls.data[uid][100:]
|
||||
entry = cls.data.get(uid)
|
||||
if entry is None:
|
||||
return
|
||||
_, messages = entry
|
||||
if len(messages) > cls._MAX_MESSAGES_PER_USER:
|
||||
messages = messages[-cls._TRIM_MESSAGES_TO :]
|
||||
cls._touch(uid, messages)
|
||||
|
||||
@classmethod
|
||||
def get(cls, uid: str) -> list[str]:
|
||||
return cls.data[uid] if uid in cls.data else []
|
||||
now = time.monotonic()
|
||||
cls._prune(now)
|
||||
entry = cls.data.get(uid)
|
||||
if entry is None:
|
||||
return []
|
||||
_, messages = entry
|
||||
cls._touch(uid, messages, now)
|
||||
return list(messages)
|
||||
|
||||
@classmethod
|
||||
def clear_all(cls) -> int:
|
||||
size = len(cls.data)
|
||||
cls.data.clear()
|
||||
return size
|
||||
|
||||
Reference in New Issue
Block a user