async-lambda-unstable 0.2.10__tar.gz → 0.2.12__tar.gz

{async-lambda-unstable-0.2.10 → async-lambda-unstable-0.2.12}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: async-lambda-unstable
-Version: 0.2.10
+Version: 0.2.12
 Summary: A framework for creating AWS Lambda Async Workflows. - Unstable Branch
 Author-email: "Nuclei, Inc" <engineering@nuclei.ai>
 Description-Content-Type: text/markdown
@@ -46,6 +46,7 @@ All task decorators share common arguments for configuring the underlying lambda
 - `memory: int = 128` Sets the memory allocation for the function.
 - `timeout: int = 60` Sets the timeout for the function (max 900 seconds).
 - `ephemeral_storage: int = 512` Sets the ephemeral storage allocation for the function.
+- `maximum_concurrency: Optional[int] = None` Sets the maximum concurrency value for the SQS trigger for the function. (only applies to `async_task` and `sqs_task` tasks.)
 
 ## Async Task
 
@@ -172,11 +173,11 @@ These environment variables will also be available during build time.
 
 [The value is passed to the `Environment` property on `SAM::Serverless::Function`](https://docs.aws.amazon.com/serverless-application-model/latest/developerguide/sam-resource-function.html#sam-function-environment)
 
-## `policy_arns`
+## `policies`
 
 ```
 [
-    'IAM_POLICY_ARN'
+    'IAM_POLICY_ARN' | STATEMENT
 ]
 ```
 

{async-lambda-unstable-0.2.10 → async-lambda-unstable-0.2.12}/README.md

@@ -39,6 +39,7 @@ All task decorators share common arguments for configuring the underlying lambda
 - `memory: int = 128` Sets the memory allocation for the function.
 - `timeout: int = 60` Sets the timeout for the function (max 900 seconds).
 - `ephemeral_storage: int = 512` Sets the ephemeral storage allocation for the function.
+- `maximum_concurrency: Optional[int] = None` Sets the maximum concurrency value for the SQS trigger for the function. (only applies to `async_task` and `sqs_task` tasks.)
 
 ## Async Task
 
@@ -165,11 +166,11 @@ These environment variables will also be available during build time.
 
 [The value is passed to the `Environment` property on `SAM::Serverless::Function`](https://docs.aws.amazon.com/serverless-application-model/latest/developerguide/sam-resource-function.html#sam-function-environment)
 
-## `policy_arns`
+## `policies`
 
 ```
 [
-    'IAM_POLICY_ARN'
+    'IAM_POLICY_ARN' | STATEMENT
 ]
 ```
 

{async-lambda-unstable-0.2.10 → async-lambda-unstable-0.2.12}/async_lambda/__init__.py

@@ -9,4 +9,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.2.10"
+__version__ = "0.2.12"

{async-lambda-unstable-0.2.10 → async-lambda-unstable-0.2.12}/async_lambda/build_config.py

@@ -1,11 +1,11 @@
 from dataclasses import dataclass
-from typing import Dict, List, Optional, Set
+from typing import Dict, List, Optional, Set, Union
 
 
 @dataclass
 class AsyncLambdaBuildConfig:
     environment_variables: Dict[str, str]
-    policy_arns: Set[str]
+    policies: List[Union[str, dict]]
     layers: List[str]
     subnet_ids: Set[str]
     security_group_ids: Set[str]
@@ -17,7 +17,7 @@ class AsyncLambdaBuildConfig:
     @classmethod
     def new(cls, config: dict) -> "AsyncLambdaBuildConfig":
         return cls(
-            policy_arns=set(config.get("policy_arns", set())),
+            policies=list(config.get("policies", list())),
             environment_variables=config.get("environment_variables", dict()),
             layers=list(config.get("layers", list())),
             subnet_ids=set(config.get("subnet_ids", set())),
@@ -29,7 +29,7 @@ class AsyncLambdaBuildConfig:
         )
 
     def merge(self, other: "AsyncLambdaBuildConfig"):
-        self.policy_arns.update(other.policy_arns)
+        self.policies += other.policies
         self.environment_variables.update(other.environment_variables)
         self.layers = list(dict.fromkeys(self.layers + other.layers))
         self.subnet_ids.update(other.subnet_ids)

{async-lambda-unstable-0.2.10 → async-lambda-unstable-0.2.12}/async_lambda/models/task.py

@@ -31,6 +31,7 @@ class AsyncLambdaTask(Generic[EventType]):
     timeout: int
     memory: int
     ephemeral_storage: int
+    maximum_concurrency: Optional[int]
 
     executable: Callable[[EventType], Any]
 
@@ -43,6 +44,7 @@ class AsyncLambdaTask(Generic[EventType]):
         timeout: int = 60,
         memory: int = 128,
         ephemeral_storage: int = 512,
+        maximum_concurrency: Optional[int] = None,
     ):
         AsyncLambdaTask.validate_task_id(task_id)
         self.executable = executable
@@ -52,6 +54,7 @@ class AsyncLambdaTask(Generic[EventType]):
         self.timeout = timeout
         self.memory = memory
         self.ephemeral_storage = ephemeral_storage
+        self.maximum_concurrency = maximum_concurrency
 
     @staticmethod
     def validate_task_id(task_id: str):
@@ -93,6 +96,11 @@ class AsyncLambdaTask(Generic[EventType]):
         return f"{self.get_function_logical_id()}Extra{index}"
 
     def get_template_events(self):
+        sqs_properties = {}
+        if self.maximum_concurrency is not None:
+            sqs_properties["ScalingConfig"] = {
+                "MaximumConcurrency": self.maximum_concurrency
+            }
         if self.trigger_type == TaskTriggerType.MANAGED_SQS:
             return {
                 "ManagedSQS": {
@@ -106,6 +114,7 @@ class AsyncLambdaTask(Generic[EventType]):
                                 "Arn",
                             ]
                         },
+                        **sqs_properties,
                     },
                 }
             }
