Skip to content

Commit 90c8a46

Browse files
committed
feat: Aws Lambda Request Handler instrumentation uses OTEL_INSTRUMENTATION_AWS_LAMBDA_FLUSH_TIMEOUT
Using `WrapperConfiguration.flushTimeout()` for consistency with `TracingRequestStreamWrapper`
1 parent b184587 commit 90c8a46

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

instrumentation/aws-lambda/aws-lambda-core-1.0/javaagent/src/main/java/io/opentelemetry/javaagent/instrumentation/awslambdacore/v1_0/AwsLambdaRequestHandlerInstrumentation.java

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@
55

66
package io.opentelemetry.javaagent.instrumentation.awslambdacore.v1_0;
77

8+
import static io.opentelemetry.instrumentation.awslambdacore.v1_0.internal.WrapperConfiguration.flushTimeout;
89
import static io.opentelemetry.javaagent.extension.matcher.AgentElementMatchers.hasClassesNamed;
910
import static io.opentelemetry.javaagent.extension.matcher.AgentElementMatchers.implementsInterface;
1011
import static io.opentelemetry.javaagent.instrumentation.awslambdacore.v1_0.AwsLambdaInstrumentationHelper.functionInstrumenter;
@@ -59,6 +60,8 @@ public void transform(TypeTransformer transformer) {
5960
@SuppressWarnings("unused")
6061
public static class HandleRequestAdvice {
6162

63+
private final static long flushTimeoutNanos = flushTimeout().toNanos();
64+
6265
@Advice.OnMethodEnter(suppress = Throwable.class)
6366
public static void onEnter(
6467
@Advice.Argument(value = 0, typing = Typing.DYNAMIC) Object arg,
@@ -90,7 +93,7 @@ public static void stopSpan(
9093
functionInstrumenter().end(functionContext, input, null, throwable);
9194
}
9295

93-
OpenTelemetrySdkAccess.forceFlush(1, TimeUnit.SECONDS);
96+
OpenTelemetrySdkAccess.forceFlush(flushTimeoutNanos, TimeUnit.NANOSECONDS);
9497
}
9598
}
9699
}

0 commit comments

Comments
 (0)