[fix] fix request, prepare run on aws lambda

dockerfiles/dockerfile-base-webserver

@@ -7,6 +7,11 @@ ARG RIE="https://github.com/aws/aws-lambda-runtime-interface-emulator/releases/l
 
 # Set working directory to function root directory
 WORKDIR ${LAMBDA_TASK_ROOT}
+COPY requirements_dev.txt ${LAMBDA_TASK_ROOT}/requirements_dev.txt
+COPY ./src ${LAMBDA_TASK_ROOT}/src
+
+RUN apt update && apt install -y curl python3-pip
+RUN python -m pip install -r ${LAMBDA_TASK_ROOT}/requirements_dev.txt --target ${LAMBDA_TASK_ROOT}
 
 RUN curl -Lo /usr/local/bin/aws-lambda-rie ${RIE}
 RUN chmod +x /usr/local/bin/aws-lambda-rie
@@ -15,11 +20,5 @@ COPY ./scripts/lambda-entrypoint.sh /lambda-entrypoint.sh
 RUN chmod +x /lambda-entrypoint.sh
 RUN ls -l /lambda-entrypoint.sh
 
-COPY requirements_dev.txt ${LAMBDA_TASK_ROOT}/requirements_dev.txt
-COPY ./src ${LAMBDA_TASK_ROOT}/src
-
-RUN apt update && apt install -y python3-pip
-RUN python -m pip install -r requirements_dev.txt
-
 ENTRYPOINT  ["/lambda-entrypoint.sh"]
 CMD [ "src.app.lambda_handler" ]

requirements_dev.txt

@@ -1,4 +1,6 @@
 awslambdaric
 aws_lambda_powertools
+fastjsonschema
+pydantic
 jmespath
 

src/app.py

@@ -1,22 +1,61 @@
 import json
-
+import time
 from aws_lambda_powertools import Logger
 from aws_lambda_powertools.utilities.typing import LambdaContext
+from pydantic import BaseModel, ValidationError
+
+from src.utilities.type_hints import input_floatlist, input_floatlist2
+
 
 logger = Logger()
 
 
-# @logger.inject_lambda_context
+class BBoxWithPointInput(BaseModel):
+    bbox: input_floatlist
+    points: input_floatlist2
+
+
+def get_response(status: int, start_time: float, output: BaseModel = None) -> dict[str]:
+    """
+    Return a response for frontend clients.
+
+    Args:
+        status: status response
+        start_time: request start time (float)
+        output: dict we embed into our response
+
+    Returns:
+        dict: response
+
+    """
+    messages = {200: "ok", 422: "validation error", 500: "internal server error"}
+    body = {"duration_run": time.time() - start_time, "message": messages[status]}
+    if status == 200:
+        body = {"output": output.model_dump_json(), **body}
+    return {
+        "statusCode": status,
+        "headers": {'Content-type': 'application/json', 'Accept': 'application/json'},
+        "body": json.dumps(body)
+    }
+
+
 def lambda_handler(event: dict, context: LambdaContext) -> dict[str]:
+    logger.info(f"start with aws_request_id:{context.aws_request_id}.")
+    start_time = time.time()
     try:
-        logger.info("start")
-        logger.debug("debug...")
-        logger.info(f"event:{json.dumps(event)}...")
-        logger.info(f"context:{context}...")
-        logger.info("end")
-        if event["test"] > 100:
-            raise ValueError(f"value test {event['test']} too high!")
-        return {"test": event["test"]}
+        logger.debug(f"event:{json.dumps(event)}...")
+        logger.debug(f"context:{context}...")
+
+        try:
+            bbox_points = BBoxWithPointInput(bbox=event["bbox"], points=event["points"])
+            logger.info(f"validation ok, bbox_points:{bbox_points}...")
+            response = get_response(200, start_time, bbox_points)
+        except ValidationError as ve:
+            logger.error(f"validation error:{ve}.")
+            response = get_response(422, start_time)
     except Exception as e:
         logger.error(f"exception:{e}.")
-        return {"msg": "internal server error"}
+        response = get_response(500, start_time)
+
+    logger.info(f"response:{response}...")
+    return response

src/utilities/type_hints.py

@@ -3,4 +3,4 @@ from typing import List
 
 input_floatlist = List[float]
 input_floatlist2 = List[input_floatlist]
-
+ts_dict_str2 = dict[str, str]