BruceSun
/
MaiBot
mirror of https://github.com/Mai-with-u/MaiBot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix(chat): Improve error handling and logging in heartFC_chat

pull/1178/head
Eric-Terminal 2025-08-12 20:54:30 +08:00
parent 04bd05c1fe
commit c03a7f71c2
1 changed files with 21 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
src/chat/chat_loop/heartFC_chat.py
View File

@ -109,6 +109,10 @@ class HeartFChatting:
self.reply_timeout_count = 0 self.reply_timeout_count = 0
self.plan_timeout_count = 0 self.plan_timeout_count = 0
# 重启控制
self._restart_attempts = 0
self._max_restart_attempts = 5
self.last_read_time = time.time() - 1 self.last_read_time = time.time() - 1
self.focus_energy = 1 self.focus_energy = 1
@ -313,7 +317,7 @@ class HeartFChatting:
platform = getattr(self.chat_stream, "platform", "unknown") platform = getattr(self.chat_stream, "platform", "unknown")
person = Person(platform = platform ,user_id = action_message.get("user_id", "")) person = Person(platform = platform ,user_id = action_message.get("user_id", ""))
person_name = person.person_name person_name = person.person_name or action_message.get("user_nickname") or action_message.get("user_id", "未知用户")
action_prompt_display = f"你对{person_name}进行了回复：{reply_text}" action_prompt_display = f"你对{person_name}进行了回复：{reply_text}"
await database_api.store_action_info( await database_api.store_action_info(
@ -612,6 +616,7 @@ class HeartFChatting:
async def _main_chat_loop(self): async def _main_chat_loop(self):
"""主循环，持续进行计划并可能回复消息，直到被外部取消。""" """主循环，持续进行计划并可能回复消息，直到被外部取消。"""
self._restart_attempts = 0 # 成功启动时重置计数器
try: try:
while self.running: while self.running:
# 主循环 # 主循环
@ -623,11 +628,21 @@ class HeartFChatting:
# 设置了关闭标志位后被取消是正常流程 # 设置了关闭标志位后被取消是正常流程
logger.info(f"{self.log_prefix} 麦麦已关闭聊天") logger.info(f"{self.log_prefix} 麦麦已关闭聊天")
except Exception: except Exception:
logger.error(f"{self.log_prefix} 麦麦聊天意外错误，将于3s后尝试重新启动") self._restart_attempts += 1
print(traceback.format_exc()) if self._restart_attempts > self._max_restart_attempts:
await asyncio.sleep(3) logger.error(f"{self.log_prefix} 麦麦聊天意外错误，达到最大重启次数 {self._max_restart_attempts}，停止重启")
self._loop_task = asyncio.create_task(self._main_chat_loop()) print(traceback.format_exc())
logger.error(f"{self.log_prefix} 结束了当前聊天循环") else:
wait_time = 2**self._restart_attempts # 指数退避
logger.error(
f"{self.log_prefix} 麦麦聊天意外错误，将于{wait_time}s后尝试重新启动 "
f"(第 {self._restart_attempts}/{self._max_restart_attempts} 次)"
)
print(traceback.format_exc())
await asyncio.sleep(wait_time)
self._loop_task = asyncio.create_task(self._main_chat_loop())
finally:
logger.info(f"{self.log_prefix} 结束了当前聊天循环")
async def _handle_action( async def _handle_action(
self, self,