Configure sdk for Lambda environment

Author: srprash

awsagentprovider/src/main/java/software/amazon/opentelemetry/javaagent/providers/AwsApplicationSignalsCustomizerProvider.java

@@ -20,8 +20,10 @@
 import io.opentelemetry.contrib.awsxray.AlwaysRecordSampler;
 import io.opentelemetry.contrib.awsxray.ResourceHolder;
 import io.opentelemetry.exporter.otlp.http.metrics.OtlpHttpMetricExporter;
+import io.opentelemetry.exporter.otlp.http.trace.OtlpHttpSpanExporter;
 import io.opentelemetry.exporter.otlp.internal.OtlpConfigUtil;
 import io.opentelemetry.exporter.otlp.metrics.OtlpGrpcMetricExporter;
+import io.opentelemetry.exporter.otlp.trace.OtlpGrpcSpanExporter;
 import io.opentelemetry.sdk.autoconfigure.spi.AutoConfigurationCustomizer;
 import io.opentelemetry.sdk.autoconfigure.spi.AutoConfigurationCustomizerProvider;
 import io.opentelemetry.sdk.autoconfigure.spi.ConfigProperties;
@@ -39,13 +41,7 @@
 import io.opentelemetry.sdk.trace.export.SpanExporter;
 import io.opentelemetry.sdk.trace.samplers.Sampler;
 import java.time.Duration;
-import java.util.ArrayList;
-import java.util.Collections;
-import java.util.HashMap;
-import java.util.HashSet;
-import java.util.List;
-import java.util.Map;
-import java.util.Set;
+import java.util.*;
 import java.util.logging.Level;
 import java.util.logging.Logger;
 
