@sentienguard/apm 1.0.18 → 1.0.21

package/src/spanExporter.js

@@ -1,232 +1,232 @@
-/**
- * OpenTelemetry SpanExporter — classifies finished spans into the same
- * in-memory metrics as the legacy HTTP/Mongo patches (mirrors sdk-python _exporter.py).
- */
-
-import { SpanKind, SpanStatusCode } from '@opentelemetry/api';
-import { hrTimeToMilliseconds, ExportResultCode } from '@opentelemetry/core';
-import { getAggregator } from './aggregator.js';
-import { normalizeRoute } from './normalizer.js';
-import { getServiceName, getDependencyType } from './dependencies.js';
-import config, { debug, getConfig } from './config.js';
-
-const HTTP_METHOD = 'http.method';
-const HTTP_ROUTE = 'http.route';
-const HTTP_TARGET = 'http.target';
-const HTTP_URL = 'http.url';
-const HTTP_STATUS = 'http.status_code';
-const HTTP_HOST = 'http.host';
-const NET_PEER_NAME = 'net.peer.name';
-const NET_PEER_PORT = 'net.peer.port';
-const DB_SYSTEM = 'db.system';
-const DB_COLLECTION = 'db.mongodb.collection';
-const DB_OPERATION = 'db.operation';
-
-function attr(attrs, key) {
-  if (!attrs) return undefined;
-  if (typeof attrs.get === 'function') return attrs.get(key);
-  return attrs[key];
-}
-
-function spanLatencyMs(span) {
-  if (span.endTime && span.startTime) {
-    const d = hrTimeToMilliseconds(span.endTime) - hrTimeToMilliseconds(span.startTime);
-    return d >= 0 ? d : 0;
-  }
-  return 0;
-}
-
-function isErrorSpan(span) {
-  if (span.status?.code === SpanStatusCode.ERROR) return true;
-  return false;
-}
-
-function isLocalHost(host) {
-  return host === 'localhost' || host === '127.0.0.1' || host === '::1';
-}
-
-function shouldSkipIngestHost(host) {
-  if (!host) return true;
-  try {
-    const endpointHost = new URL(config.endpoint).hostname;
-    if (host === endpointHost) return true;
-  } catch {
-    // ignore
-  }
-  return false;
-}
-
-/** HTTP client port from OTel attrs (for localhost peer labels). */
-function extractHttpPeerPort(attrs) {
-  const p = attr(attrs, NET_PEER_PORT);
-  if (p != null && p !== '') return String(p);
-  const url = attr(attrs, HTTP_URL);
-  if (url) {
-    try {
-      const u = new URL(url);
-      if (u.port) return u.port;
-    } catch {
-      // ignore
-    }
-  }
-  return '';
-}
-
-/**
- * Resolve outgoing HTTP peer label and whether to skip (ingest, OpenAI, local when disabled).
- * Caller service is always config.service (see flush payload); this is the callee label.
- */
-export function resolveHttpPeerForSpan(attrs) {
-  const host = hostFromClientSpan(attrs);
-  if (shouldSkipIngestHost(host)) return { skip: true, peerLabel: '' };
-
-  const cfg = getConfig();
-  const traceLocal = cfg.tracing?.traceLocalHttp === true;
-  const peerMap = cfg.tracing?.peerServiceMap || {};
-
-  if (host && isLocalHost(host)) {
-    if (!traceLocal) return { skip: true, peerLabel: '' };
-    const port = extractHttpPeerPort(attrs);
-    const label = (port && peerMap[port]) || (port ? `localhost:${port}` : host);
-    return { skip: false, peerLabel: label };
-  }
-
-  if (host && /openai\.com/i.test(host)) return { skip: true, peerLabel: '' };
-
-  const peerLabel = host ? getServiceName(host) : 'unknown';
-  return { skip: false, peerLabel };
-}
-
-function hostFromClientSpan(attrs) {
-  let host = attr(attrs, NET_PEER_NAME) || attr(attrs, HTTP_HOST) || '';
-  if (!host) {
-    const url = attr(attrs, HTTP_URL);
-    if (url) {
-      try {
-        host = new URL(url).hostname || '';
-      } catch {
-        host = '';
-      }
-    }
-  }
-  return host;
-}
-
-function normalizeMongoOp(op) {
-  const o = String(op || '').toLowerCase();
-  if (o.includes('find')) return 'find';
-  if (o.includes('insert')) return 'insert';
-  if (o.includes('update')) return 'update';
-  if (o.includes('delete') || o.includes('remove')) return 'delete';
-  if (o === 'aggregate') return 'aggregate';
-  if (o.includes('count')) return 'count';
-  if (o.includes('index')) return 'index';
-  return o || 'unknown';
-}
-
-function dbDepInfo(system) {
-  const mapping = {
-    postgresql: ['PostgreSQL', 'db'],
-    mysql: ['MySQL', 'db'],
-    sqlite: ['SQLite', 'db'],
-    mssql: ['SQL Server', 'db'],
-    oracle: ['Oracle', 'db'],
-    redis: ['Redis', 'cache'],
-    memcached: ['Memcached', 'cache']
-  };
-  return mapping[system] || [system, 'db'];
-}
-
-/**
- * Map one OTel span into SentienGuard aggregator buckets (same rules as Python SDK).
- */
-export function classifyAndRecordSpan(span) {
-  const attrs = span.attributes || {};
-  const latency = spanLatencyMs(span);
-  let error = isErrorSpan(span);
-  const kind = span.kind;
-
-  if (kind === SpanKind.SERVER) {
-    const method = attr(attrs, HTTP_METHOD) || 'GET';
-    let route = attr(attrs, HTTP_ROUTE) || attr(attrs, HTTP_TARGET) || '/';
-    route = normalizeRoute(typeof route === 'string' ? route : '/');
-    const status = parseInt(String(attr(attrs, HTTP_STATUS) || 0), 10);
-    if (status >= 400) error = true;
-    getAggregator().recordRequest(method, route, latency, error);
-    return;
-  }
-
-  if (kind === SpanKind.CLIENT) {
-    const dbSystem = attr(attrs, DB_SYSTEM);
-
-    if (dbSystem === 'mongodb') {
-      const collection = attr(attrs, DB_COLLECTION) || 'unknown';
-      const operation = normalizeMongoOp(attr(attrs, DB_OPERATION) || span.name);
-      getAggregator().recordMongoOperation(collection, operation, latency, error);
-      return;
-    }
-
-    if (dbSystem) {
-      const [name, dtype] = dbDepInfo(dbSystem);
-      getAggregator().recordDependency(name, dtype, latency, error);
-      return;
-    }
-
-    const { skip, peerLabel } = resolveHttpPeerForSpan(attrs);
-    if (skip) return;
-
-    const host = hostFromClientSpan(attrs);
-    const dtype = getDependencyType(host || '', '');
-    const status = parseInt(String(attr(attrs, HTTP_STATUS) || 0), 10);
-    if (status >= 400) error = true;
-    debug(
-      `Service call: ${config.service} -> ${peerLabel} ${latency.toFixed(2)}ms` +
-        (status ? ` (HTTP ${status})` : '')
-    );
-    getAggregator().recordDependency(peerLabel, dtype, latency, error);
-    return;
-  }
-
-  const nameLower = (span.name || '').toLowerCase();
-  if (nameLower.includes('openai') || attr(attrs, 'gen_ai.system') === 'openai') {
-    const operation = attr(attrs, 'gen_ai.operation.name') || span.name;
-    const model = attr(attrs, 'gen_ai.request.model') || 'unknown';
-    const promptTokens = parseInt(String(attr(attrs, 'gen_ai.usage.prompt_tokens') || 0), 10);
-    const completionTokens = parseInt(String(attr(attrs, 'gen_ai.usage.completion_tokens') || 0), 10);
-    const totalTokens = promptTokens + completionTokens;
-    getAggregator().recordOpenAIOperation({
-      operation,
-      model,
-      latency,
-      promptTokens,
-      completionTokens,
-      totalTokens,
-      cost: 0,
-      error: error ? 'error' : null,
-      statusCode: parseInt(String(attr(attrs, HTTP_STATUS) || 200), 10)
-    });
-  }
-}
-
-export class SentienGuardSpanExporter {
-  export(spans, resultCallback) {
-    try {
-      for (const span of spans) {
-        try {
-          classifyAndRecordSpan(span);
-        } catch {
-          // never break the app
-        }
-      }
-      resultCallback({ code: ExportResultCode.SUCCESS });
-    } catch (err) {
-      resultCallback({ code: ExportResultCode.FAILED, error: err });
-    }
-  }
-
-  shutdown() {
-    return Promise.resolve();
-  }
-}
-
-export default SentienGuardSpanExporter;
+/**
+ * OpenTelemetry SpanExporter — classifies finished spans into the same
+ * in-memory metrics as the legacy HTTP/Mongo patches (mirrors sdk-python _exporter.py).
+ */
+
+import { SpanKind, SpanStatusCode } from '@opentelemetry/api';
+import { hrTimeToMilliseconds, ExportResultCode } from '@opentelemetry/core';
+import { getAggregator } from './aggregator.js';
+import { normalizeRoute } from './normalizer.js';
+import { getServiceName, getDependencyType } from './dependencies.js';
+import config, { debug, getConfig } from './config.js';
+
+const HTTP_METHOD = 'http.method';
+const HTTP_ROUTE = 'http.route';
+const HTTP_TARGET = 'http.target';
+const HTTP_URL = 'http.url';
+const HTTP_STATUS = 'http.status_code';
+const HTTP_HOST = 'http.host';
+const NET_PEER_NAME = 'net.peer.name';
+const NET_PEER_PORT = 'net.peer.port';
+const DB_SYSTEM = 'db.system';
+const DB_COLLECTION = 'db.mongodb.collection';
+const DB_OPERATION = 'db.operation';
+
+function attr(attrs, key) {
+  if (!attrs) return undefined;
+  if (typeof attrs.get === 'function') return attrs.get(key);
+  return attrs[key];
+}
+
+function spanLatencyMs(span) {
+  if (span.endTime && span.startTime) {
+    const d = hrTimeToMilliseconds(span.endTime) - hrTimeToMilliseconds(span.startTime);
+    return d >= 0 ? d : 0;
+  }
+  return 0;
+}
+
+function isErrorSpan(span) {
+  if (span.status?.code === SpanStatusCode.ERROR) return true;
+  return false;
+}
+
+function isLocalHost(host) {
+  return host === 'localhost' || host === '127.0.0.1' || host === '::1';
+}
+
+function shouldSkipIngestHost(host) {
+  if (!host) return true;
+  try {
+    const endpointHost = new URL(config.endpoint).hostname;
+    if (host === endpointHost) return true;
+  } catch {
+    // ignore
+  }
+  return false;
+}
+
+/** HTTP client port from OTel attrs (for localhost peer labels). */
+function extractHttpPeerPort(attrs) {
+  const p = attr(attrs, NET_PEER_PORT);
+  if (p != null && p !== '') return String(p);
+  const url = attr(attrs, HTTP_URL);
+  if (url) {
+    try {
+      const u = new URL(url);
+      if (u.port) return u.port;
+    } catch {
+      // ignore
+    }
+  }
+  return '';
+}
+
+/**
+ * Resolve outgoing HTTP peer label and whether to skip (ingest, OpenAI, local when disabled).
+ * Caller service is always config.service (see flush payload); this is the callee label.
+ */
+export function resolveHttpPeerForSpan(attrs) {
+  const host = hostFromClientSpan(attrs);
+  if (shouldSkipIngestHost(host)) return { skip: true, peerLabel: '' };
+
+  const cfg = getConfig();
+  const traceLocal = cfg.tracing?.traceLocalHttp === true;
+  const peerMap = cfg.tracing?.peerServiceMap || {};
+
+  if (host && isLocalHost(host)) {
+    if (!traceLocal) return { skip: true, peerLabel: '' };
+    const port = extractHttpPeerPort(attrs);
+    const label = (port && peerMap[port]) || (port ? `localhost:${port}` : host);
+    return { skip: false, peerLabel: label };
+  }
+
+  if (host && /openai\.com/i.test(host)) return { skip: true, peerLabel: '' };
+
+  const peerLabel = host ? getServiceName(host) : 'unknown';
+  return { skip: false, peerLabel };
+}
+
+function hostFromClientSpan(attrs) {
+  let host = attr(attrs, NET_PEER_NAME) || attr(attrs, HTTP_HOST) || '';
+  if (!host) {
+    const url = attr(attrs, HTTP_URL);
+    if (url) {
+      try {
+        host = new URL(url).hostname || '';
+      } catch {
+        host = '';
+      }
+    }
+  }
+  return host;
+}
+
+function normalizeMongoOp(op) {
+  const o = String(op || '').toLowerCase();
+  if (o.includes('find')) return 'find';
+  if (o.includes('insert')) return 'insert';
+  if (o.includes('update')) return 'update';
+  if (o.includes('delete') || o.includes('remove')) return 'delete';
+  if (o === 'aggregate') return 'aggregate';
+  if (o.includes('count')) return 'count';
+  if (o.includes('index')) return 'index';
+  return o || 'unknown';
+}
+
+function dbDepInfo(system) {
+  const mapping = {
+    postgresql: ['PostgreSQL', 'db'],
+    mysql: ['MySQL', 'db'],
+    sqlite: ['SQLite', 'db'],
+    mssql: ['SQL Server', 'db'],
+    oracle: ['Oracle', 'db'],
+    redis: ['Redis', 'cache'],
+    memcached: ['Memcached', 'cache']
+  };
+  return mapping[system] || [system, 'db'];
+}
+
+/**
+ * Map one OTel span into SentienGuard aggregator buckets (same rules as Python SDK).
+ */
+export function classifyAndRecordSpan(span) {
+  const attrs = span.attributes || {};
+  const latency = spanLatencyMs(span);
+  let error = isErrorSpan(span);
+  const kind = span.kind;
+
+  if (kind === SpanKind.SERVER) {
+    const method = attr(attrs, HTTP_METHOD) || 'GET';
+    let route = attr(attrs, HTTP_ROUTE) || attr(attrs, HTTP_TARGET) || '/';
+    route = normalizeRoute(typeof route === 'string' ? route : '/');
+    const status = parseInt(String(attr(attrs, HTTP_STATUS) || 0), 10);
+    if (status >= 400) error = true;
+    getAggregator().recordRequest(method, route, latency, error);
+    return;
+  }
+
+  if (kind === SpanKind.CLIENT) {
+    const dbSystem = attr(attrs, DB_SYSTEM);
+
+    if (dbSystem === 'mongodb') {
+      const collection = attr(attrs, DB_COLLECTION) || 'unknown';
+      const operation = normalizeMongoOp(attr(attrs, DB_OPERATION) || span.name);
+      getAggregator().recordMongoOperation(collection, operation, latency, error);
+      return;
+    }
+
+    if (dbSystem) {
+      const [name, dtype] = dbDepInfo(dbSystem);
+      getAggregator().recordDependency(name, dtype, latency, error);
+      return;
+    }
+
+    const { skip, peerLabel } = resolveHttpPeerForSpan(attrs);
+    if (skip) return;
+
+    const host = hostFromClientSpan(attrs);
+    const dtype = getDependencyType(host || '', '');
+    const status = parseInt(String(attr(attrs, HTTP_STATUS) || 0), 10);
+    if (status >= 400) error = true;
+    debug(
+      `Service call: ${config.service} -> ${peerLabel} ${latency.toFixed(2)}ms` +
+        (status ? ` (HTTP ${status})` : '')
+    );
+    getAggregator().recordDependency(peerLabel, dtype, latency, error);
+    return;
+  }
+
+  const nameLower = (span.name || '').toLowerCase();
+  if (nameLower.includes('openai') || attr(attrs, 'gen_ai.system') === 'openai') {
+    const operation = attr(attrs, 'gen_ai.operation.name') || span.name;
+    const model = attr(attrs, 'gen_ai.request.model') || 'unknown';
+    const promptTokens = parseInt(String(attr(attrs, 'gen_ai.usage.prompt_tokens') || 0), 10);
+    const completionTokens = parseInt(String(attr(attrs, 'gen_ai.usage.completion_tokens') || 0), 10);
+    const totalTokens = promptTokens + completionTokens;
+    getAggregator().recordOpenAIOperation({
+      operation,
+      model,
+      latency,
+      promptTokens,
+      completionTokens,
+      totalTokens,
+      cost: 0,
+      error: error ? 'error' : null,
+      statusCode: parseInt(String(attr(attrs, HTTP_STATUS) || 200), 10)
+    });
+  }
+}
+
+export class SentienGuardSpanExporter {
+  export(spans, resultCallback) {
+    try {
+      for (const span of spans) {
+        try {
+          classifyAndRecordSpan(span);
+        } catch {
+          // never break the app
+        }
+      }
+      resultCallback({ code: ExportResultCode.SUCCESS });
+    } catch (err) {
+      resultCallback({ code: ExportResultCode.FAILED, error: err });
+    }
+  }
+
+  shutdown() {
+    return Promise.resolve();
+  }
+}
+
+export default SentienGuardSpanExporter;