const { NodeSDK } = require("@opentelemetry/sdk-node")

const { OTLPTraceExporter } = require("@opentelemetry/exporter-trace-otlp-proto")

const { AwsLambdaInstrumentation } = require("@opentelemetry/instrumentation-aws-lambda")

const { AwsInstrumentation } = require("@opentelemetry/instrumentation-aws-sdk")

const { PgInstrumentation } = require("@opentelemetry/instrumentation-pg")

const { AsyncLocalStorageContextManager } = require("@opentelemetry/context-async-hooks")

const { propagation, context, diag, DiagConsoleLogger, DiagLogLevel } = require("@opentelemetry/api")

diag.setLogger(new DiagConsoleLogger(), DiagLogLevel.ERROR)

const sqsGetter = {

keys: (carrier) => (carrier ? Object.keys(carrier) : []),

get: (carrier, key) => {

const a = carrier && carrier[key]

return a ? a.stringValue || a.StringValue || a.Value : undefined

},

const sdk = new NodeSDK({

// Explicit AsyncLocalStorage manager so the invocation span's context carries

// into the bundled handler's async pg queries → DSQL spans nest under it.

contextManager: new AsyncLocalStorageContextManager(),

traceExporter: new OTLPTraceExporter({ url: "http://localhost:4318/v1/traces" }),

instrumentations: [

new AwsLambdaInstrumentation({

// Make the producer's traceparent the PARENT → producer & consumer share

// ONE trace id (correct for our batch-size-1 FIFO source).

eventContextExtractor: (event) => {

const attrs = event && event.Records && event.Records[0] && event.Records[0].messageAttributes

return attrs ? propagation.extract(context.active(), attrs, sqsGetter) : context.active()

},

}),

new AwsInstrumentation(),

new PgInstrumentation(),

],