make tracing key _baselime

baselime · Mar 28, 2024 · 2b983b9 · 2b983b9
1 parent afededa
commit 2b983b9
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 2 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -5,6 +5,10 @@ All notable changes to this project will be documented in this file.
 
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/).
 
+## [Unreleased]
+
+* make tracing key _baselime for step-functions
+
 ## [0.5.1] - 2024-03-28
 
 * Improve step function tracing

diff --git a/src/lambda/parse-event.ts b/src/lambda/parse-event.ts
@@ -135,7 +135,11 @@ function detectService(event: any) {
         return "eventbridge";
     }
 
-    if (process.env.BASELIME_TRACE_STEP_FUNCTION === "true" || event.Payload?.baselime?.traceparent || (Array.isArray(event) && event[0]?.Payload?.baselime?.traceparent)) {
+    if (
+        process.env.BASELIME_TRACE_STEP_FUNCTION === "true" ||
+        event.Payload?._baselime?.traceparent || event._baselime?.traceparent ||
+        (Array.isArray(event) && (event[0]?.Payload?.baselime?.traceparent || event[0]?._baselime?.traceparent))
+    ) {
         return "step-function";
     }
     return 'unknown'

diff --git a/src/lambda/propation.ts b/src/lambda/propation.ts
@@ -51,9 +51,10 @@ export function extractContext(service: string, event: any): { parent?: Context,
                     }).filter(el => el)
                 }
             }
+            const traceparent = event._baselime?.traceparent || event.Payload?._baselime.traceparent;
 
             return {
-                parent: propagation.extract(context.active(), event?.baselime || event.Payload?.baselime, headerGetter)
+                parent: propagation.extract(context.active(), { traceparent }, headerGetter)
             }
         default:
             return {