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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: async-lambda-unstable
-Version: 0.6.9
+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.9 → 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.9"
+__version__ = "0.6.11"

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

@@ -431,6 +431,12 @@ class AsyncLambdaController:
                             "ASYNC_LAMBDA_DELAY_SCHEDULE_GROUP": {
                                 "Ref": "AsyncLambdaDelayScheduleGroup"
                             },
+                            "ASYNC_LAMBDA_DELAY_SCHEDULE_DLQ_ARN": {
+                                "Fn::GetAtt": [
+                                    "AsyncLambdaDelayDLQ",
+                                    "Arn",
+                                ]
+                            },
                             **build_config.environment_variables,
                         },
                     },
@@ -1036,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

{async_lambda_unstable-0.6.9 → async_lambda_unstable-0.6.11}/async_lambda/models/mock/mock_event.py

@@ -1,6 +1,7 @@
 from time import time
 from typing import List, Optional, Tuple
 from urllib.parse import parse_qs, parse_qsl
+from uuid import uuid4
 
 
 def MockSQSLambdaEvent(
@@ -27,7 +28,7 @@ def MockSQSLambdaEvent(
     return {
         "Records": [
             {
-                "messageId": "059f36b4-87a3-44ab-83d2-661975830a7d",
+                "messageId": str(uuid4()),
                 "receiptHandle": "AQEBwJnKyrHigUMZj6rYigCgxlaS3SLy0a",
                 "body": body,
                 "attributes": {
@@ -107,7 +108,7 @@ def MockSQSBatchLambdaEvent(
     for body in bodies:
         records.append(
             {
-                "messageId": "059f36b4-87a3-44ab-83d2-661975830a7e",
+                "messageId": str(uuid4()),
                 "receiptHandle": "AQEBwJnKyrHigUMZj6rYigCgxlaS3SLy0b",
                 "body": body,
                 "attributes": {