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

refactor(chat): 将消息获取相关函数重构为异步

Browse Source 
将 `chat_message_builder` 中的多个同步消息获取函数（如 `get_raw_msg_by_timestamp`）及其调用全部修改为异步函数。这统一了数据库查询的异步模式，提高了代码一致性和可维护性。

主要改动包括：
- 将 `chat_message_builder.py` 中的数据库查询函数标记为 `async` 并使用 `await`。
- 更新了 `message_api.py`、`mood_manager.py` 和 `qzone_service.py` 中对这些函数的调用，以适应异步接口。
- 调整了 `message_api.py` 中的函数签名和返回类型提示，以反映异步特性。
This commit is contained in:
minecraft1024a
2025-10-02 17:32:02 +08:00
committed by Windpicker-owo
parent 00a9a71702
commit 944876227a
4 changed files with 44 additions and 40 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
src/chat/utils/chat_message_builder.py
View File

@@ -466,9 +466,7 @@ async def get_raw_msg_before_timestamp_with_chat(chat_id: str, timestamp: float,
return await find_messages(message_filter=filter_query, sort=sort_order, limit=limit)
async def get_raw_msg_before_timestamp_with_users(
timestamp: float, person_ids: list, limit: int = 0
) -> List[Dict[str, Any]]:
async def get_raw_msg_before_timestamp_with_users(timestamp: float, person_ids: list, limit: int = 0) -> List[Dict[str, Any]]:
"""获取指定时间戳之前的消息，按时间升序排序，返回消息列表
limit: 限制返回的消息数量，0为不限制
"""
@@ -644,7 +642,7 @@ async def _build_readable_messages_internal(
person_name = f"{person_name}({user_id})"
# 使用独立函数处理用户引用格式
content = replace_user_references_sync(content, platform, replace_bot_name=replace_bot_name)
content = await replace_user_references_async(content, platform, replace_bot_name=replace_bot_name)
target_str = "这是QQ的一个功能，用于提及某人，但没那么明显"
if target_str in content and random.random() < 0.6:
@@ -1221,13 +1219,15 @@ async def build_anonymous_messages(messages: List[Dict[str, Any]]) -> str:
# print(f"anon_name:{anon_name}")
# 使用独立函数处理用户引用格式，传入自定义的匿名名称解析器
def anon_name_resolver(platform: str, user_id: str) -> str:
async def anon_name_resolver(platform: str, user_id: str) -> str:
try:
return get_anon_name(platform, user_id)
except Exception:
return "?"
content = replace_user_references_sync(content, platform, anon_name_resolver, replace_bot_name=False)
content = await replace_user_references_async(
content, platform, anon_name_resolver, replace_bot_name=False
)
header = f"{anon_name}说 "
output_lines.append(header)

2
src/mood/mood_manager.py
View File

@@ -103,7 +103,7 @@ class ChatMood:
logger.debug(
f"{self.log_prefix} 更新情绪状态，感兴趣度: {interested_rate:.2f}, 更新概率: {update_probability:.2f}"
)
message_list_before_now = get_raw_msg_by_timestamp_with_chat_inclusive(
message_list_before_now = await get_raw_msg_by_timestamp_with_chat_inclusive(
chat_id=self.chat_id,
timestamp_start=self.last_change_time,
timestamp_end=message_time,

68
src/plugin_system/apis/message_api.py
View File

@@ -34,9 +34,9 @@ from src.chat.utils.chat_message_builder import (
# =============================================================================
def get_messages_by_time(
async def get_messages_by_time(
start_time: float, end_time: float, limit: int = 0, limit_mode: str = "latest", filter_mai: bool = False
) -> Coroutine[Any, Any, list[dict[str, Any]]]:
) -> List[Dict[str, Any]]:
"""
获取指定时间范围内的消息
@@ -58,8 +58,8 @@ def get_messages_by_time(
if limit < 0:
raise ValueError("limit 不能为负数")
if filter_mai:
return filter_mai_messages(get_raw_msg_by_timestamp(start_time, end_time, limit, limit_mode))
return get_raw_msg_by_timestamp(start_time, end_time, limit, limit_mode)
return await filter_mai_messages(await get_raw_msg_by_timestamp(start_time, end_time, limit, limit_mode))
return await get_raw_msg_by_timestamp(start_time, end_time, limit, limit_mode)
async def get_messages_by_time_in_chat(
@@ -148,14 +148,14 @@ async def get_messages_by_time_in_chat_inclusive(
)
def get_messages_by_time_in_chat_for_users(
async def get_messages_by_time_in_chat_for_users(
chat_id: str,
start_time: float,
end_time: float,
person_ids: List[str],
limit: int = 0,
limit_mode: str = "latest",
) -> Coroutine[Any, Any, list[dict[str, Any]]]:
) -> List[Dict[str, Any]]:
"""
获取指定聊天中指定用户在指定时间范围内的消息
@@ -181,12 +181,14 @@ def get_messages_by_time_in_chat_for_users(
raise ValueError("chat_id 不能为空")
if not isinstance(chat_id, str):
raise ValueError("chat_id 必须是字符串类型")
return get_raw_msg_by_timestamp_with_chat_users(chat_id, start_time, end_time, person_ids, limit, limit_mode)
return await get_raw_msg_by_timestamp_with_chat_users(
chat_id, start_time, end_time, person_ids, limit, limit_mode
)
def get_random_chat_messages(
async def get_random_chat_messages(
start_time: float, end_time: float, limit: int = 0, limit_mode: str = "latest", filter_mai: bool = False
) -> Coroutine[Any, Any, list[dict[str, Any]]]:
) -> List[Dict[str, Any]]:
"""
随机选择一个聊天，返回该聊天在指定时间范围内的消息
@@ -208,13 +210,13 @@ def get_random_chat_messages(
if limit < 0:
raise ValueError("limit 不能为负数")
if filter_mai:
return filter_mai_messages(get_raw_msg_by_timestamp_random(start_time, end_time, limit, limit_mode))
return get_raw_msg_by_timestamp_random(start_time, end_time, limit, limit_mode)
return await filter_mai_messages(await get_raw_msg_by_timestamp_random(start_time, end_time, limit, limit_mode))
return await get_raw_msg_by_timestamp_random(start_time, end_time, limit, limit_mode)
def get_messages_by_time_for_users(
async def get_messages_by_time_for_users(
start_time: float, end_time: float, person_ids: List[str], limit: int = 0, limit_mode: str = "latest"
) -> Coroutine[Any, Any, list[dict[str, Any]]]:
) -> List[Dict[str, Any]]:
"""
获取指定用户在所有聊天中指定时间范围内的消息
@@ -235,11 +237,10 @@ def get_messages_by_time_for_users(
raise ValueError("start_time 和 end_time 必须是数字类型")
if limit < 0:
raise ValueError("limit 不能为负数")
return get_raw_msg_by_timestamp_with_users(start_time, end_time, person_ids, limit, limit_mode)
return await get_raw_msg_by_timestamp_with_users(start_time, end_time, person_ids, limit, limit_mode)
def get_messages_before_time(timestamp: float, limit: int = 0, filter_mai: bool = False) -> Coroutine[
Any, Any, list[dict[str, Any]]]:
async def get_messages_before_time(timestamp: float, limit: int = 0, filter_mai: bool = False) -> List[Dict[str, Any]]:
"""
获取指定时间戳之前的消息
@@ -259,8 +260,8 @@ def get_messages_before_time(timestamp: float, limit: int = 0, filter_mai: bool
if limit < 0:
raise ValueError("limit 不能为负数")
if filter_mai:
return filter_mai_messages(get_raw_msg_before_timestamp(timestamp, limit))
return get_raw_msg_before_timestamp(timestamp, limit)
return await filter_mai_messages(await get_raw_msg_before_timestamp(timestamp, limit))
return await get_raw_msg_before_timestamp(timestamp, limit)
async def get_messages_before_time_in_chat(
@@ -294,8 +295,9 @@ async def get_messages_before_time_in_chat(
return await get_raw_msg_before_timestamp_with_chat(chat_id, timestamp, limit)
def get_messages_before_time_for_users(timestamp: float, person_ids: List[str], limit: int = 0) -> Coroutine[
Any, Any, list[dict[str, Any]]]:
async def get_messages_before_time_for_users(
timestamp: float, person_ids: List[str], limit: int = 0
) -> List[Dict[str, Any]]:
"""
获取指定用户在指定时间戳之前的消息
@@ -314,12 +316,12 @@ def get_messages_before_time_for_users(timestamp: float, person_ids: List[str],
raise ValueError("timestamp 必须是数字类型")
if limit < 0:
raise ValueError("limit 不能为负数")
return get_raw_msg_before_timestamp_with_users(timestamp, person_ids, limit)
return await get_raw_msg_before_timestamp_with_users(timestamp, person_ids, limit)
def get_recent_messages(
async def get_recent_messages(
chat_id: str, hours: float = 24.0, limit: int = 100, limit_mode: str = "latest", filter_mai: bool = False
) -> Coroutine[Any, Any, list[dict[str, Any]]]:
) -> List[Dict[str, Any]]:
"""
获取指定聊天中最近一段时间的消息
@@ -347,8 +349,10 @@ def get_recent_messages(
now = time.time()
start_time = now - hours * 3600
if filter_mai:
return filter_mai_messages(get_raw_msg_by_timestamp_with_chat(chat_id, start_time, now, limit, limit_mode))
return get_raw_msg_by_timestamp_with_chat(chat_id, start_time, now, limit, limit_mode)
return await filter_mai_messages(
await get_raw_msg_by_timestamp_with_chat(chat_id, start_time, now, limit, limit_mode)
)
return await get_raw_msg_by_timestamp_with_chat(chat_id, start_time, now, limit, limit_mode)
# =============================================================================
@@ -356,8 +360,7 @@ def get_recent_messages(
# =============================================================================
def count_new_messages(chat_id: str, start_time: float = 0.0, end_time: Optional[float] = None) -> Coroutine[
Any, Any, int]:
async def count_new_messages(chat_id: str, start_time: float = 0.0, end_time: Optional[float] = None) -> int:
"""
计算指定聊天中从开始时间到结束时间的新消息数量
@@ -378,11 +381,12 @@ def count_new_messages(chat_id: str, start_time: float = 0.0, end_time: Optional
raise ValueError("chat_id 不能为空")
if not isinstance(chat_id, str):
raise ValueError("chat_id 必须是字符串类型")
return num_new_messages_since(chat_id, start_time, end_time)
return await num_new_messages_since(chat_id, start_time, end_time)
def count_new_messages_for_users(chat_id: str, start_time: float, end_time: float, person_ids: List[str]) -> Coroutine[
Any, Any, int]:
async def count_new_messages_for_users(
chat_id: str, start_time: float, end_time: float, person_ids: List[str]
) -> int:
"""
计算指定聊天中指定用户从开始时间到结束时间的新消息数量
@@ -404,7 +408,7 @@ def count_new_messages_for_users(chat_id: str, start_time: float, end_time: floa
raise ValueError("chat_id 不能为空")
if not isinstance(chat_id, str):
raise ValueError("chat_id 必须是字符串类型")
return num_new_messages_since_with_users(chat_id, start_time, end_time, person_ids)
return await num_new_messages_since_with_users(chat_id, start_time, end_time, person_ids)
# =============================================================================
@@ -420,7 +424,7 @@ async def build_readable_messages_to_str(
read_mark: float = 0.0,
truncate: bool = False,
show_actions: bool = False,
) -> Coroutine[Any, Any, str]:
) -> str:
"""
将消息列表构建成可读的字符串

2
src/plugins/built_in/maizone_refactored/services/qzone_service.py
View File

@@ -221,7 +221,7 @@ class QZoneService:
for chat_id in group.chat_ids:
# 使用正确的函数获取历史消息
messages = get_raw_msg_by_timestamp_with_chat(
messages = await get_raw_msg_by_timestamp_with_chat(
chat_id=chat_id,
timestamp_start=start_time,
timestamp_end=end_time,