Function handler

python/opensearch/ctcwl-oss/lambda/index.py:20–64

(event, context)

# Lambda handler
def handler(event, context):
    try:
        host = os.environ["COLLECTION_ENDPOINT"]
        if host.startswith("https://"):
            host = host[8:]
        region = os.environ["REGION"]
        service = "aoss"
        credentials = boto3.Session().get_credentials()

        awsauth = AWS4Auth(
            credentials.access_key,
            credentials.secret_key,
            region,
            service,
            session_token=credentials.token,
        )

        # Build the OpenSearch client
        os_client = OpenSearch(
            hosts=[{"host": host, "port": 443}],
            http_auth=awsauth,
            use_ssl=True,
            verify_certs=False,
            connection_class=RequestsHttpConnection,
            timeout=300,
        )

        cw_data = str(event["awslogs"]["data"])
        cw_logs = gzip.GzipFile(
            fileobj=BytesIO(base64.b64decode(cw_data, validate=True))
        ).read()
        cw_logs = json.loads(cw_logs)
        if cw_logs["messageType"] == "CONTROL_MESSAGE":
            print("Skipping control message")
            return

        parse_and_send(os_client, cw_logs)

    except Exception as e:
        logging.exception("Failed to process CloudWatch data")
        response_status = "FAILED"
        error_message = f"{response_status} Error: {str(e)}. "
        print(error_message)
    finally:
        return "Succeeded"


def parse_and_send(os_client, cw_logs):