fix potential log excheptions

Lancetnik · May 22, 2023 · 921248b · 921248b
1 parent 0877ceb
commit 921248b
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 4 deletions.
diff --git a/propan/__about__.py b/propan/__about__.py
@@ -1,3 +1,3 @@
 """Simple and fast framework to create message brokers based microservices"""
 
-__version__ = "0.1.1.3"
+__version__ = "0.1.1.4"
diff --git a/propan/brokers/model/broker_usecase.py b/propan/brokers/model/broker_usecase.py
@@ -205,15 +205,15 @@ async def wrapper(message: PropanMessage) -> T:
                 log_context = self._get_log_context(message=message, **broker_args)
 
                 with context.scope("log_context", log_context):
-                    self._log("Received")
+                    self._log("Received", extra=log_context)
 
                     try:
                         r = await func(message)
                     except Exception as e:
                         self._log(repr(e), logging.ERROR)
                         raise e
                     else:
-                        self._log("Processed")
+                        self._log("Processed", extra=log_context)
                         return r
 
             return wrapper

diff --git a/propan/brokers/rabbit/rabbit_broker.py b/propan/brokers/rabbit/rabbit_broker.py
@@ -64,7 +64,8 @@ async def _connect(
             self._channel = await connection.channel()
 
             if max_consumers:
-                self._log(f"Set max consumers to {max_consumers}")
+                c = self._get_log_context(None, RabbitQueue(""), RabbitExchange(""))
+                self._log(f"Set max consumers to {max_consumers}", extra=c)
                 await self._channel.set_qos(prefetch_count=int(self._max_consumers))
 
         return connection