瀏覽代碼

fix(domain-events-infra): add parent context on internal span

Karol Sójko 1 年之前
父節點
當前提交
099c6e10c6
共有 1 個文件被更改,包括 4 次插入1 次删除
  1. 4 1
      packages/domain-events-infra/src/Infra/SQS/SQSOpenTelemetryEventMessageHandler.ts

+ 4 - 1
packages/domain-events-infra/src/Infra/SQS/SQSOpenTelemetryEventMessageHandler.ts

@@ -38,11 +38,14 @@ export class SQSOpenTelemetryEventMessageHandler implements DomainEventMessageHa
     const tracer = OpenTelemetryApi.trace.getTracer('sqs-handler')
 
     this.currentSpan = tracer.startSpan(this.serviceName, { kind: OpenTelemetryApi.SpanKind.CONSUMER })
-    this.internalSpan = tracer.startSpan(domainEvent.type, { kind: OpenTelemetryApi.SpanKind.INTERNAL })
+    const ctx = OpenTelemetryApi.trace.setSpan(OpenTelemetryApi.context.active(), this.currentSpan)
+
+    this.internalSpan = tracer.startSpan(domainEvent.type, { kind: OpenTelemetryApi.SpanKind.INTERNAL }, ctx)
 
     await handler.handle(domainEvent)
 
     this.internalSpan.end()
+
     this.currentSpan.end()
 
     this.internalSpan = undefined