Send telemetry putlog api (#351)

* use PutLogEvents API

Author: chaudharysaket

python/newrelic_lambda/agent_protocol.py

@@ -17,6 +17,13 @@
 
 logger = logging.getLogger(__name__)
 
+def put_payload_cloudwatch(payload):
+    try:
+        cloudwatch_logging = __import__('newrelic_lambda.cloudwatch_logging', fromlist=['put_log_to_cloudwatch'])
+        cloudwatch_logging.put_log_to_cloudwatch(payload)
+    except Exception as e:
+        print(f"Failed to send payload to CloudWatch: {e}, resorting to print payload.")
+        print(payload)
 
 if ServerlessModeProtocol is not None:
     # New Relic Agent >=5.16
@@ -44,7 +51,10 @@ def protocol_finalize(self):
                     "Failed to write to named pipe %s: %s" % (NAMED_PIPE_PATH, e)
                 )
         else:
-            print(payload)
+            if os.getenv("NEW_RELIC_MAX_PAYLOAD", "false").lower() == "true":
+                put_payload_cloudwatch(payload)
+            else:
+                print(payload)
 
             return payload
 

python/newrelic_lambda/cloudwatch_logging.py

@@ -0,0 +1,32 @@
+import os
+import boto3
+from time import time
+
+log_group_name = os.getenv("AWS_LAMBDA_LOG_GROUP_NAME", "")
+log_stream_name = os.getenv("AWS_LAMBDA_LOG_STREAM_NAME", "")
+log_level = os.getenv("NEW_RELIC_LOG_LEVEL", "info").lower()
+
+def put_log_to_cloudwatch(payload):
+    logs_client = boto3.client('logs')
+
+    def ensure_log_stream_exists(log_group_name, log_stream_name):
+        try:
+            logs_client.create_log_stream(logGroupName=log_group_name, logStreamName=log_stream_name)
+        except Exception as e:
+            if log_level == "debug":
+                print(f"Attempt create log stream {log_stream_name} in log group {log_group_name}: {e}")
+
+    ensure_log_stream_exists(log_group_name, log_stream_name)
+
+    log_event = {
+        'timestamp': int(time() * 1000),
+        'message': payload,
+    }
+
+    logs_client.put_log_events(
+        logGroupName=log_group_name,
+        logStreamName=log_stream_name,
+        logEvents=[log_event]
+    )
+    if log_level == "debug":
+        print(f"Log event successfully sent to CloudWatch: {len(payload)} bytes, log group: {log_group_name}, log stream: {log_stream_name}")