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

优化消息管理中的异步任务处理

Browse Source
This commit is contained in:
Windpicker-owo
2025-12-13 21:06:57 +08:00
parent 8366d5aaad
commit ee30fa5d1d
1 changed files with 5 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
src/chat/message_manager/message_manager.py
View File

@@ -109,15 +109,11 @@ class MessageManager:
context = chat_stream.context context = chat_stream.context
if not (context.stream_loop_task and not context.stream_loop_task.done()): if not (context.stream_loop_task and not context.stream_loop_task.done()):
# 异步启动驱动器任务；避免在高并发下阻塞消息入队 # 异步启动驱动器任务；避免在高并发下阻塞消息入队
context.stream_loop_task = asyncio.create_task( await stream_loop_manager.start_stream_loop(stream_id)
stream_loop_manager.start_stream_loop(stream_id)
) # 检查并处理消息打断
await self._check_and_handle_interruption(chat_stream, message)
# 并行触发打断检查，不阻塞消息入队
context.interruption_task = asyncio.create_task(
self._check_and_handle_interruption(chat_stream, message)
)
# 入队消息 # 入队消息
await chat_stream.context.add_message(message) await chat_stream.context.add_message(message)