@ai-sdk/otel 0.0.0-6b196531-20260710185421

package/src/mock-tracer.ts

@@ -0,0 +1,156 @@
+import type {
+  AttributeValue,
+  Attributes,
+  Context,
+  Exception,
+  Span,
+  SpanContext,
+  SpanOptions,
+  SpanStatus,
+  TimeInput,
+  Tracer,
+} from '@opentelemetry/api';
+
+export class MockTracer implements Tracer {
+  spans: MockSpan[] = [];
+
+  get jsonSpans() {
+    return this.spans.map(span => ({
+      name: span.name,
+      attributes: span.attributes,
+      events: span.events,
+      ...(span.status && { status: span.status }),
+    }));
+  }
+
+  startSpan(name: string, options?: SpanOptions, context?: Context): Span {
+    const span = new MockSpan({
+      name,
+      options,
+      context,
+    });
+    this.spans.push(span);
+    return span;
+  }
+
+  startActiveSpan<F extends (span: Span) => unknown>(
+    name: string,
+    arg1: unknown,
+    arg2?: unknown,
+    arg3?: F,
+  ): ReturnType<any> {
+    if (typeof arg1 === 'function') {
+      const span = new MockSpan({
+        name,
+      });
+      this.spans.push(span);
+      return arg1(span);
+    }
+    if (typeof arg2 === 'function') {
+      const span = new MockSpan({
+        name,
+        options: arg1 as SpanOptions,
+      });
+      this.spans.push(span);
+      return arg2(span);
+    }
+    if (typeof arg3 === 'function') {
+      const span = new MockSpan({
+        name,
+        options: arg1 as SpanOptions,
+        context: arg2 as Context,
+      });
+      this.spans.push(span);
+      return arg3(span);
+    }
+  }
+}
+
+class MockSpan implements Span {
+  name: string;
+  context?: Context;
+  options?: SpanOptions;
+  attributes: Attributes;
+  events: Array<{
+    name: string;
+    attributes: Attributes | undefined;
+    time?: [number, number];
+  }> = [];
+  status?: SpanStatus;
+
+  readonly _spanContext: SpanContext = new MockSpanContext();
+
+  constructor({
+    name,
+    options,
+    context,
+  }: {
+    name: string;
+    options?: SpanOptions;
+    context?: Context;
+  }) {
+    this.name = name;
+    this.context = context;
+    this.options = options;
+    this.attributes = options?.attributes ?? {};
+  }
+
+  spanContext(): SpanContext {
+    return this._spanContext;
+  }
+
+  setAttribute(key: string, value: AttributeValue): this {
+    this.attributes = { ...this.attributes, [key]: value };
+    return this;
+  }
+
+  setAttributes(attributes: Attributes): this {
+    this.attributes = { ...this.attributes, ...attributes };
+    return this;
+  }
+
+  addEvent(name: string, attributes?: Attributes): this {
+    this.events.push({ name, attributes });
+    return this;
+  }
+
+  addLink() {
+    return this;
+  }
+  addLinks() {
+    return this;
+  }
+  setStatus(status: SpanStatus) {
+    this.status = status;
+    return this;
+  }
+  updateName() {
+    return this;
+  }
+  end() {
+    return this;
+  }
+  isRecording() {
+    return false;
+  }
+  recordException(exception: Exception, time?: TimeInput) {
+    const error =
+      typeof exception === 'string' ? new Error(exception) : exception;
+    this.events.push({
+      name: 'exception',
+      attributes: {
+        'exception.type': error.constructor?.name || 'Error',
+        'exception.name': error.name || 'Error',
+        'exception.message': error.message || '',
+        'exception.stack': error.stack || '',
+      },
+      time: Array.isArray(time) ? time : [0, 0],
+    });
+  }
+}
+
+class MockSpanContext implements SpanContext {
+  traceId = 'test-trace-id';
+  spanId = 'test-span-id';
+  traceFlags = 0;
+}

package/src/noop-tracer.ts

@@ -0,0 +1,69 @@
+import type { Span, SpanContext, Tracer } from '@opentelemetry/api';
+
+/**
+ * Tracer implementation that does nothing (null object).
+ */
+export const noopTracer: Tracer = {
+  startSpan(): Span {
+    return noopSpan;
+  },
+
+  startActiveSpan<F extends (span: Span) => unknown>(
+    name: unknown,
+    arg1: unknown,
+    arg2?: unknown,
+    arg3?: F,
+  ): ReturnType<any> {
+    if (typeof arg1 === 'function') {
+      return arg1(noopSpan);
+    }
+    if (typeof arg2 === 'function') {
+      return arg2(noopSpan);
+    }
+    if (typeof arg3 === 'function') {
+      return arg3(noopSpan);
+    }
+  },
+};
+
+const noopSpan: Span = {
+  spanContext() {
+    return noopSpanContext;
+  },
+  setAttribute() {
+    return this;
+  },
+  setAttributes() {
+    return this;
+  },
+  addEvent() {
+    return this;
+  },
+  addLink() {
+    return this;
+  },
+  addLinks() {
+    return this;
+  },
+  setStatus() {
+    return this;
+  },
+  updateName() {
+    return this;
+  },
+  end() {
+    return this;
+  },
+  isRecording() {
+    return false;
+  },
+  recordException() {
+    return this;
+  },
+};
+
+const noopSpanContext: SpanContext = {
+  traceId: '',
+  spanId: '',
+  traceFlags: 0,
+};