add automatic endpoint configuration for lambda environments

Author: yiyuan-he

exporters/aws-otel-otlp-udp-exporter/src/amazon/opentelemetry/exporters/otlp/udp/exporter.py

@@ -1,5 +1,6 @@
 # Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
 # SPDX-License-Identifier: Apache-2.0
+import os
 import base64
 import socket
 from logging import Logger, getLogger
@@ -29,6 +30,10 @@
 
 class UdpExporter:
     def __init__(self, endpoint: Optional[str] = None):
+        if endpoint is None and "AWS_LAMBDA_FUNCTION_NAME" in os.environ:
+            # If in an AWS Lambda Environment, `AWS_XRAY_DAEMON_ADDRESS` will be defined
+            endpoint = os.environ.get("AWS_XRAY_DAEMON_ADDRESS", DEFAULT_ENDPOINT)
+
         self._endpoint = endpoint or DEFAULT_ENDPOINT
         self._host, self._port = self._parse_endpoint(self._endpoint)
         self._socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)