async-lambda-unstable 0.5.1__tar.gz → 0.5.2__tar.gz

{async-lambda-unstable-0.5.1 → async-lambda-unstable-0.5.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: async-lambda-unstable
-Version: 0.5.1
+Version: 0.5.2
 Summary: A framework for creating AWS Lambda Async Workflows. - Unstable Branch
 Author-email: "Nuclei, Inc" <engineering@nuclei.ai>
 Description-Content-Type: text/markdown

{async-lambda-unstable-0.5.1 → async-lambda-unstable-0.5.2}/async_lambda/__init__.py

@@ -19,4 +19,4 @@ from .models.events.managed_sqs_event import ManagedSQSEvent as ManagedSQSEvent
 from .models.events.scheduled_event import ScheduledEvent as ScheduledEvent
 from .models.events.unmanaged_sqs_event import UnmanagedSQSEvent as UnmanagedSQSEvent
 
-__version__ = "0.5.1"
+__version__ = "0.5.2"

{async-lambda-unstable-0.5.1 → async-lambda-unstable-0.5.2}/async_lambda/controller.py

@@ -525,7 +525,7 @@ class AsyncLambdaController:
                 self.set_current_lane(current_lane)
                 self.set_current_task_id(current_task_id)
         else:
-            entries = []
+            entries: List[dict] = []
             for i, sqs_payload in enumerate(sqs_payloads):
                 if isinstance(delay, Sequence):
                     _delay = delay[i]
@@ -542,13 +542,23 @@ class AsyncLambdaController:
                 url = f"https://sqs.{env.get_aws_region()}.amazonaws.com/{env.get_aws_account_id()}/{destination_task_id}"
             else:
                 url = destination_task.get_managed_queue_url(lane=lane)
-            response = get_sqs_client().send_message_batch(
-                QueueUrl=url,
-                Entries=entries,
-            )
-            if len(response.get("Failed", [])) > 0:
-                logger.error(response["Failed"])
-                raise Exception(f"Failed to send {len(response['Failed'])} messages.")
+            failed_messages = []
+            for _ in range(5):
+                response = get_sqs_client().send_message_batch(
+                    QueueUrl=url,
+                    Entries=entries,
+                )
+                failed_messages: List[dict] = response.get("Failed", [])
+                if len(failed_messages) == 0:
+                    return
+                logger.warning(failed_messages)
+                logger.warning(f"{len(failed_messages)} messages failed to send.")
+                failed_message_ids = {message["Id"] for message in failed_messages}
+                entries = [
+                    entry for entry in entries if entry["Id"] in failed_message_ids
+                ]
+            logger.error(failed_messages)
+            raise Exception(f"Failed to send {len(failed_messages)} messages.")
 
     def new_payload(
         self,

{async-lambda-unstable-0.5.1 → async-lambda-unstable-0.5.2}/async_lambda_unstable.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: async-lambda-unstable
-Version: 0.5.1
+Version: 0.5.2
 Summary: A framework for creating AWS Lambda Async Workflows. - Unstable Branch
 Author-email: "Nuclei, Inc" <engineering@nuclei.ai>
 Description-Content-Type: text/markdown