@ydking0911/observr 0.3.1 → 0.4.7

package/dist/index.d.mts

@@ -4,6 +4,7 @@ interface ObservrEvent {
     level: "debug" | "info" | "warn" | "error";
     trace_id?: string;
     span_id?: string;
+    parent_span_id?: string;
     message: string;
     service?: string;
     duration_ms?: number;
@@ -39,10 +40,11 @@ declare class Span {
     readonly name: string;
     readonly spanId: string;
     readonly traceId: string;
+    readonly parentSpanId: string | undefined;
     private readonly transport;
     private readonly attributes;
     private startTime;
-    constructor(name: string, transport: Transport, attributes?: Record<string, unknown>);
+    constructor(name: string, transport: Transport, attributes?: Record<string, unknown>, parentSpanId?: string);
     setAttribute(key: string, value: unknown): this;
     /** Run an async function inside this span, emit on completion. */
     run<T>(fn: (span: this) => Promise<T>): Promise<T>;

package/dist/index.d.ts

@@ -4,6 +4,7 @@ interface ObservrEvent {
     level: "debug" | "info" | "warn" | "error";
     trace_id?: string;
     span_id?: string;
+    parent_span_id?: string;
     message: string;
     service?: string;
     duration_ms?: number;
@@ -39,10 +40,11 @@ declare class Span {
     readonly name: string;
     readonly spanId: string;
     readonly traceId: string;
+    readonly parentSpanId: string | undefined;
     private readonly transport;
     private readonly attributes;
     private startTime;
-    constructor(name: string, transport: Transport, attributes?: Record<string, unknown>);
+    constructor(name: string, transport: Transport, attributes?: Record<string, unknown>, parentSpanId?: string);
     setAttribute(key: string, value: unknown): this;
     /** Run an async function inside this span, emit on completion. */
     run<T>(fn: (span: this) => Promise<T>): Promise<T>;

package/dist/index.js

@@ -285,15 +285,27 @@ function unpatchConsole() {
 }
 
 // src/span.ts
+var import_node_async_hooks = require("async_hooks");
 var import_node_crypto = require("crypto");
+var _activeSpan = new import_node_async_hooks.AsyncLocalStorage();
 var Span = class {
-  constructor(name, transport, attributes = {}) {
+  constructor(name, transport, attributes = {}, parentSpanId) {
     this.startTime = 0;
     this.name = name;
     this.spanId = (0, import_node_crypto.randomBytes)(8).toString("hex");
-    this.traceId = (0, import_node_crypto.randomBytes)(16).toString("hex");
     this.transport = transport;
     this.attributes = { ...attributes };
+    const active = _activeSpan.getStore();
+    if (parentSpanId !== void 0) {
+      this.parentSpanId = parentSpanId;
+      this.traceId = active !== void 0 ? active.traceId : (0, import_node_crypto.randomBytes)(16).toString("hex");
+    } else if (active !== void 0) {
+      this.parentSpanId = active.spanId;
+      this.traceId = active.traceId;
+    } else {
+      this.parentSpanId = void 0;
+      this.traceId = (0, import_node_crypto.randomBytes)(16).toString("hex");
+    }
   }
   setAttribute(key, value) {
     this.attributes[key] = value;
@@ -302,31 +314,34 @@ var Span = class {
   /** Run an async function inside this span, emit on completion. */
   async run(fn) {
     this.startTime = performance.now();
-    let level = "info";
-    let exceptionMsg;
-    try {
-      const result = await fn(this);
-      return result;
-    } catch (err) {
-      level = "error";
-      exceptionMsg = err instanceof Error ? err.stack ?? err.message : String(err);
-      throw err;
-    } finally {
-      const durationMs = parseFloat(
-        (performance.now() - this.startTime).toFixed(2)
-      );
-      if (exceptionMsg) this.attributes["exception"] = exceptionMsg;
-      this.transport.send({
-        timestamp: (/* @__PURE__ */ new Date()).toISOString(),
-        type: "span",
-        level,
-        trace_id: this.traceId,
-        span_id: this.spanId,
-        message: this.name,
-        duration_ms: durationMs,
-        attributes: this.attributes
-      });
-    }
+    return _activeSpan.run(this, async () => {
+      let level = "info";
+      let exceptionMsg;
+      try {
+        const result = await fn(this);
+        return result;
+      } catch (err) {
+        level = "error";
+        exceptionMsg = err instanceof Error ? err.stack ?? err.message : String(err);
+        throw err;
+      } finally {
+        const durationMs = parseFloat(
+          (performance.now() - this.startTime).toFixed(2)
+        );
+        if (exceptionMsg) this.attributes["exception"] = exceptionMsg;
+        this.transport.send({
+          timestamp: (/* @__PURE__ */ new Date()).toISOString(),
+          type: "span",
+          level,
+          trace_id: this.traceId,
+          span_id: this.spanId,
+          ...this.parentSpanId !== void 0 && { parent_span_id: this.parentSpanId },
+          message: this.name,
+          duration_ms: durationMs,
+          attributes: { ...this.attributes }
+        });
+      }
+    });
   }
 };
 

package/dist/index.mjs

@@ -264,15 +264,27 @@ function unpatchConsole() {
 }
 
 // src/span.ts
+import { AsyncLocalStorage } from "async_hooks";
 import { randomBytes } from "crypto";
+var _activeSpan = new AsyncLocalStorage();
 var Span = class {
-  constructor(name, transport, attributes = {}) {
+  constructor(name, transport, attributes = {}, parentSpanId) {
     this.startTime = 0;
     this.name = name;
     this.spanId = randomBytes(8).toString("hex");
-    this.traceId = randomBytes(16).toString("hex");
     this.transport = transport;
     this.attributes = { ...attributes };
+    const active = _activeSpan.getStore();
+    if (parentSpanId !== void 0) {
+      this.parentSpanId = parentSpanId;
+      this.traceId = active !== void 0 ? active.traceId : randomBytes(16).toString("hex");
+    } else if (active !== void 0) {
+      this.parentSpanId = active.spanId;
+      this.traceId = active.traceId;
+    } else {
+      this.parentSpanId = void 0;
+      this.traceId = randomBytes(16).toString("hex");
+    }
   }
   setAttribute(key, value) {
     this.attributes[key] = value;
@@ -281,31 +293,34 @@ var Span = class {
   /** Run an async function inside this span, emit on completion. */
   async run(fn) {
     this.startTime = performance.now();
-    let level = "info";
-    let exceptionMsg;
-    try {
-      const result = await fn(this);
-      return result;
-    } catch (err) {
-      level = "error";
-      exceptionMsg = err instanceof Error ? err.stack ?? err.message : String(err);
-      throw err;
-    } finally {
-      const durationMs = parseFloat(
-        (performance.now() - this.startTime).toFixed(2)
-      );
-      if (exceptionMsg) this.attributes["exception"] = exceptionMsg;
-      this.transport.send({
-        timestamp: (/* @__PURE__ */ new Date()).toISOString(),
-        type: "span",
-        level,
-        trace_id: this.traceId,
-        span_id: this.spanId,
-        message: this.name,
-        duration_ms: durationMs,
-        attributes: this.attributes
-      });
-    }
+    return _activeSpan.run(this, async () => {
+      let level = "info";
+      let exceptionMsg;
+      try {
+        const result = await fn(this);
+        return result;
+      } catch (err) {
+        level = "error";
+        exceptionMsg = err instanceof Error ? err.stack ?? err.message : String(err);
+        throw err;
+      } finally {
+        const durationMs = parseFloat(
+          (performance.now() - this.startTime).toFixed(2)
+        );
+        if (exceptionMsg) this.attributes["exception"] = exceptionMsg;
+        this.transport.send({
+          timestamp: (/* @__PURE__ */ new Date()).toISOString(),
+          type: "span",
+          level,
+          trace_id: this.traceId,
+          span_id: this.spanId,
+          ...this.parentSpanId !== void 0 && { parent_span_id: this.parentSpanId },
+          message: this.name,
+          duration_ms: durationMs,
+          attributes: { ...this.attributes }
+        });
+      }
+    });
   }
 };
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ydking0911/observr",
-  "version": "0.3.1",
+  "version": "0.4.7",
   "description": "Zero-config observability for AI agents and developers",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
@@ -60,4 +60,4 @@
     "dist",
     "README.md"
   ]
-}
+}