feat: update logger logic

src/application/commands/create_payment_command.py

@@ -21,6 +21,8 @@ class CreatePaymentCommand:
         metadata = payment.metadata or {}
         order_id = str(metadata.get('order_id') or '')
         user_id = str(metadata.get('user_id') or '')
+        trace_id = str(metadata.get('trace_id') or self._logger.get_trace_id())
+        self._logger.set_trace_id(trace_id)
         if not order_id:
             raise ApplicationException(status_code=400, message='Itpay webhook metadata missing order_id')
         if not user_id:
@@ -40,7 +42,7 @@ class CreatePaymentCommand:
         message: dict[str,str] = {
             'order_id': order_id,
             'user_id': user_id,
-            'trace_id': self._logger.get_trace_id(),
+            'trace_id': trace_id,
             'message_id': message_id,
         }
         await self._queue_messanger.publish_to_queue(
@@ -48,4 +50,5 @@ class CreatePaymentCommand:
             message=message,
             message_id=message_id,
             correlation_id=message['trace_id'],
+            headers={'trace_id': trace_id},
         )