mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2026-09-29 00:32:06 +08:00
46 lines
1.3 KiB
Python
46 lines
1.3 KiB
Python
"""
|
|
优化后的权限检查系统入口 (V2)
|
|
|
|
主要改进：
|
|
1. 使用预聚合的权限快照，将查询次数从6-10次降低到1-2次
|
|
2. 本地内存缓存 + Redis缓存双层结构
|
|
3. 所有权限检查基于内存数据，无额外I/O
|
|
|
|
使用方式：
|
|
1. 在 hooks/__init__.py 中将 auth_checker 替换为 auth_checker_v2
|
|
2. 或者通过配置开关选择使用哪个版本
|
|
|
|
性能对比：
|
|
- 原版本：6-10次查询，平均延迟~50ms
|
|
- V2版本：1-2次查询，平均延迟~10ms
|
|
"""
|
|
|
|
import nonebot
|
|
|
|
from zhenxun.services.auth_snapshot import (
|
|
AuthSnapshotService,
|
|
PluginSnapshotService,
|
|
)
|
|
from zhenxun.services.log import logger
|
|
from zhenxun.utils.manager.priority_manager import PriorityLifecycle
|
|
|
|
driver = nonebot.get_driver()
|
|
|
|
|
|
# 启动时预热插件缓存
|
|
@PriorityLifecycle.on_startup(priority=10)
|
|
async def _warmup_plugin_cache():
|
|
"""预热插件快照缓存"""
|
|
logger.info("开始预热插件快照缓存...", "auth_checker_v2")
|
|
await PluginSnapshotService.warmup()
|
|
logger.info("插件快照缓存预热完成", "auth_checker_v2")
|
|
|
|
|
|
# 关闭时清理缓存
|
|
@driver.on_shutdown
|
|
async def _cleanup_cache():
|
|
"""清理快照缓存"""
|
|
AuthSnapshotService.clear_all_cache()
|
|
PluginSnapshotService.clear_all_cache()
|
|
logger.info("快照缓存已清理", "auth_checker_v2")
|