async-lambda-unstable 0.6.10__tar.gz → 0.6.11__tar.gz

{async_lambda_unstable-0.6.10 → async_lambda_unstable-0.6.11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: async-lambda-unstable
-Version: 0.6.10
+Version: 0.6.11
 Summary: A framework for creating AWS Lambda Async Workflows. - Unstable Branch
 Author-email: "Nuclei, Inc" <engineering@nuclei.ai>
 Requires-Dist: click>=8.0.0

{async_lambda_unstable-0.6.10 → async_lambda_unstable-0.6.11}/async_lambda/__init__.py

@@ -21,4 +21,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.6.10"
+__version__ = "0.6.11"

{async_lambda_unstable-0.6.10 → async_lambda_unstable-0.6.11}/async_lambda/controller.py

@@ -1042,12 +1042,24 @@ class AsyncLambdaController:
             delay (int): Delay in seconds before delivering the message.
             message_group_id (Optional[str]): MessageGroupId for FIFO queues. Defaults to None.
         """
-        schedule_time = datetime.now(tz=timezone.utc) + timedelta(seconds=delay)
+        if delay <= _SQS_MAX_DELAY_SECONDS:
+            raise ValueError(
+                f"Delays with EventBridge Scheduler must be greater than {_SQS_MAX_DELAY_SECONDS} seconds. Use SQS delay instead."
+            )
+
+        # Schedule names are unique and ActionAfterCompletion="DELETE" is an asynchronous action. This can take a few
+        # seconds, during which time, re-scheduling yourself on the same schedule_name would fail. Utilizing full
+        # _SQS_MAX_DELAY_SECONDS as a buffer in case AWS ActionAfterCompletion becomes delayed.
+        scheduler_delay = delay - _SQS_MAX_DELAY_SECONDS
+        schedule_time = datetime.now(tz=timezone.utc) + timedelta(
+            seconds=scheduler_delay
+        )
         schedule_expression = f"at({schedule_time.strftime('%Y-%m-%dT%H:%M:%S')})"
 
         sqs_input: dict = {
             "QueueUrl": queue_url,
             "MessageBody": message_body,
+            "DelaySeconds": _SQS_MAX_DELAY_SECONDS,
         }
         if message_group_id:
             sqs_input["MessageGroupId"] = message_group_id