fix: pino trace linking issue

Author: wrn14897

packages/node-logger/src/logger.ts

@@ -1,13 +1,19 @@
 import stringifySafe from 'json-stringify-safe';
-import { Attributes, diag, DiagConsoleLogger } from '@opentelemetry/api';
+import {
+  Attributes,
+  context as apiContext,
+  diag,
+  trace,
+  DiagConsoleLogger,
+} from '@opentelemetry/api';
 import { getEnvWithoutDefaults } from '@opentelemetry/core';
 import {
   BatchLogRecordProcessor,
   BufferConfig,
   LoggerProvider,
 } from '@opentelemetry/sdk-logs';
 import { OTLPLogExporter } from '@opentelemetry/exporter-logs-otlp-http';
-import { Logger as OtelLogger, SeverityNumber } from '@opentelemetry/api-logs';
+import { Logger as OtelLogger, LogRecord } from '@opentelemetry/api-logs';
 import {
   Resource,
   defaultServiceName,
@@ -47,6 +53,47 @@ const DEFAULT_SERVICE_NAME = otelEnv.OTEL_SERVICE_NAME ?? defaultServiceName();
 
 const LOG_PREFIX = `⚠️  ${_LOG_PREFIX}`;
 
+/**
+ * Load context from attributes and set it to logRecord.context
+ *
+ * @param {LogRecord} logRecord
+ * @returns {LogRecord}
+ */
+function loadContext(logRecord: LogRecord): LogRecord {
+  let context = apiContext.active();
+  let attributes = logRecord.attributes;
+
+  if (typeof attributes !== 'undefined') {
+    const { trace_id, span_id, trace_flags, ...otherAttributes } =
+      logRecord.attributes as Attributes & {
+        trace_id?: string;
+        span_id?: string;
+        trace_flags?: number;
+      };
+
+    if (
+      typeof trace_id !== 'undefined' &&
+      typeof span_id !== 'undefined' &&
+      typeof trace_flags !== 'undefined'
+    ) {
+      context = trace.setSpanContext(context, {
+        traceId: trace_id,
+        spanId: span_id,
+        traceFlags: trace_flags,
+        isRemote: true,
+      });
+    }
+
+    attributes = otherAttributes;
+  }
+
+  return {
+    ...logRecord,
+    attributes,
+    context,
+  };
+}
+
 export const jsonToString = (json) => {
   try {
     return JSON.stringify(json);
@@ -159,14 +206,16 @@ export class Logger {
 
   postMessage(level: string, body: string, attributes: Attributes = {}): void {
     hdx('Emitting log from HyperDX node logger...');
-    this.logger.emit({
-      // TODO: should map to otel severity number
-      severityNumber: 0,
-      // TODO: set up the mapping between different downstream log levels
-      severityText: level,
-      body,
-      attributes,
-      timestamp: this.parseTimestamp(attributes),
-    });
+    this.logger.emit(
+      loadContext({
+        // TODO: should map to otel severity number
+        severityNumber: 0,
+        // TODO: set up the mapping between different downstream log levels
+        severityText: level,
+        body,
+        attributes,
+        timestamp: this.parseTimestamp(attributes),
+      }),
+    );
   }
 }