From c52606f4b1eeb23197e4a9a91f25cc2a7fa8998c Mon Sep 17 00:00:00 2001 From: Olivier John Ndjike Nzia Date: Mon, 17 Nov 2025 16:23:27 -0500 Subject: [PATCH 1/4] add request-id to lambda start/end invocation calls --- .../v1/lambda/LambdaHandlerInstrumentation.java | 15 +++++++++++---- .../main/java/datadog/trace/core/CoreTracer.java | 8 ++++---- .../java/datadog/trace/lambda/LambdaHandler.java | 9 +++++++-- .../instrumentation/api/AgentTracer.java | 9 +++++---- 4 files changed, 27 insertions(+), 14 deletions(-) diff --git a/dd-java-agent/instrumentation/aws-java/aws-java-lambda-handler-1.2/src/main/java/datadog/trace/instrumentation/aws/v1/lambda/LambdaHandlerInstrumentation.java b/dd-java-agent/instrumentation/aws-java/aws-java-lambda-handler-1.2/src/main/java/datadog/trace/instrumentation/aws/v1/lambda/LambdaHandlerInstrumentation.java index 91648f09ff0..69dc44fbcf4 100644 --- a/dd-java-agent/instrumentation/aws-java/aws-java-lambda-handler-1.2/src/main/java/datadog/trace/instrumentation/aws/v1/lambda/LambdaHandlerInstrumentation.java +++ b/dd-java-agent/instrumentation/aws-java/aws-java-lambda-handler-1.2/src/main/java/datadog/trace/instrumentation/aws/v1/lambda/LambdaHandlerInstrumentation.java @@ -88,15 +88,15 @@ static AgentScope enter( if (CallDepthThreadLocalMap.incrementCallDepth(RequestHandler.class) > 0) { return null; } - - AgentSpanContext lambdaContext = AgentTracer.get().notifyExtensionStart(in); + String lambdaRequestId = awsContext.getAwsRequestId(); + AgentSpanContext lambdaContext = AgentTracer.get().notifyExtensionStart(in, lambdaRequestId); final AgentSpan span; if (null == lambdaContext) { span = startSpan(INVOCATION_SPAN_NAME); } else { span = startSpan(INVOCATION_SPAN_NAME, lambdaContext); } - span.setTag("request_id", awsContext.getAwsRequestId()); + span.setTag("request_id", lambdaRequestId); final AgentScope scope = activateSpan(span); return scope; @@ -107,6 +107,7 @@ static void exit( @Origin String method, @Enter final AgentScope scope, @Advice.Argument(1) final Object result, + @Advice.Argument(2) final Context awsContext, @Advice.Thrown final Throwable throwable) { if (scope == null) { @@ -120,8 +121,14 @@ static void exit( if (throwable != null) { span.addThrowable(throwable); } + String lambdaRequestId = awsContext.getAwsRequestId(); + + // QUESTION would it be better to get request id from span tag instead? So we dont have to + // change method signature + // E.G: + // String lambdaRequestIdFromSpan = (String) span.getTag("request_id"); span.finish(); - AgentTracer.get().notifyExtensionEnd(span, result, null != throwable); + AgentTracer.get().notifyExtensionEnd(span, result, null != throwable, lambdaRequestId); } finally { scope.close(); } diff --git a/dd-trace-core/src/main/java/datadog/trace/core/CoreTracer.java b/dd-trace-core/src/main/java/datadog/trace/core/CoreTracer.java index 05c213a9c18..7a7e30fe1c9 100644 --- a/dd-trace-core/src/main/java/datadog/trace/core/CoreTracer.java +++ b/dd-trace-core/src/main/java/datadog/trace/core/CoreTracer.java @@ -1194,13 +1194,13 @@ public void closeActive() { } @Override - public AgentSpanContext notifyExtensionStart(Object event) { - return LambdaHandler.notifyStartInvocation(this, event); + public AgentSpanContext notifyExtensionStart(Object event, String requestId) { + return LambdaHandler.notifyStartInvocation(this, event, requestId); } @Override - public void notifyExtensionEnd(AgentSpan span, Object result, boolean isError) { - LambdaHandler.notifyEndInvocation(span, result, isError); + public void notifyExtensionEnd(AgentSpan span, Object result, boolean isError, String requestId) { + LambdaHandler.notifyEndInvocation(span, result, isError, requestId); } @Override diff --git a/dd-trace-core/src/main/java/datadog/trace/lambda/LambdaHandler.java b/dd-trace-core/src/main/java/datadog/trace/lambda/LambdaHandler.java index cd7fae33672..ca6ad9f2b02 100644 --- a/dd-trace-core/src/main/java/datadog/trace/lambda/LambdaHandler.java +++ b/dd-trace-core/src/main/java/datadog/trace/lambda/LambdaHandler.java @@ -43,6 +43,7 @@ public class LambdaHandler { private static final String DATADOG_INVOCATION_ERROR_MSG = "x-datadog-invocation-error-msg"; private static final String DATADOG_INVOCATION_ERROR_TYPE = "x-datadog-invocation-error-type"; private static final String DATADOG_INVOCATION_ERROR_STACK = "x-datadog-invocation-error-stack"; + private static final String LAMBDA_RUNTIME_AWS_REQUEST_ID = "lambda-runtime-aws-request-id"; private static final String START_INVOCATION = "/lambda/start-invocation"; private static final String END_INVOCATION = "/lambda/end-invocation"; @@ -72,7 +73,8 @@ public class LambdaHandler { private static String EXTENSION_BASE_URL = "http://127.0.0.1:8124"; - public static AgentSpanContext notifyStartInvocation(CoreTracer tracer, Object event) { + public static AgentSpanContext notifyStartInvocation( + CoreTracer tracer, Object event, String requestId) { RequestBody body = RequestBody.create(jsonMediaType, writeValueAsString(event)); try (Response response = HTTP_CLIENT @@ -80,6 +82,7 @@ public static AgentSpanContext notifyStartInvocation(CoreTracer tracer, Object e new Request.Builder() .url(EXTENSION_BASE_URL + START_INVOCATION) .addHeader(DATADOG_META_LANG, "java") + .addHeader(LAMBDA_RUNTIME_AWS_REQUEST_ID, requestId) .post(body) .build()) .execute()) { @@ -99,7 +102,8 @@ public static AgentSpanContext notifyStartInvocation(CoreTracer tracer, Object e return null; } - public static boolean notifyEndInvocation(AgentSpan span, Object result, boolean isError) { + public static boolean notifyEndInvocation( + AgentSpan span, Object result, boolean isError, String requestId) { if (null == span || null == span.getSamplingPriority()) { log.error( "could not notify the extension as the lambda span is null or no sampling priority has been found"); @@ -113,6 +117,7 @@ public static boolean notifyEndInvocation(AgentSpan span, Object result, boolean .addHeader(DATADOG_SPAN_ID, DDSpanId.toString(span.getSpanId())) .addHeader(DATADOG_SAMPLING_PRIORITY, span.getSamplingPriority().toString()) .addHeader(DATADOG_META_LANG, "java") + .addHeader(LAMBDA_RUNTIME_AWS_REQUEST_ID, requestId) .post(body); Object errorMessage = span.getTag(DDTags.ERROR_MSG); diff --git a/internal-api/src/main/java/datadog/trace/bootstrap/instrumentation/api/AgentTracer.java b/internal-api/src/main/java/datadog/trace/bootstrap/instrumentation/api/AgentTracer.java index 59405a0aef1..eac5683c007 100644 --- a/internal-api/src/main/java/datadog/trace/bootstrap/instrumentation/api/AgentTracer.java +++ b/internal-api/src/main/java/datadog/trace/bootstrap/instrumentation/api/AgentTracer.java @@ -406,9 +406,9 @@ default SpanBuilder singleSpanBuilder(CharSequence spanName) { CallbackProvider getUniversalCallbackProvider(); - AgentSpanContext notifyExtensionStart(Object event); + AgentSpanContext notifyExtensionStart(Object event, String requestId); - void notifyExtensionEnd(AgentSpan span, Object result, boolean isError); + void notifyExtensionEnd(AgentSpan span, Object result, boolean isError, String requestId); AgentDataStreamsMonitoring getDataStreamsMonitoring(); @@ -653,12 +653,13 @@ public EndpointTracker onRootSpanStarted(AgentSpan root) { } @Override - public AgentSpanContext notifyExtensionStart(Object event) { + public AgentSpanContext notifyExtensionStart(Object event, String requestId) { return null; } @Override - public void notifyExtensionEnd(AgentSpan span, Object result, boolean isError) {} + public void notifyExtensionEnd( + AgentSpan span, Object result, boolean isError, String requestId) {} @Override public AgentDataStreamsMonitoring getDataStreamsMonitoring() { From dcf196f91eabf5ef1c2f9abbe12f4656cab3565e Mon Sep 17 00:00:00 2001 From: Olivier John Ndjike Nzia Date: Tue, 18 Nov 2025 13:11:35 -0500 Subject: [PATCH 2/4] tests --- .../java/datadog/trace/core/CoreTracer.java | 9 +++--- .../datadog/trace/lambda/LambdaHandler.java | 13 ++++---- .../trace/lambda/LambdaHandlerTest.groovy | 30 +++++++++++-------- .../instrumentation/api/AgentTracer.java | 6 ++-- 4 files changed, 34 insertions(+), 24 deletions(-) diff --git a/dd-trace-core/src/main/java/datadog/trace/core/CoreTracer.java b/dd-trace-core/src/main/java/datadog/trace/core/CoreTracer.java index 7a7e30fe1c9..d03ccb0a52b 100644 --- a/dd-trace-core/src/main/java/datadog/trace/core/CoreTracer.java +++ b/dd-trace-core/src/main/java/datadog/trace/core/CoreTracer.java @@ -1194,13 +1194,14 @@ public void closeActive() { } @Override - public AgentSpanContext notifyExtensionStart(Object event, String requestId) { - return LambdaHandler.notifyStartInvocation(this, event, requestId); + public AgentSpanContext notifyExtensionStart(Object event, String lambdaRequestId) { + return LambdaHandler.notifyStartInvocation(this, event, lambdaRequestId); } @Override - public void notifyExtensionEnd(AgentSpan span, Object result, boolean isError, String requestId) { - LambdaHandler.notifyEndInvocation(span, result, isError, requestId); + public void notifyExtensionEnd( + AgentSpan span, Object result, boolean isError, String lambdaRequestId) { + LambdaHandler.notifyEndInvocation(span, result, isError, lambdaRequestId); } @Override diff --git a/dd-trace-core/src/main/java/datadog/trace/lambda/LambdaHandler.java b/dd-trace-core/src/main/java/datadog/trace/lambda/LambdaHandler.java index ca6ad9f2b02..41603f44fc0 100644 --- a/dd-trace-core/src/main/java/datadog/trace/lambda/LambdaHandler.java +++ b/dd-trace-core/src/main/java/datadog/trace/lambda/LambdaHandler.java @@ -74,15 +74,18 @@ public class LambdaHandler { private static String EXTENSION_BASE_URL = "http://127.0.0.1:8124"; public static AgentSpanContext notifyStartInvocation( - CoreTracer tracer, Object event, String requestId) { - RequestBody body = RequestBody.create(jsonMediaType, writeValueAsString(event)); + CoreTracer tracer, Object event, String lambdaRequestId) { + RequestBody body = + RequestBody.create( + jsonMediaType, + writeValueAsString(event)); // MAYBE remove CoreTracer parameter since it is not used try (Response response = HTTP_CLIENT .newCall( new Request.Builder() .url(EXTENSION_BASE_URL + START_INVOCATION) .addHeader(DATADOG_META_LANG, "java") - .addHeader(LAMBDA_RUNTIME_AWS_REQUEST_ID, requestId) + .addHeader(LAMBDA_RUNTIME_AWS_REQUEST_ID, lambdaRequestId) .post(body) .build()) .execute()) { @@ -103,7 +106,7 @@ public static AgentSpanContext notifyStartInvocation( } public static boolean notifyEndInvocation( - AgentSpan span, Object result, boolean isError, String requestId) { + AgentSpan span, Object result, boolean isError, String lambdaRequestId) { if (null == span || null == span.getSamplingPriority()) { log.error( "could not notify the extension as the lambda span is null or no sampling priority has been found"); @@ -117,7 +120,7 @@ public static boolean notifyEndInvocation( .addHeader(DATADOG_SPAN_ID, DDSpanId.toString(span.getSpanId())) .addHeader(DATADOG_SAMPLING_PRIORITY, span.getSamplingPriority().toString()) .addHeader(DATADOG_META_LANG, "java") - .addHeader(LAMBDA_RUNTIME_AWS_REQUEST_ID, requestId) + .addHeader(LAMBDA_RUNTIME_AWS_REQUEST_ID, lambdaRequestId) .post(body); Object errorMessage = span.getTag(DDTags.ERROR_MSG); diff --git a/dd-trace-core/src/test/groovy/datadog/trace/lambda/LambdaHandlerTest.groovy b/dd-trace-core/src/test/groovy/datadog/trace/lambda/LambdaHandlerTest.groovy index d06e9f7f151..45969111924 100644 --- a/dd-trace-core/src/test/groovy/datadog/trace/lambda/LambdaHandlerTest.groovy +++ b/dd-trace-core/src/test/groovy/datadog/trace/lambda/LambdaHandlerTest.groovy @@ -50,11 +50,12 @@ class LambdaHandlerTest extends DDCoreSpecification { LambdaHandler.setExtensionBaseUrl(server.address.toString()) when: - def objTest = LambdaHandler.notifyStartInvocation(ct, obj) + def objTest = LambdaHandler.notifyStartInvocation(ct, obj, "lambda-request-123") then: objTest.getTraceId().toString() == traceId objTest.getSamplingPriority() == samplingPriority + server.lastRequest.headers.get("lambda-runtime-aws-request-id") == "lambda-request-123" cleanup: server.close() @@ -84,11 +85,12 @@ class LambdaHandlerTest extends DDCoreSpecification { LambdaHandler.setExtensionBaseUrl(server.address.toString()) when: - def objTest = LambdaHandler.notifyStartInvocation(ct, obj) + def objTest = LambdaHandler.notifyStartInvocation(ct, obj, "lambda-request-123") then: objTest.getTraceId().toHexString() == traceId objTest.getSamplingPriority() == samplingPriority + server.lastRequest.headers.get("lambda-runtime-aws-request-id") == "lambda-request-123" cleanup: server.close() @@ -115,10 +117,11 @@ class LambdaHandlerTest extends DDCoreSpecification { LambdaHandler.setExtensionBaseUrl(server.address.toString()) when: - def objTest = LambdaHandler.notifyStartInvocation(ct, obj) + def objTest = LambdaHandler.notifyStartInvocation(ct, obj, "my-lambda-request") then: objTest == expected + server.lastRequest.headers.get("lambda-runtime-aws-request-id") == "my-lambda-request" cleanup: server.close() @@ -148,22 +151,23 @@ class LambdaHandlerTest extends DDCoreSpecification { } when: - def result = LambdaHandler.notifyEndInvocation(span, lambdaResult, boolValue) + def result = LambdaHandler.notifyEndInvocation(span, lambdaResult, boolValue, lambdaReqIdHeaderValue) then: server.lastRequest.headers.get("x-datadog-invocation-error") == eHeaderValue server.lastRequest.headers.get("x-datadog-trace-id") == tIdHeaderValue server.lastRequest.headers.get("x-datadog-span-id") == sIdHeaderValue server.lastRequest.headers.get("x-datadog-sampling-priority") == sPIdHeaderValue + server.lastRequest.headers.get("lambda-runtime-aws-request-id") == lambdaReqIdHeaderValue result == expected cleanup: server.close() where: - expected | eHeaderValue | tIdHeaderValue | sIdHeaderValue | sPIdHeaderValue | lambdaResult | boolValue - true | "true" | "1234" | "5678" | "2" | {} | true - true | null | "1234" | "5678" | "2" | "12345" | false + expected | eHeaderValue | tIdHeaderValue | sIdHeaderValue | sPIdHeaderValue | lambdaResult | boolValue | lambdaReqIdHeaderValue + true | "true" | "1234" | "5678" | "2" | {} | true | "request123" + true | null | "1234" | "5678" | "2" | "12345" | false | "request456" } def "test end invocation failure"() { @@ -185,19 +189,20 @@ class LambdaHandlerTest extends DDCoreSpecification { } when: - def result = LambdaHandler.notifyEndInvocation(span, lambdaResult, boolValue) + def result = LambdaHandler.notifyEndInvocation(span, lambdaResult, boolValue, lambdaReqIdHeaderValue) then: result == expected server.lastRequest.headers.get("x-datadog-invocation-error") == headerValue + server.lastRequest.headers.get("lambda-runtime-aws-request-id") == lambdaReqIdHeaderValue cleanup: server.close() where: - expected | headerValue | lambdaResult | boolValue - false | "true" | {} | true - false | null | "12345" | false + expected | headerValue | lambdaResult | boolValue | lambdaReqIdHeaderValue + false | "true" | {} | true | "request123" + false | null | "12345" | false | "request456" } def "test end invocation success with error metadata"() { @@ -222,13 +227,14 @@ class LambdaHandlerTest extends DDCoreSpecification { } when: - LambdaHandler.notifyEndInvocation(span, {}, true) + LambdaHandler.notifyEndInvocation(span, {}, true, "lambda-request-123") then: server.lastRequest.headers.get("x-datadog-invocation-error") == "true" server.lastRequest.headers.get("x-datadog-invocation-error-msg") == "custom error message" server.lastRequest.headers.get("x-datadog-invocation-error-type") == "java.lang.Throwable" server.lastRequest.headers.get("x-datadog-invocation-error-stack") == "ZXJyb3JTdGFjawogCXRlc3Q=" + server.lastRequest.headers.get("lambda-runtime-aws-request-id") == "lambda-request-123" cleanup: server.close() diff --git a/internal-api/src/main/java/datadog/trace/bootstrap/instrumentation/api/AgentTracer.java b/internal-api/src/main/java/datadog/trace/bootstrap/instrumentation/api/AgentTracer.java index eac5683c007..fde1472d77d 100644 --- a/internal-api/src/main/java/datadog/trace/bootstrap/instrumentation/api/AgentTracer.java +++ b/internal-api/src/main/java/datadog/trace/bootstrap/instrumentation/api/AgentTracer.java @@ -406,9 +406,9 @@ default SpanBuilder singleSpanBuilder(CharSequence spanName) { CallbackProvider getUniversalCallbackProvider(); - AgentSpanContext notifyExtensionStart(Object event, String requestId); + AgentSpanContext notifyExtensionStart(Object event, String lambdaRequestId); - void notifyExtensionEnd(AgentSpan span, Object result, boolean isError, String requestId); + void notifyExtensionEnd(AgentSpan span, Object result, boolean isError, String lambdaRequestId); AgentDataStreamsMonitoring getDataStreamsMonitoring(); @@ -653,7 +653,7 @@ public EndpointTracker onRootSpanStarted(AgentSpan root) { } @Override - public AgentSpanContext notifyExtensionStart(Object event, String requestId) { + public AgentSpanContext notifyExtensionStart(Object event, String lambdaRequestId) { return null; } From 1f69e8ee8727d4ccc7988b133e5f30f176c9439d Mon Sep 17 00:00:00 2001 From: Olivier John Ndjike Nzia Date: Tue, 18 Nov 2025 16:38:11 -0500 Subject: [PATCH 3/4] commenting and naming --- .../aws/v1/lambda/LambdaHandlerInstrumentation.java | 4 ---- .../trace/bootstrap/instrumentation/api/AgentTracer.java | 2 +- 2 files changed, 1 insertion(+), 5 deletions(-) diff --git a/dd-java-agent/instrumentation/aws-java/aws-java-lambda-handler-1.2/src/main/java/datadog/trace/instrumentation/aws/v1/lambda/LambdaHandlerInstrumentation.java b/dd-java-agent/instrumentation/aws-java/aws-java-lambda-handler-1.2/src/main/java/datadog/trace/instrumentation/aws/v1/lambda/LambdaHandlerInstrumentation.java index 69dc44fbcf4..1f75e292327 100644 --- a/dd-java-agent/instrumentation/aws-java/aws-java-lambda-handler-1.2/src/main/java/datadog/trace/instrumentation/aws/v1/lambda/LambdaHandlerInstrumentation.java +++ b/dd-java-agent/instrumentation/aws-java/aws-java-lambda-handler-1.2/src/main/java/datadog/trace/instrumentation/aws/v1/lambda/LambdaHandlerInstrumentation.java @@ -123,10 +123,6 @@ static void exit( } String lambdaRequestId = awsContext.getAwsRequestId(); - // QUESTION would it be better to get request id from span tag instead? So we dont have to - // change method signature - // E.G: - // String lambdaRequestIdFromSpan = (String) span.getTag("request_id"); span.finish(); AgentTracer.get().notifyExtensionEnd(span, result, null != throwable, lambdaRequestId); } finally { diff --git a/internal-api/src/main/java/datadog/trace/bootstrap/instrumentation/api/AgentTracer.java b/internal-api/src/main/java/datadog/trace/bootstrap/instrumentation/api/AgentTracer.java index fde1472d77d..e32f201a3df 100644 --- a/internal-api/src/main/java/datadog/trace/bootstrap/instrumentation/api/AgentTracer.java +++ b/internal-api/src/main/java/datadog/trace/bootstrap/instrumentation/api/AgentTracer.java @@ -659,7 +659,7 @@ public AgentSpanContext notifyExtensionStart(Object event, String lambdaRequestI @Override public void notifyExtensionEnd( - AgentSpan span, Object result, boolean isError, String requestId) {} + AgentSpan span, Object result, boolean isError, String lambdaRequestId) {} @Override public AgentDataStreamsMonitoring getDataStreamsMonitoring() { From 3241b0f7c3a26a42c4569745ff14ec3f7f9b1c14 Mon Sep 17 00:00:00 2001 From: Olivier John Ndjike Nzia Date: Wed, 19 Nov 2025 09:49:58 -0500 Subject: [PATCH 4/4] remove unused coreTracer param --- .../src/main/java/datadog/trace/core/CoreTracer.java | 2 +- .../main/java/datadog/trace/lambda/LambdaHandler.java | 9 ++------- .../groovy/datadog/trace/lambda/LambdaHandlerTest.groovy | 6 +++--- 3 files changed, 6 insertions(+), 11 deletions(-) diff --git a/dd-trace-core/src/main/java/datadog/trace/core/CoreTracer.java b/dd-trace-core/src/main/java/datadog/trace/core/CoreTracer.java index e64357e03cd..77f14baa122 100644 --- a/dd-trace-core/src/main/java/datadog/trace/core/CoreTracer.java +++ b/dd-trace-core/src/main/java/datadog/trace/core/CoreTracer.java @@ -1195,7 +1195,7 @@ public void closeActive() { @Override public AgentSpanContext notifyExtensionStart(Object event, String lambdaRequestId) { - return LambdaHandler.notifyStartInvocation(this, event, lambdaRequestId); + return LambdaHandler.notifyStartInvocation(event, lambdaRequestId); } @Override diff --git a/dd-trace-core/src/main/java/datadog/trace/lambda/LambdaHandler.java b/dd-trace-core/src/main/java/datadog/trace/lambda/LambdaHandler.java index 41603f44fc0..4ee2add2940 100644 --- a/dd-trace-core/src/main/java/datadog/trace/lambda/LambdaHandler.java +++ b/dd-trace-core/src/main/java/datadog/trace/lambda/LambdaHandler.java @@ -9,7 +9,6 @@ import datadog.trace.api.DDTags; import datadog.trace.bootstrap.instrumentation.api.AgentSpan; import datadog.trace.bootstrap.instrumentation.api.AgentSpanContext; -import datadog.trace.core.CoreTracer; import java.io.ByteArrayInputStream; import java.io.ByteArrayOutputStream; import java.nio.charset.StandardCharsets; @@ -73,12 +72,8 @@ public class LambdaHandler { private static String EXTENSION_BASE_URL = "http://127.0.0.1:8124"; - public static AgentSpanContext notifyStartInvocation( - CoreTracer tracer, Object event, String lambdaRequestId) { - RequestBody body = - RequestBody.create( - jsonMediaType, - writeValueAsString(event)); // MAYBE remove CoreTracer parameter since it is not used + public static AgentSpanContext notifyStartInvocation(Object event, String lambdaRequestId) { + RequestBody body = RequestBody.create(jsonMediaType, writeValueAsString(event)); try (Response response = HTTP_CLIENT .newCall( diff --git a/dd-trace-core/src/test/groovy/datadog/trace/lambda/LambdaHandlerTest.groovy b/dd-trace-core/src/test/groovy/datadog/trace/lambda/LambdaHandlerTest.groovy index 45969111924..ab8b188eb8b 100644 --- a/dd-trace-core/src/test/groovy/datadog/trace/lambda/LambdaHandlerTest.groovy +++ b/dd-trace-core/src/test/groovy/datadog/trace/lambda/LambdaHandlerTest.groovy @@ -50,7 +50,7 @@ class LambdaHandlerTest extends DDCoreSpecification { LambdaHandler.setExtensionBaseUrl(server.address.toString()) when: - def objTest = LambdaHandler.notifyStartInvocation(ct, obj, "lambda-request-123") + def objTest = LambdaHandler.notifyStartInvocation(obj, "lambda-request-123") then: objTest.getTraceId().toString() == traceId @@ -85,7 +85,7 @@ class LambdaHandlerTest extends DDCoreSpecification { LambdaHandler.setExtensionBaseUrl(server.address.toString()) when: - def objTest = LambdaHandler.notifyStartInvocation(ct, obj, "lambda-request-123") + def objTest = LambdaHandler.notifyStartInvocation(obj, "lambda-request-123") then: objTest.getTraceId().toHexString() == traceId @@ -117,7 +117,7 @@ class LambdaHandlerTest extends DDCoreSpecification { LambdaHandler.setExtensionBaseUrl(server.address.toString()) when: - def objTest = LambdaHandler.notifyStartInvocation(ct, obj, "my-lambda-request") + def objTest = LambdaHandler.notifyStartInvocation(obj, "my-lambda-request") then: objTest == expected