zlei6/ragflow
1
0
Fork 0
mirror of https://github.com/infiniflow/ragflow.git synced 2026-08-08 16:38:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix: SIGTERM not handle well in ragflow server chat channel (#17828)

Browse Source
This commit is contained in:
Wang Qi
2026-08-05 11:52:53 +08:00
committed by GitHub
parent 9b05e5c67e
commit f057544e31
2 changed files with 25 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
api/channels/bootstrap.py
View File

@@ -219,13 +219,17 @@ async def _start_channel(running: dict, account_id: str, channel: str, credentia
return True
async def _reconcile(running: dict, failed: dict) -> None:
async def _reconcile(running: dict, failed: dict, stop_event: threading.Event) -> None:
"""Diff desired (DB) vs running channels and apply start/stop/restart.
``failed`` remembers configs that could not be started so they are not
retried (and re-logged) every tick until their credentials change.
"""
if stop_event.is_set():
return
desired = await asyncio.to_thread(_desired_channels)
if stop_event.is_set():
return
# Stop channels that were removed or whose credentials/type changed.
for account_id in list(running.keys()):
@@ -266,7 +270,12 @@ async def run_channels(stop_event: threading.Event) -> None:
try:
while not stop_event.is_set():
try:
await _reconcile(running, failed)
await _reconcile(running, failed, stop_event)
except RuntimeError as ex:
if stop_event.is_set():
LOGGER.info("chat channel reconcile stopped")
break
LOGGER.error("chat channel reconcile failed: %s", ex)
except Exception as ex:
LOGGER.error("chat channel reconcile failed: %s", ex)