def handler()

in platform/src/register/main.py [0:0]


def handler(event, _):
    """
    Lambda handler
    """

    try:
        connection_id = event["requestContext"]["connectionId"]
    except (KeyError, TypeError):
        logger.error({
            "message": "Missing connection ID in event",
            "event": event
        })
        return response("Missing connection ID", 400)

    try:
        body = json.loads(event["body"])
    except json.decoder.JSONDecodeError:
        logger.error({
            "message": "Failed to parse request body",
            "event": event
        })
        return response("Failed to parse request body", 400)

    try:
        body = json.loads(event["body"])
        service_name = body["serviceName"]
    except (KeyError, TypeError):
        logger.warning({
            "message": "Missing 'serviceName' in request body",
            "event": event
        })
        return response("Missing 'serviceName' in request body", 400)

    logger.debug({
        "message": f"Register {connection_id} with service '{service_name}'",
        "event": event
    })

    register_service(connection_id, service_name)

    return response("Connected")