update format

Author: laspsandoval

sds_data_manager/constructs/ialirt_api_manager_construct.py

@@ -223,12 +223,14 @@ def __init__(
             handler="IAlirtCode.ialirt_db_query_api.lambda_handler",
             runtime=lambda_.Runtime.PYTHON_3_12,
             timeout=cdk.Duration.minutes(1),
-            memory_size=1000,
+            # Lambda allocates CPU proportionally to memory,
+            # and DynamoDB queries are often CPU-bound due to
+            # JSON parsing and network serialization.
+            memory_size=2048,  # MB
             environment={
                 "ALGORITHM_TABLE": algorithm_table.table_name,
                 "REGION": env.region,
             },
-            layers=layers,
         )
 
         # Grant the lambda function read/write permissions on the DynamoDB table.
@@ -250,12 +252,14 @@ def __init__(
             handler="IAlirtCode.ialirt_db_query_api_formatted.lambda_handler",
             runtime=lambda_.Runtime.PYTHON_3_12,
             timeout=cdk.Duration.minutes(1),
-            memory_size=1000,
+            # Lambda allocates CPU proportionally to memory,
+            # and DynamoDB queries are often CPU-bound due to
+            # JSON parsing and network serialization.
+            memory_size=2048,
             environment={
                 "ALGORITHM_TABLE": algorithm_table.table_name,
                 "REGION": env.region,
             },
-            layers=layers,
         )
 
         # Grant the lambda function read/write permissions on the DynamoDB table.

sds_data_manager/lambda_code/IAlirtCode/ialirt_db_query_api.py

@@ -3,6 +3,7 @@
 import json
 import logging
 import os
+import time
 from decimal import Decimal
 
 import boto3
@@ -50,7 +51,7 @@ def process_item_types(item: dict) -> dict:
     return result
 
 
-def lambda_handler(event, context):  # noqa: PLR0912
+def lambda_handler(event, context):  # noqa: PLR0912, PLR0915
     """Create metadata and add it to the database.
 
     This function is an event handler for s3 ingest bucket.
@@ -67,12 +68,16 @@ def lambda_handler(event, context):  # noqa: PLR0912
         and runtime environment.
 
     """
+    t0 = time.perf_counter()
     table_name = os.environ.get("ALGORITHM_TABLE")
     region = os.environ.get("AWS_DEFAULT_REGION", "us-west-2")
     dynamodb = boto3.resource("dynamodb", region_name=region)
     table = dynamodb.Table(table_name)
+    t1 = time.perf_counter()
 
     logger.info(f"Received event: {json.dumps(event)}")
+
+    # --- Parse event ---
     params = event.get("queryStringParameters", {})
 
     if not params:
@@ -83,7 +88,9 @@ def lambda_handler(event, context):  # noqa: PLR0912
 
     key_expr = Key("apid").eq(478)
     query_kwargs = {"KeyConditionExpression": key_expr}
+    t2 = time.perf_counter()
 
+    # --- Determine key condition ---
     allowed_params = {
         "met_start",
         "met_end",
@@ -183,10 +190,30 @@ def lambda_handler(event, context):  # noqa: PLR0912
         }
 
     query_kwargs["KeyConditionExpression"] = key_expr
+    t3 = time.perf_counter()
 
+    # --- Query DynamoDB ---
     response = table.query(**query_kwargs)
+    t4 = time.perf_counter()
 
+    # --- Process items ---
     items = response.get("Items", [])
     processed_items = [process_item_types(item) for item in items]
+    t5 = time.perf_counter()
+
+    # --- Serialize to JSON ---
+    json_body = json.dumps(processed_items)
+    t6 = time.perf_counter()
+
+    num_items = len(processed_items)
+
+    text = (
+        f"[TIMER SUMMARY] DynamoDB init: {t1 - t0:.3f}s | "
+        f"Param parse: {t2 - t1:.3f}s | KeyCondition setup: {t3 - t2:.3f}s | "
+        f"Query: {t4 - t3:.3f}s | Process: {t5 - t4:.3f}s | "
+        f"JSON: {t6 - t5:.3f}s | TOTAL: {t6 - t0:.3f}s | "
+        f"Items: {num_items}"
+    )
+    logger.info(text)
 
-    return {"statusCode": 200, "body": json.dumps(processed_items)}
+    return {"statusCode": 200, "body": json_body}