@@ -66,6 +62,8 @@
  */
 public class AwsApplicationSignalsCustomizerProvider
     implements AutoConfigurationCustomizerProvider {
+  static final String AWS_LAMBDA_FUNCTION_NAME_CONFIG = "AWS_LAMBDA_FUNCTION_NAME";
+
   private static final Duration DEFAULT_METRIC_EXPORT_INTERVAL = Duration.ofMinutes(1);
   private static final Logger logger =
       Logger.getLogger(AwsApplicationSignalsCustomizerProvider.class.getName());
@@ -85,6 +83,11 @@ public class AwsApplicationSignalsCustomizerProvider
       "otel.aws.application.signals.exporter.endpoint";
 
   private static final String OTEL_JMX_TARGET_SYSTEM_CONFIG = "otel.jmx.target.system";
+  private static final String OTEL_EXPORTER_OTLP_TRACES_ENDPOINT_CONFIG =
+      "OTEL_EXPORTER_OTLP_TRACES_ENDPOINT";
+  private static final String AWS_XRAY_DAEMON_ADDRESS_CONFIG = "AWS_XRAY_DAEMON_ADDRESS";
+
+  private static final String DEFAULT_UDP_ENDPOINT = "127.0.0.1:2000";
 
   public void customize(AutoConfigurationCustomizer autoConfiguration) {
     autoConfiguration.addPropertiesCustomizer(this::customizeProperties);
@@ -95,6 +98,10 @@ public void customize(AutoConfigurationCustomizer autoConfiguration) {
     autoConfiguration.addSpanExporterCustomizer(this::customizeSpanExporter);
   }
 
+  static boolean isLambdaEnvironment() {
+    return System.getenv(AWS_LAMBDA_FUNCTION_NAME_CONFIG) != null;
+  }
+
   private boolean isApplicationSignalsEnabled(ConfigProperties configProps) {
     return configProps.getBoolean(
         APPLICATION_SIGNALS_ENABLED_CONFIG,
@@ -156,6 +163,15 @@ private SdkTracerProviderBuilder customizeTracerProviderBuilder(
       // Construct and set local and remote attributes span processor
       tracerProviderBuilder.addSpanProcessor(
           AttributePropagatingSpanProcessorBuilder.create().build());
+
+      // If running on Lambda, we just need to export 100% spans and skip generating any Application
+      // Signals metrics.
+      if (isLambdaEnvironment()) {
+        tracerProviderBuilder.addSpanProcessor(
+            AwsUnsampledOnlySpanProcessorBuilder.create().build());
+        return tracerProviderBuilder;
+      }
+
       // Construct meterProvider
       MetricExporter metricsExporter =
           ApplicationSignalsExporterProvider.INSTANCE.createExporter(configProps);
@@ -207,6 +223,21 @@ private SdkMeterProviderBuilder customizeMeterProvider(
 
   private SpanExporter customizeSpanExporter(
       SpanExporter spanExporter, ConfigProperties configProps) {
+    // When running in Lambda, override the default OTLP exporter with UDP exporter
+    if (isLambdaEnvironment()) {
+      if (isOtlpSpanExporter(spanExporter)
+          && System.getenv(OTEL_EXPORTER_OTLP_TRACES_ENDPOINT_CONFIG) == null) {
+        String tracesEndpoint =
+            Optional.ofNullable(System.getenv(AWS_XRAY_DAEMON_ADDRESS_CONFIG))
+                .orElse(DEFAULT_UDP_ENDPOINT);
+        spanExporter =
+            new OtlpUdpSpanExporterBuilder()
+                .setPayloadSampleDecision(TracePayloadSampleDecision.SAMPLED)
+                .setEndpoint(tracesEndpoint)
+                .build();
+      }
+    }
+
     if (isApplicationSignalsEnabled(configProps)) {
       return AwsMetricAttributesSpanExporterBuilder.create(
               spanExporter, ResourceHolder.getResource())
@@ -216,6 +247,11 @@ private SpanExporter customizeSpanExporter(
     return spanExporter;
   }
 
+  private boolean isOtlpSpanExporter(SpanExporter spanExporter) {
+    return spanExporter instanceof OtlpGrpcSpanExporter
+        || spanExporter instanceof OtlpHttpSpanExporter;
+  }
+
   private enum ApplicationSignalsExporterProvider {
     INSTANCE;
 

awsagentprovider/src/main/java/software/amazon/opentelemetry/javaagent/providers/AwsSpanProcessingUtil.java

@@ -23,6 +23,8 @@
 import static io.opentelemetry.semconv.SemanticAttributes.MESSAGING_OPERATION;
 import static io.opentelemetry.semconv.SemanticAttributes.MessagingOperationValues.PROCESS;
 import static io.opentelemetry.semconv.SemanticAttributes.RPC_SYSTEM;
+import static software.amazon.opentelemetry.javaagent.providers.AwsApplicationSignalsCustomizerProvider.AWS_LAMBDA_FUNCTION_NAME_CONFIG;
+import static software.amazon.opentelemetry.javaagent.providers.AwsApplicationSignalsCustomizerProvider.isLambdaEnvironment;
 import static software.amazon.opentelemetry.javaagent.providers.AwsAttributeKeys.AWS_LOCAL_OPERATION;
 
 import com.fasterxml.jackson.core.type.TypeReference;
@@ -82,9 +84,13 @@ static List<String> getDialectKeywords() {
   /**
    * Ingress operation (i.e. operation for Server and Consumer spans) will be generated from
    * "http.method + http.target/with the first API path parameter" if the default span name equals
-   * null, UnknownOperation or http.method value.
+   * null, UnknownOperation or http.method value. If running in Lambda, the ingress operation will
+   * be the function name + /FunctionHandler.
    */
   static String getIngressOperation(SpanData span) {
+    if (isLambdaEnvironment()) {
+      return System.getenv(AWS_LAMBDA_FUNCTION_NAME_CONFIG) + "/FunctionHandler";
+    }
     String operation = span.getName();
     if (shouldUseInternalOperation(span)) {
       operation = INTERNAL_OPERATION;

awsagentprovider/src/main/java/software/amazon/opentelemetry/javaagent/providers/AwsUnsampledOnlySpanProcessor.java

@@ -34,10 +34,6 @@ final class AwsUnsampledOnlySpanProcessor implements SpanProcessor {
     this.delegate = delegate;
   }
 
-  public static AwsUnsampledOnlySpanProcessorBuilder builder() {
-    return new AwsUnsampledOnlySpanProcessorBuilder();
-  }
-
   @Override
   public void onStart(Context parentContext, ReadWriteSpan span) {
     if (!span.getSpanContext().isSampled()) {

awsagentprovider/src/main/java/software/amazon/opentelemetry/javaagent/providers/AwsUnsampledOnlySpanProcessorBuilder.java

@@ -21,6 +21,9 @@
 import io.opentelemetry.sdk.trace.export.SpanExporter;
 
 final class AwsUnsampledOnlySpanProcessorBuilder {
+  public static AwsUnsampledOnlySpanProcessorBuilder create() {
+    return new AwsUnsampledOnlySpanProcessorBuilder();
+  }
 
   // Default exporter is OtlpUdpSpanExporter with unsampled payload prefix
   private SpanExporter exporter =

awsagentprovider/src/test/java/software/amazon/opentelemetry/javaagent/providers/AwsUnsampledOnlySpanProcessorTest.java

@@ -35,19 +35,19 @@ public class AwsUnsampledOnlySpanProcessorTest {
 
   @Test
   public void testIsStartRequired() {
-    SpanProcessor processor = AwsUnsampledOnlySpanProcessor.builder().build();
+    SpanProcessor processor = AwsUnsampledOnlySpanProcessorBuilder.create().build();
     assertThat(processor.isStartRequired()).isTrue();
   }
 
   @Test
   public void testIsEndRequired() {
-    SpanProcessor processor = AwsUnsampledOnlySpanProcessor.builder().build();
+    SpanProcessor processor = AwsUnsampledOnlySpanProcessorBuilder.create().build();
     assertThat(processor.isEndRequired()).isTrue();
   }
 
   @Test
   public void testDefaultSpanProcessor() {
-    AwsUnsampledOnlySpanProcessorBuilder builder = AwsUnsampledOnlySpanProcessor.builder();
+    AwsUnsampledOnlySpanProcessorBuilder builder = AwsUnsampledOnlySpanProcessorBuilder.create();
     AwsUnsampledOnlySpanProcessor unsampledSP = builder.build();
 
     assertThat(builder.getSpanExporter()).isInstanceOf(OtlpUdpSpanExporter.class);
@@ -64,7 +64,8 @@ public void testDefaultSpanProcessor() {
   @Test
   public void testSpanProcessorWithExporter() {
     AwsUnsampledOnlySpanProcessorBuilder builder =
-        AwsUnsampledOnlySpanProcessor.builder().setSpanExporter(InMemorySpanExporter.create());
+        AwsUnsampledOnlySpanProcessorBuilder.create()
+            .setSpanExporter(InMemorySpanExporter.create());
     AwsUnsampledOnlySpanProcessor unsampledSP = builder.build();
 
     assertThat(builder.getSpanExporter()).isInstanceOf(InMemorySpanExporter.class);
@@ -85,7 +86,7 @@ public void testStartAddsAttributeToSampledSpan() {
     ReadWriteSpan spanMock = mock(ReadWriteSpan.class);
     when(spanMock.getSpanContext()).thenReturn(mockSpanContext);
 
-    AwsUnsampledOnlySpanProcessor processor = AwsUnsampledOnlySpanProcessor.builder().build();
+    AwsUnsampledOnlySpanProcessor processor = AwsUnsampledOnlySpanProcessorBuilder.create().build();
     processor.onStart(parentContextMock, spanMock);
 
     // verify setAttribute was never called
@@ -100,7 +101,7 @@ public void testStartAddsAttributeToUnsampledSpan() {
     ReadWriteSpan spanMock = mock(ReadWriteSpan.class);
     when(spanMock.getSpanContext()).thenReturn(mockSpanContext);
 
-    AwsUnsampledOnlySpanProcessor processor = AwsUnsampledOnlySpanProcessor.builder().build();
+    AwsUnsampledOnlySpanProcessor processor = AwsUnsampledOnlySpanProcessorBuilder.create().build();
     processor.onStart(parentContextMock, spanMock);
 
     // verify setAttribute was called with the correct arguments