fix: handle undefined traceparent

CHANGELOG.md

@@ -4,6 +4,9 @@ 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]
+
+* Handle undefined traceparent for step function propagation
 
 ## [0.5.2] - 2024-03-28
 

examples/state-machine/packages/functions/src/lambda.ts

@@ -1,4 +1,4 @@
-import { withOpenTelemetry } from "./tracing";
+// import { withOpenTelemetry } from "./tracing";
 import { ApiHandler } from "sst/node/api";
 
 export const handler = withOpenTelemetry(ApiHandler(async (_evt) => {

examples/state-machine/packages/functions/src/task-two.ts

@@ -1,9 +1,8 @@
-import { withOpenTelemetry } from "./tracing";
+import { ApiHandler } from "sst/node/api";
 
-export const handler = withOpenTelemetry(async (_evt) => {
-  console.log(_evt)
+export const handler = ApiHandler(async (_evt) => {
   return {
     statusCode: 200,
     body: `Hello world. The time is ${new Date().toISOString()}`,
   };
-});
+});

src/lambda/propation.ts

@@ -37,7 +37,7 @@ export function extractContext(service: string, event: any): { parent?: Context,
             if (Array.isArray(event)) {
                 return {
                     links: event.map((parent) => {
-                        const traceparent = parent._baselime?.traceparent || parent.Payload?._baselime.traceparent;
+                        const traceparent = parent._baselime?.traceparent || parent.Payload?._baselime?.traceparent;
                         if (!traceparent) {
                             return
                         }
@@ -51,7 +51,7 @@ export function extractContext(service: string, event: any): { parent?: Context,
                     }).filter(el => el)
                 }
             }
-            const traceparent = event._baselime?.traceparent || event.Payload?._baselime.traceparent;
+            const traceparent = event._baselime?.traceparent || event.Payload?._baselime?.traceparent;
 
             return {
                 parent: propagation.extract(context.active(), { traceparent }, headerGetter)