@sentienguard/apm 1.0.18 → 1.0.21

package/src/traceSpanExporter.js

@@ -1,165 +1,186 @@
-/**
- * SpanExporter that ships sampled raw spans to SentienGuard trace ingest.
- *
- * This is intentionally "lossy": it enqueues serialized spans to an async transport
- * and returns SUCCESS quickly to avoid blocking the app.
- */
-
-import { ExportResultCode, hrTimeToMilliseconds } from '@opentelemetry/core';
-import { SpanStatusCode } from '@opentelemetry/api';
-import { enqueueSpans } from './traceTransport.js';
-import { getConfig } from './config.js';
-
-let debugLogged = 0;
-const DEBUG_LOG_LIMIT = 10;
-
-function debugSpanShape(span, serialized, cfg) {
-  try {
-    if (!cfg?.debug) return;
-    if (debugLogged >= DEBUG_LOG_LIMIT) return;
-    debugLogged++;
-
-    const attrObj = serialized?.attributes && typeof serialized.attributes === 'object' ? serialized.attributes : {};
-    const keys = Object.keys(attrObj);
-    const sample = keys.slice(0, 12).reduce((acc, k) => {
-      acc[k] = attrObj[k];
-      return acc;
-    }, {});
-
-    // eslint-disable-next-line no-console
-    console.log(
-      '[SentienGuard APM][trace-debug]',
-      JSON.stringify(
-        {
-          name: serialized?.name,
-          kind: serialized?.kind,
-          status: serialized?.status?.code,
-          trace_id: serialized?.trace_id,
-          span_id: serialized?.span_id,
-          parent_span_id: serialized?.parent_span_id,
-          attributes_count: keys.length,
-          attributes_sample: sample
-        },
-        null,
-        0
-      )
-    );
-  } catch {
-    // never break the app
-  }
-}
-
-function hrTimeToUnixNanoString(hrTime) {
-  // hrTime is [seconds, nanoseconds]
-  if (!Array.isArray(hrTime) || hrTime.length !== 2) return '';
-  const sec = BigInt(hrTime[0] || 0);
-  const ns = BigInt(hrTime[1] || 0);
-  return String(sec * 1000000000n + ns);
-}
-
-function statusForSpan(span) {
-  const code = span?.status?.code;
-  if (code === SpanStatusCode.ERROR) return { code: 'ERROR', message: span.status?.message || '' };
-  if (code === SpanStatusCode.OK) return { code: 'OK', message: span.status?.message || '' };
-  return { code: 'UNSET', message: span?.status?.message || '' };
-}
-
-function safeAttrs(attrs) {
-  if (!attrs) return {};
-  if (typeof attrs.get === 'function') {
-    const out = {};
-    for (const [k, v] of attrs.entries()) out[k] = v;
-    return out;
-  }
-  if (typeof attrs === 'object') return attrs;
-  return {};
-}
-
-function shouldSampleTraceId(traceId, sampleRate) {
-  if (sampleRate == null) return true;
-  const r = Number(sampleRate);
-  if (!Number.isFinite(r)) return true;
-  if (r <= 0) return false;
-  if (r >= 1) return true;
-
-  // Deterministic sampling based on trace_id (stable across services).
-  // Use the first 8 hex chars (32 bits) -> [0,1).
-  try {
-    const prefix = String(traceId).slice(0, 8);
-    if (!/^[0-9a-f]{8}$/i.test(prefix)) return Math.random() < r;
-    const n = parseInt(prefix, 16) >>> 0;
-    const p = n / 0x100000000; // 2^32
-    return p < r;
-  } catch {
-    return Math.random() < r;
-  }
-}
-
-function serializeSpan(span) {
-  const ctx = span?.spanContext?.();
-  if (!ctx?.traceId || !ctx?.spanId) return null;
-
-  const startNano = hrTimeToUnixNanoString(span.startTime);
-  const endNano = hrTimeToUnixNanoString(span.endTime);
-  if (!startNano || !endNano) return null;
-
-  const parentSpanId = span?.parentSpanId || span?.parentSpanContext?.spanId || null;
-  const status = statusForSpan(span);
-
-  const durationMs =
-    span.endTime && span.startTime
-      ? Math.max(0, hrTimeToMilliseconds(span.endTime) - hrTimeToMilliseconds(span.startTime))
-      : 0;
-
-  return {
-    trace_id: ctx.traceId,
-    span_id: ctx.spanId,
-    parent_span_id: parentSpanId || null,
-    name: span.name || '',
-    kind: span.kind != null ? String(span.kind) : undefined,
-    start_time_unix_nano: startNano,
-    end_time_unix_nano: endNano,
-    status,
-    attributes: safeAttrs(span.attributes),
-    events: Array.isArray(span.events) ? span.events : [],
-    links: Array.isArray(span.links) ? span.links : [],
-    duration_ms: Math.round(durationMs)
-  };
-}
-
-export class SentienGuardTraceSpanExporter {
-  export(spans, resultCallback) {
-    try {
-      const cfg = getConfig();
-      const rate = cfg?.tracing?.sampleRate;
-
-      const serialized = [];
-      for (const span of spans) {
-        try {
-          const s = serializeSpan(span);
-          if (s && shouldSampleTraceId(s.trace_id, rate)) {
-            debugSpanShape(span, s, cfg);
-            serialized.push(s);
-          }
-        } catch {
-          // ignore
-        }
-      }
-
-      if (serialized.length) {
-        enqueueSpans(serialized);
-      }
-
-      resultCallback({ code: ExportResultCode.SUCCESS });
-    } catch (err) {
-      resultCallback({ code: ExportResultCode.FAILED, error: err });
-    }
-  }
-
-  shutdown() {
-    return Promise.resolve();
-  }
-}
-
-export default SentienGuardTraceSpanExporter;
-
+/**
+ * SpanExporter that ships sampled raw spans to SentienGuard trace ingest.
+ *
+ * This is intentionally "lossy": it enqueues serialized spans to an async transport
+ * and returns SUCCESS quickly to avoid blocking the app.
+ */
+
+import { ExportResultCode, hrTimeToMilliseconds } from '@opentelemetry/core';
+import { SpanStatusCode } from '@opentelemetry/api';
+import { enqueueSpans } from './traceTransport.js';
+import { getConfig } from './config.js';
+
+let debugLogged = 0;
+const DEBUG_LOG_LIMIT = 10;
+
+function debugSpanShape(span, serialized, cfg) {
+  try {
+    if (!cfg?.debug) return;
+    if (debugLogged >= DEBUG_LOG_LIMIT) return;
+    debugLogged++;
+
+    const attrObj = serialized?.attributes && typeof serialized.attributes === 'object' ? serialized.attributes : {};
+    const keys = Object.keys(attrObj);
+    const sample = keys.slice(0, 12).reduce((acc, k) => {
+      acc[k] = attrObj[k];
+      return acc;
+    }, {});
+
+    // eslint-disable-next-line no-console
+    console.log(
+      '[SentienGuard APM][trace-debug]',
+      JSON.stringify(
+        {
+          name: serialized?.name,
+          kind: serialized?.kind,
+          status: serialized?.status?.code,
+          trace_id: serialized?.trace_id,
+          span_id: serialized?.span_id,
+          parent_span_id: serialized?.parent_span_id,
+          attributes_count: keys.length,
+          attributes_sample: sample
+        },
+        null,
+        0
+      )
+    );
+  } catch {
+    // never break the app
+  }
+}
+
+function hrTimeToUnixNanoString(hrTime) {
+  // hrTime is [seconds, nanoseconds]
+  if (!Array.isArray(hrTime) || hrTime.length !== 2) return '';
+  const sec = BigInt(hrTime[0] || 0);
+  const ns = BigInt(hrTime[1] || 0);
+  return String(sec * 1000000000n + ns);
+}
+
+function statusForSpan(span) {
+  const code = span?.status?.code;
+  if (code === SpanStatusCode.ERROR) return { code: 'ERROR', message: span.status?.message || '' };
+  if (code === SpanStatusCode.OK) return { code: 'OK', message: span.status?.message || '' };
+  return { code: 'UNSET', message: span?.status?.message || '' };
+}
+
+function safeAttrs(attrs) {
+  if (!attrs) return {};
+  if (typeof attrs.get === 'function') {
+    const out = {};
+    for (const [k, v] of attrs.entries()) out[k] = v;
+    return out;
+  }
+  if (typeof attrs === 'object') return attrs;
+  return {};
+}
+
+function shouldSampleTraceId(traceId, sampleRate) {
+  if (sampleRate == null) return true;
+  const r = Number(sampleRate);
+  if (!Number.isFinite(r)) return true;
+  if (r <= 0) return false;
+  if (r >= 1) return true;
+
+  // Deterministic sampling based on trace_id (stable across services).
+  // Use the first 8 hex chars (32 bits) -> [0,1).
+  try {
+    const prefix = String(traceId).slice(0, 8);
+    if (!/^[0-9a-f]{8}$/i.test(prefix)) return Math.random() < r;
+    const n = parseInt(prefix, 16) >>> 0;
+    const p = n / 0x100000000; // 2^32
+    return p < r;
+  } catch {
+    return Math.random() < r;
+  }
+}
+
+function getParentSpanId(span) {
+  try {
+    const direct = span?.parentSpanId;
+    if (typeof direct === 'string' && direct) return direct;
+
+    const psc = span?.parentSpanContext;
+    if (psc && typeof psc === 'object') {
+      const id = psc.spanId;
+      if (typeof id === 'string' && id) return id;
+    }
+    if (typeof psc === 'function') {
+      const ctx = psc();
+      const id = ctx?.spanId;
+      if (typeof id === 'string' && id) return id;
+    }
+  } catch {
+    // ignore
+  }
+  return null;
+}
+
+function serializeSpan(span) {
+  const ctx = span?.spanContext?.();
+  if (!ctx?.traceId || !ctx?.spanId) return null;
+
+  const startNano = hrTimeToUnixNanoString(span.startTime);
+  const endNano = hrTimeToUnixNanoString(span.endTime);
+  if (!startNano || !endNano) return null;
+
+  const parentSpanId = getParentSpanId(span);
+  const status = statusForSpan(span);
+
+  const durationMs =
+    span.endTime && span.startTime
+      ? Math.max(0, hrTimeToMilliseconds(span.endTime) - hrTimeToMilliseconds(span.startTime))
+      : 0;
+
+  return {
+    trace_id: ctx.traceId,
+    span_id: ctx.spanId,
+    parent_span_id: parentSpanId || null,
+    name: span.name || '',
+    kind: span.kind != null ? String(span.kind) : undefined,
+    start_time_unix_nano: startNano,
+    end_time_unix_nano: endNano,
+    status,
+    attributes: safeAttrs(span.attributes),
+    events: Array.isArray(span.events) ? span.events : [],
+    links: Array.isArray(span.links) ? span.links : [],
+    duration_ms: Math.round(durationMs)
+  };
+}
+
+export class SentienGuardTraceSpanExporter {
+  export(spans, resultCallback) {
+    try {
+      const cfg = getConfig();
+      const rate = cfg?.tracing?.sampleRate;
+
+      const serialized = [];
+      for (const span of spans) {
+        try {
+          const s = serializeSpan(span);
+          if (s && shouldSampleTraceId(s.trace_id, rate)) {
+            debugSpanShape(span, s, cfg);
+            serialized.push(s);
+          }
+        } catch {
+          // ignore
+        }
+      }
+
+      if (serialized.length) {
+        enqueueSpans(serialized);
+      }
+
+      resultCallback({ code: ExportResultCode.SUCCESS });
+    } catch (err) {
+      resultCallback({ code: ExportResultCode.FAILED, error: err });
+    }
+  }
+
+  shutdown() {
+    return Promise.resolve();
+  }
+}
+
+export default SentienGuardTraceSpanExporter;
+