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

refactor: message counter

Browse Source
This commit is contained in:
Jonathan Kössler 2024-08-01 17:42:59 +02:00
parent 47b42e95e2
commit cb8509b120
2 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
pyinfra/queue/async_manager.py
View File

@ -164,8 +164,9 @@ class AsyncQueueManager:
await self.shutdown() await self.shutdown()
return return
self.message_count += 1
try: try:
self.message_count += 1
tenant_id = message.routing_key tenant_id = message.routing_key
filtered_message_headers = ( filtered_message_headers = (
@ -183,7 +184,6 @@ class AsyncQueueManager:
await self.publish_to_output_exchange(tenant_id, result, filtered_message_headers) await self.publish_to_output_exchange(tenant_id, result, filtered_message_headers)
await message.ack() await message.ack()
logger.debug(f"Message with {message.delivery_tag=} acknowledged.") logger.debug(f"Message with {message.delivery_tag=} acknowledged.")
self.message_count -= 1
else: else:
raise ValueError(f"Could not process message with {message.body=}.") raise ValueError(f"Could not process message with {message.body=}.")
@ -197,6 +197,8 @@ class AsyncQueueManager:
await message.nack(requeue=False) await message.nack(requeue=False)
logger.error(f"Error processing input message: {e}", exc_info=True) logger.error(f"Error processing input message: {e}", exc_info=True)
raise raise
finally:
self.message_count -= 1
async def publish_to_output_exchange(self, tenant_id: str, result: Dict[str, Any], headers: Dict[str, Any]) -> None: async def publish_to_output_exchange(self, tenant_id: str, result: Dict[str, Any], headers: Dict[str, Any]) -> None:
await self.output_exchange.publish( await self.output_exchange.publish(

2
pyproject.toml
View File

@ -1,6 +1,6 @@
[tool.poetry] [tool.poetry]
name = "pyinfra" name = "pyinfra"
version = "3.0.0" version = "3.1.0"
description = "" description = ""
authors = ["Team Research <research@knecon.com>"] authors = ["Team Research <research@knecon.com>"]
license = "All rights reseverd" license = "All rights reseverd"