Marmelator/pyinfra
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 36 Wiki Activity

refactor: raise error if tenant service is not available

Browse Source
This commit is contained in:
Jonathan Kössler 2024-11-12 13:30:21 +01:00
parent 9906f68e0a
commit ed2bd1ec86
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
pyinfra/queue/async_manager.py
View File

@ -273,15 +273,20 @@ class AsyncQueueManager:
@retry(
tries=5,
exceptions=(AMQPConnectionError, ChannelInvalidStateError),
exceptions=(
AMQPConnectionError,
ChannelInvalidStateError,
aiohttp.ClientResponseError,
aiohttp.ClientConnectorError,
),
reraise=True,
)
async def initialize_tenant_queues(self) -> None:
try:
active_tenants = await self.fetch_active_tenants()
except (aiohttp.ClientResponseError, aiohttp.ClientConnectorError):
logger.warning("API calls to tenant server failed. No tenant queues initialized.", exc_info=True)
active_tenants = set()
logger.error("API calls to tenant server failed. No tenant queues initialized.", exc_info=True)
raise asyncio.CancelledError
for tenant_id in active_tenants:
await self.create_tenant_queues(tenant_id)