mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2026-09-29 00:32:06 +08:00
ce94f63d9a
* ✨ feat(hook): 增强认证钩子和运行时缓存管理 ``` ♻ refactor(hook): 移除未使用的配置项并优化缓存设置 移除 AUTH_HOOKS_CONCURRENCY_LIMIT 配置项，该配置项不再使用 ✨feat(auth_ban): 简化缓存配置并添加实体参数支持 将 BAN_CACHE_TTL 相关配置从动态配置改为常量定义， 移除复杂的 TTL 值转换逻辑，并为 auth_ban 函数添加可选的 entity 参数以支持外部传入实体信息 ♻ refactor(auth_limit): 移除未使用的配置依赖 移除 AUTH_LIMIT_NOTICE_CD 配置项，直接使用常量值 2 作为限制通知冷却时间 📦 依赖更新: update playwright dependency to version 1.57.0 in pyproject.toml and requirements.txt ``` * ✨ feat(auth_checker): 增强插件模块处理和预过滤逻辑，支持用户插件兼容性 * ✨ feat(http_utils): 添加内容缓存机制以优化HTTP响应处理 * ✨ feat(cache): 添加群组插件设置视图缓存类型并更新相关逻辑 * ✨ feat(renderer): 优化渲染引擎，增加内存缓存管理和HTML文档处理逻辑 * ✨ feat(sign_in): 添加好感度排行和好感度总排行命令 * ✨ feat(renderer): 增强浏览器实例管理和模板预处理，支持历史 include 语法兼容 * ✨ feat(renderer): 优化 Playwright 环境检查逻辑，增加结果缓存以提高性能 * ✨ feat(renderer): 增强模板文件渲染策略，优化资产加载路径处理 * 🚨 auto fix by pre-commit hooks * ✨ feat(log): 增加日志内容安全序列化，避免超长 base64 等污染日志 ✨ feat(log_sanitizer): 添加对嵌入超长 base64/data URI 的清理功能 ✨ feat(auth_checker): 添加 Alconna 快捷方式缓存检查，优化路由匹配逻辑 * feat(renderer): 添加渲染结果内存缓存功能以优化性能 feat(help): 实现帮助菜单图像缓存机制 feat(sign_in): 更新HTML卡片生成以支持动画禁用和剪裁 feat(superuser): 在启动时预热超级用户帮助缓存 feat(theme): 优化主题管理器的资源解析缓存 * 🚨 auto fix by pre-commit hooks * feat(renderer): 增加全页面视口最大宽度限制并优化内容尺寸计算 --------- Co-authored-by: ATTomatoo <1126160939@qq.com> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: HibiKier <45528451+HibiKier@users.noreply.github.com>
92 lines
3.0 KiB
Python
92 lines
3.0 KiB
Python
from __future__ import annotations
|
|
|
|
import asyncio
|
|
from collections import OrderedDict
|
|
import hashlib
|
|
import time
|
|
from typing import Any
|
|
|
|
from zhenxun.utils.pydantic_compat import dump_json_safely
|
|
|
|
|
|
class RenderResultMemoryCache:
|
|
def __init__(
|
|
self,
|
|
ttl_seconds: float,
|
|
max_items: int,
|
|
max_total_bytes: int | None = None,
|
|
):
|
|
self._ttl_seconds = max(ttl_seconds, 0.0)
|
|
self._max_items = max(max_items, 1)
|
|
self._max_total_bytes = (
|
|
max_total_bytes
|
|
if isinstance(max_total_bytes, int) and max_total_bytes > 0
|
|
else None
|
|
)
|
|
self._cache: OrderedDict[str, tuple[float, bytes]] = OrderedDict()
|
|
self._total_bytes = 0
|
|
self._lock = asyncio.Lock()
|
|
|
|
@staticmethod
|
|
def build_key(payload: Any) -> str:
|
|
payload_text = dump_json_safely(
|
|
payload,
|
|
ensure_ascii=False,
|
|
sort_keys=True,
|
|
separators=(",", ":"),
|
|
)
|
|
return hashlib.sha256(payload_text.encode("utf-8")).hexdigest()
|
|
|
|
def _pop_oldest(self) -> None:
|
|
if not self._cache:
|
|
return
|
|
_, (_, value) = self._cache.popitem(last=False)
|
|
self._total_bytes -= len(value)
|
|
if self._total_bytes < 0:
|
|
self._total_bytes = 0
|
|
|
|
def _cleanup(self, now: float) -> None:
|
|
while self._cache:
|
|
expire_at, _ = next(iter(self._cache.values()))
|
|
if expire_at > now:
|
|
break
|
|
self._pop_oldest()
|
|
while len(self._cache) > self._max_items:
|
|
self._pop_oldest()
|
|
if self._max_total_bytes is not None:
|
|
while self._total_bytes > self._max_total_bytes and self._cache:
|
|
self._pop_oldest()
|
|
|
|
async def get(self, key: str) -> bytes | None:
|
|
now = time.monotonic()
|
|
async with self._lock:
|
|
self._cleanup(now)
|
|
item = self._cache.get(key)
|
|
if item is None:
|
|
return None
|
|
expire_at, value = item
|
|
if expire_at <= now:
|
|
removed = self._cache.pop(key, None)
|
|
if removed:
|
|
self._total_bytes -= len(removed[1])
|
|
if self._total_bytes < 0:
|
|
self._total_bytes = 0
|
|
return None
|
|
self._cache.move_to_end(key)
|
|
return value
|
|
|
|
async def set(self, key: str, value: bytes) -> None:
|
|
value_size = len(value)
|
|
if self._max_total_bytes is not None and value_size > self._max_total_bytes:
|
|
return
|
|
now = time.monotonic()
|
|
async with self._lock:
|
|
if old := self._cache.pop(key, None):
|
|
self._total_bytes -= len(old[1])
|
|
if self._total_bytes < 0:
|
|
self._total_bytes = 0
|
|
self._cache[key] = (now + self._ttl_seconds, value)
|
|
self._total_bytes += value_size
|
|
self._cache.move_to_end(key)
|
|
self._cleanup(now)
|