@@ -117,6 +126,7 @@ class AsyncLambdaTask(Generic[EventType]):
                         "BatchSize": 1,
                         "Enabled": True,
                         "Queue": self.trigger_config["queue_arn"],
+                        **sqs_properties,
                     },
                 }
             }
@@ -257,7 +267,7 @@ class AsyncLambdaTask(Generic[EventType]):
                     "Events": events,
                     "Policies": [
                         {"Statement": policy_statements},
-                        *build_config.policy_arns,
+                        *build_config.policies,
                     ],
                     **function_properties,
                 },

{async-lambda-unstable-0.2.10 → async-lambda-unstable-0.2.12}/async_lambda_unstable.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: async-lambda-unstable
-Version: 0.2.10
+Version: 0.2.12
 Summary: A framework for creating AWS Lambda Async Workflows. - Unstable Branch
 Author-email: "Nuclei, Inc" <engineering@nuclei.ai>
 Description-Content-Type: text/markdown
@@ -46,6 +46,7 @@ All task decorators share common arguments for configuring the underlying lambda
 - `memory: int = 128` Sets the memory allocation for the function.
 - `timeout: int = 60` Sets the timeout for the function (max 900 seconds).
 - `ephemeral_storage: int = 512` Sets the ephemeral storage allocation for the function.
+- `maximum_concurrency: Optional[int] = None` Sets the maximum concurrency value for the SQS trigger for the function. (only applies to `async_task` and `sqs_task` tasks.)
 
 ## Async Task
 
@@ -172,11 +173,11 @@ These environment variables will also be available during build time.
 
 [The value is passed to the `Environment` property on `SAM::Serverless::Function`](https://docs.aws.amazon.com/serverless-application-model/latest/developerguide/sam-resource-function.html#sam-function-environment)
 
-## `policy_arns`
+## `policies`
 
 ```
 [
-    'IAM_POLICY_ARN'
+    'IAM_POLICY_ARN' | STATEMENT
 ]
 ```