Adding additional check to SpanExporterBuilder to see if we are already working in Lambda environment or not

Author: Jeel Mehta

exporters/aws-otel-otlp-udp-exporter/src/main/java/software/opentelemetry/exporters/otlp/udp/OtlpUdpSpanExporterBuilder.java

@@ -16,6 +16,8 @@
 package software.opentelemetry.exporters.otlp.udp;
 
 import static java.util.Objects.requireNonNull;
+//import static software.amazon.opentelemetry.javaagent.providers.AwsApplicationSignalsCustomizerProvider.AWS_LAMBDA_FUNCTION_NAME_CONFIG;
+//import static software.amazon.opentelemetry.javaagent.providers.AwsApplicationSignalsCustomizerProvider.AWS_XRAY_DAEMON_ADDRESS_CONFIG;
 
 public final class OtlpUdpSpanExporterBuilder {
 
@@ -35,6 +37,9 @@ public final class OtlpUdpSpanExporterBuilder {
   private UdpSender sender;
   private String tracePayloadPrefix = FORMAT_OTEL_SAMPLED_TRACES_BINARY_PREFIX;
 
+  private static final String AWS_LAMBDA_FUNCTION_NAME_CONFIG = "AWS_LAMBDA_FUNCTION_NAME";
+  private static final String AWS_XRAY_DAEMON_ADDRESS_CONFIG = "AWS_XRAY_DAEMON_ADDRESS";
+
   public OtlpUdpSpanExporterBuilder setEndpoint(String endpoint) {
     requireNonNull(endpoint, "endpoint must not be null");
     try {
@@ -58,10 +63,34 @@ public OtlpUdpSpanExporterBuilder setPayloadSampleDecision(TracePayloadSampleDec
 
   public OtlpUdpSpanExporter build() {
     if (sender == null) {
-      this.sender = new UdpSender(DEFAULT_HOST, DEFAULT_PORT);
+        String endpoint = null;
+        
+        // If in Lambda environment, try to get X-Ray daemon address
+        if (isLambdaEnvironment()) {
+            endpoint = System.getenv(AWS_XRAY_DAEMON_ADDRESS_CONFIG);
+        }
+        
+        if (endpoint != null) {
+            // Use the endpoint from Lambda environment
+            try {
+                String[] parts = endpoint.split(":");
+                String host = parts[0];
+                int port = Integer.parseInt(parts[1]);
+                this.sender = new UdpSender(host, port);
+            } catch (Exception e) {
+                // Fallback to defaults if parsing fails
+                this.sender = new UdpSender(DEFAULT_HOST, DEFAULT_PORT);
+            }
+        } else {
+            // Use defaults if not in Lambda or if daemon address is not available
+            this.sender = new UdpSender(DEFAULT_HOST, DEFAULT_PORT);
+        }
     }
     return new OtlpUdpSpanExporter(
         this.sender, PROTOCOL_HEADER + PROTOCOL_DELIMITER + tracePayloadPrefix);
+}
+  private static boolean isLambdaEnvironment() {
+    return System.getenv(AWS_LAMBDA_FUNCTION_NAME_CONFIG) != null;
   }
 
   // Only for testing