renamed _X_AMZN_TRACE_ID to com.amazonaws.xray.traceHeader (#7)

aws-xray renamed _X_AMZN_TRACE_ID environment property to System Property com.amazonaws.xray.traceHeader aws/aws-xray-sdk-java#251
formkiq · Jan 22, 2021 · e0f44dc · e0f44dc
1 parent bd9cb54
commit e0f44dc
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/build.gradle b/build.gradle
@@ -22,7 +22,7 @@ sourceCompatibility = JavaVersion.VERSION_11
 targetCompatibility = JavaVersion.VERSION_11
 
 group 'com.formkiq'
-version '2.2.0'
+version '2.2.1'
 
 spotless {
   java {

diff --git a/src/main/java/com/formkiq/lambda/runtime/graalvm/LambdaRuntime.java b/src/main/java/com/formkiq/lambda/runtime/graalvm/LambdaRuntime.java
@@ -195,7 +195,7 @@ private static void invokeClass(
 
         String xamazTraceId = event.getHeaderValue("Lambda-Runtime-Trace-Id");
         if (xamazTraceId != null) {
-          System.setProperty("_X_AMZN_TRACE_ID", xamazTraceId);
+          System.setProperty("com.amazonaws.xray.traceHeader", xamazTraceId);
         }
 
         context = new LambdaContext(requestId);

diff --git a/src/test/java/com/formkiq/lambda/runtime/graalvm/LambdaRuntimeTest.java b/src/test/java/com/formkiq/lambda/runtime/graalvm/LambdaRuntimeTest.java
@@ -313,6 +313,6 @@ public void testInvoke11() throws Exception {
     // then
     String expected = "{\"body\":\"this is some data\"}";
     assertEquals(expected, invocationResponseHandler.getResponse());
-    assertNotNull(System.getProperty("_X_AMZN_TRACE_ID"));
+    assertNotNull(System.getProperty("com.amazonaws.xray.traceHeader"));
   }
 }