[PRMP-1048] handler

Author: SWhyteAnswer

.github/workflows/base-lambdas-reusable-deploy-all.yml

@@ -780,3 +780,17 @@ jobs:
       lambda_layer_names: "core_lambda_layer"
     secrets:
       AWS_ASSUME_ROLE: ${{ secrets.AWS_ASSUME_ROLE }}
+
+  deploy_concurrency_controller_lambda:
+    name: Deploy Concurrency Controller Lambda
+    uses: ./.github/workflows/base-lambdas-reusable-deploy.yml
+    with:
+      environment: ${{ inputs.environment }}
+      python_version: ${{ inputs.python_version }}
+      build_branch: ${{ inputs.build_branch }}
+      sandbox: ${{ inputs.sandbox }}
+      lambda_handler_name: concurrency_controller_handler
+      lambda_aws_name: ConcurrencyController
+      lambda_layer_names: "core_lambda_layer"
+    secrets:
+      AWS_ASSUME_ROLE: ${{ secrets.AWS_ASSUME_ROLE }}

lambdas/handlers/concurrency_controller_handler.py

@@ -0,0 +1,32 @@
+from services.concurrency_controller_service import ConcurrencyControllerService
+from utils.audit_logging_setup import LoggingService
+from utils.decorators.handle_lambda_exceptions import handle_lambda_exceptions
+from utils.decorators.override_error_check import override_error_check
+from utils.decorators.set_audit_arg import set_request_context_for_logging
+
+logger = LoggingService(__name__)
+
+
+def validate_event(event):
+    target_function = event.get("targetFunction")
+    reserved_concurrency = event.get("reservedConcurrency")
+
+    if not target_function:
+        logger.error("Missing required parameter: targetFunction")
+        raise ValueError("targetFunction is required")
+
+    if reserved_concurrency is None:
+        logger.error("Missing required parameter: reservedConcurrency")
+        raise ValueError("reservedConcurrency is required")
+
+    return target_function, reserved_concurrency
+
+
+@set_request_context_for_logging
+@override_error_check
+@handle_lambda_exceptions
+def lambda_handler(event, _context):
+    target_function, reserved_concurrency = validate_event(event)
+    
+    service = ConcurrencyControllerService()
+    return service.update_function_concurrency(target_function, reserved_concurrency)

lambdas/services/concurrency_controller_service.py

@@ -0,0 +1,42 @@
+import boto3
+from utils.audit_logging_setup import LoggingService
+
+logger = LoggingService(__name__)
+
+
+class ConcurrencyControllerService:
+    def __init__(self):
+        self.lambda_client = boto3.client("lambda")
+
+    def update_function_concurrency(self, target_function, reserved_concurrency):
+        logger.info(
+            f"Updating reserved concurrency for function '{target_function}' to {reserved_concurrency}"
+        )
+
+        try:
+            response = self.lambda_client.put_function_concurrency(
+                FunctionName=target_function,
+                ReservedConcurrentExecutions=reserved_concurrency
+            )
+            
+            updated_concurrency = response.get("ReservedConcurrentExecutions")
+            
+            logger.info(
+                f"Successfully updated concurrency for '{target_function}'. "
+                f"Reserved concurrency set to: {updated_concurrency}"
+            )
+            
+            return {
+                "statusCode": 200,
+                "body": {
+                    "message": "Concurrency updated successfully",
+                    "function": target_function,
+                    "reservedConcurrency": updated_concurrency
+                }
+            }
+        except self.lambda_client.exceptions.ResourceNotFoundException:
+            logger.error(f"Lambda function '{target_function}' not found")
+            raise
+        except self.lambda_client.exceptions.InvalidParameterValueException as e:
+            logger.error(f"Invalid concurrency value: {e}")
+            raise