Atualizar SQSService para enviar múltiplas mensagens como uma lista e remover função de formatação de resultados não utilizada no lambda_handler, simplificando o código.

Author: maxsonferovante

aws/sqs_service.py

@@ -1,7 +1,7 @@
 from aws.boto_aws import get_instance_aws, ServiceNameAWS
 from config import config
 from botocore.exceptions import ClientError
-from typing import Dict
+from typing import Dict, List
 import json
 import logging
 import uuid
@@ -13,15 +13,16 @@ def __init__(self):
         self.aws = get_instance_aws(ServiceNameAWS.SQS)
         self.queue_url = config.QUEUE_URL        
 
-    def send_message(self, message: Dict):
+    def send_message(self, messagens: List[Dict]):
         try:
             # Add MessageGroupId for FIFO queue
             # Using order_id as MessageGroupId to ensure messages for the same order are processed in order
             message_group_id = str(uuid.uuid4())
             message_deduplication_id = str(uuid.uuid4())
+            logger.info(f"Enviando mensagem para a fila {self.queue_url} com MessageGroupId: {message_group_id} e MessageDeduplicationId: {message_deduplication_id}, com {len(messagens)} mensagens")
             response = self.aws.send_message(
                 QueueUrl=self.queue_url,
-                MessageBody=json.dumps(message),
+                MessageBody=json.dumps(messagens),
                 MessageGroupId=message_group_id,
                 MessageDeduplicationId=message_deduplication_id,
             )

lambda_function.py

@@ -78,19 +78,6 @@ def create_participant_object(participant_data):
     )
     return participant
 
-def format_result(result):
-    if isinstance(result, dict):
-        formatted = {}
-        for key, value in result.items():
-            if isinstance(value, (Participant, Certificate, Event)):
-                formatted[key] = value.model_dump()
-            elif isinstance(value, datetime):
-                formatted[key] = value.strftime('%Y-%m-%d %H:%M:%S')
-            else:
-                formatted[key] = value
-        return formatted
-    return result
-
 def lambda_handler(event, context):
     # Log the start of the Lambda execution
     try:
@@ -134,15 +121,16 @@ def lambda_handler(event, context):
             builder = CertifiedBuilder()
             certificates_results = builder.build_certificates(participants)
             # Format results before adding to response
-            formatted_results = []
+            certificates_results_messagens = []
 
             for result in certificates_results:
-                formatted_results.extend(format_result(result))            
+                
 
                 if result.get('success'):
                     s3_service.upload_file(result.get('certificate_path'), result.get('certificate_key'))
 
-                sqs_service.send_message({
+
+                certificates_results_messagens.append({                
                     "order_id": result.get('participant', {}).get('event', {}).get('order_id', ""),
                     "product_id": result.get('participant', {}).get('event', {}).get('product_id', ""),
                     "product_name": result.get('participant', {}).get('event', {}).get('product_name', ""),
@@ -151,6 +139,7 @@ def lambda_handler(event, context):
                     "success": result.get('success', False)
                 })
 
+            sqs_service.send_message(certificates_results_messagens)
 
             logger.info("Certificados gerados com sucesso")
             return {