mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2026-09-29 00:32:06 +08:00
29 lines
1.0 KiB
Python
29 lines
1.0 KiB
Python
"""
|
|
缓存初始化模块
|
|
|
|
负责注册各种缓存类型，实现按需缓存机制
|
|
"""
|
|
|
|
from zhenxun.models.bot_console import BotConsole
|
|
from zhenxun.models.level_user import LevelUser
|
|
from zhenxun.models.plugin_info import PluginInfo
|
|
from zhenxun.models.user_console import UserConsole
|
|
from zhenxun.services.cache import CacheRegistry, cache_config
|
|
from zhenxun.services.cache.config import CacheMode
|
|
from zhenxun.services.log import logger
|
|
from zhenxun.utils.enum import CacheType
|
|
|
|
|
|
# 注册缓存类型
|
|
def register_cache_types():
|
|
"""注册所有缓存类型"""
|
|
CacheRegistry.register(CacheType.PLUGINS, PluginInfo)
|
|
CacheRegistry.register(CacheType.BOT, BotConsole)
|
|
CacheRegistry.register(CacheType.USERS, UserConsole)
|
|
CacheRegistry.register(
|
|
CacheType.LEVEL, LevelUser, key_format="{user_id}_{group_id}"
|
|
)
|
|
|
|
if cache_config.cache_mode == CacheMode.REDIS and cache_config.redis_host:
|
|
logger.info(f"已注册 Redis 模型缓存类型，缓存模式: {cache_config.cache_mode}")
|