wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-30 00:00:26 +08:00
Code Issues Actions 20 Packages Projects Releases Wiki Activity

More graceful health check handling of this connection not being ready

Browse Source
This commit is contained in:
doctorpangloss 2025-02-06 11:08:09 -08:00
parent 4c72ef5bac
commit ef74b9fdda
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/distributed/distributed_prompt_worker.py
View File

@ -42,7 +42,7 @@ class DistributedPromptWorker:
self._health_check_site: Optional[web.TCPSite] = None self._health_check_site: Optional[web.TCPSite] = None
async def _health_check(self, request): async def _health_check(self, request):
if self._connection is None: if not hasattr(self, "_connection") or self._connection is None:
return web.Response(text="UNHEALTHY: RabbitMQ connection is not established", status=503) return web.Response(text="UNHEALTHY: RabbitMQ connection is not established", status=503)
is_healthy = await self._is_connection_healthy() is_healthy = await self._is_connection_healthy()