gardel/Mofox-Core
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(emoji): 使用QueryBuilder优化数据库查询并启用缓存

Browse Source
This commit is contained in:
Windpicker-owo
2025-11-03 15:36:30 +08:00
parent ecef8edd28
commit f822bfb6ee
2 changed files with 20 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
src/chat/emoji_system/emoji_manager.py
View File

@@ -678,12 +678,13 @@ class EmojiManager:
async def get_all_emoji_from_db(self) -> None: async def get_all_emoji_from_db(self) -> None:
"""获取所有表情包并初始化为MaiEmoji类对象，更新 self.emoji_objects""" """获取所有表情包并初始化为MaiEmoji类对象，更新 self.emoji_objects"""
try: try:
async with get_db_session() as session: # 🔧 使用 QueryBuilder 以启用数据库缓存
logger.debug("[数据库] 开始加载所有表情包记录 ...") from src.common.database.api.query import QueryBuilder
result = await session.execute(select(Emoji)) logger.debug("[数据库] 开始加载所有表情包记录 ...")
emoji_instances = result.scalars().all()
emoji_objects, load_errors = _to_emoji_objects(emoji_instances) emoji_instances = await QueryBuilder(Emoji).all()
emoji_objects, load_errors = _to_emoji_objects(emoji_instances)
# 更新内存中的列表和数量 # 更新内存中的列表和数量
self.emoji_objects = emoji_objects self.emoji_objects = emoji_objects
@@ -798,12 +799,11 @@ class EmojiManager:
logger.info(f"[缓存命中] 从内存获取表情包描述: {emoji.description[:50]}...") logger.info(f"[缓存命中] 从内存获取表情包描述: {emoji.description[:50]}...")
return emoji.description return emoji.description
# 如果内存中没有，从数据库查找 # 如果内存中没有，从数据库查找（使用 QueryBuilder 启用数据库缓存）
try: try:
async with get_db_session() as session: from src.common.database.api.query import QueryBuilder
stmt = select(Emoji).where(Emoji.emoji_hash == emoji_hash)
result = await session.execute(stmt) emoji_record = await QueryBuilder(Emoji).filter(emoji_hash=emoji_hash).first()
emoji_record = result.scalar_one_or_none()
if emoji_record and emoji_record.description: if emoji_record and emoji_record.description:
logger.info(f"[缓存命中] 从数据库获取表情包描述: {emoji_record.description[:50]}...") logger.info(f"[缓存命中] 从数据库获取表情包描述: {emoji_record.description[:50]}...")
return emoji_record.description return emoji_record.description
@@ -962,16 +962,15 @@ class EmojiManager:
image_hash = hashlib.md5(image_bytes).hexdigest() image_hash = hashlib.md5(image_bytes).hexdigest()
image_format = (Image.open(io.BytesIO(image_bytes)).format or "jpeg").lower() image_format = (Image.open(io.BytesIO(image_bytes)).format or "jpeg").lower()
# 2. 检查数据库中是否已存在该表情包的描述，实现复用 # 2. 检查数据库中是否已存在该表情包的描述，实现复用（使用 QueryBuilder 启用数据库缓存）
existing_description = None existing_description = None
try: try:
async with get_db_session() as session: from src.common.database.api.query import QueryBuilder
stmt = select(Images).where(Images.emoji_hash == image_hash, Images.type == "emoji")
result = await session.execute(stmt) existing_image = await QueryBuilder(Images).filter(emoji_hash=image_hash, type="emoji").first()
existing_image = result.scalar_one_or_none() if existing_image and existing_image.description:
if existing_image and existing_image.description: existing_description = existing_image.description
existing_description = existing_image.description logger.info(f"[复用描述] 找到已有详细描述: {existing_description[:50]}...")
logger.info(f"[复用描述] 找到已有详细描述: {existing_description[:50]}...")
except Exception as e: except Exception as e:
logger.debug(f"查询已有表情包描述时出错: {e}") logger.debug(f"查询已有表情包描述时出错: {e}")

4
src/common/database/api/specialized.py
View File

@@ -251,7 +251,7 @@ async def update_person_affinity(
# ===== ChatStreams 业务API ===== # ===== ChatStreams 业务API =====
@cached(ttl=300, key_prefix="chat_stream") # 缓存5分钟 @cached(ttl=600, key_prefix="chat_stream") # 缓存10分钟
async def get_or_create_chat_stream( async def get_or_create_chat_stream(
stream_id: str, stream_id: str,
platform: str, platform: str,
@@ -401,7 +401,7 @@ async def get_usage_statistics(
# ===== UserRelationships 业务API ===== # ===== UserRelationships 业务API =====
@cached(ttl=300, key_prefix="user_relationship") # 缓存5分钟 @cached(ttl=600, key_prefix="user_relationship") # 缓存10分钟
async def get_user_relationship( async def get_user_relationship(
platform: str, platform: str,
user_id: str, user_id: str,