@mastra/observability 1.17.2-alpha.1 → 1.17.2-alpha.2

package/CHANGELOG.md

@@ -1,5 +1,21 @@
 # @mastra/observability
 
+## 1.17.2-alpha.2
+
+### Patch Changes
+
+- Added `span.endTree()` for closing a span together with every descendant span that is still open, so an operation that is abandoned rather than completed can still emit a full trace. The options you pass are applied to every span it closes, so a force-closed child is distinguishable from one that finished on its own. ([#22278](https://github.com/mastra-ai/mastra/pull/22278))
+
+  ```ts
+  // Ends the span and any child spans still open beneath it, marking each canceled
+  workflowSpan.endTree({ attributes: { status: 'canceled' } });
+  ```
+
+  Repeat calls to `span.end()` are now ignored. A span that was force-closed this way keeps the state it was closed with and reports its end exactly once, even if the work it covered finishes later and ends the span again.
+
+- Updated dependencies [[`b05f486`](https://github.com/mastra-ai/mastra/commit/b05f48612984d5fe2447ea2d6cdd5c604d285b97), [`7960688`](https://github.com/mastra-ai/mastra/commit/7960688828e04eaf3106e34f7758fa580257eef6)]:
+  - @mastra/core@1.62.0-alpha.10
+
 ## 1.17.2-alpha.1
 
 ### Patch Changes

package/dist/index.cjs

@@ -2718,7 +2718,7 @@ function getExternalParentId(options) {
 	if (parent.isInternal || parent.isExcluded) return parent.getParentSpanId(false);
 	return parent.id;
 }
-var BaseSpan = class {
+var BaseSpan = class BaseSpan {
 	name;
 	type;
 	attributes;
@@ -2768,6 +2768,8 @@ var BaseSpan = class {
 	isExcluded;
 	/** Cached canonical correlation context for this live span */
 	correlationContext;
+	/** Child spans that have started but not yet ended */
+	#openChildren;
 	/**
 	* Subclasses can override to unconditionally mark the span as excluded.
 	* NoOpSpan uses this because it is never exported regardless of config.
@@ -2789,6 +2791,8 @@ var BaseSpan = class {
 		this.isEvent = options.isEvent ?? false;
 		this.tracingPolicy = options.tracingPolicy;
 		this.traceState = options.traceState;
+		const parent = options.parent;
+		if (!this.isEvent && parent instanceof BaseSpan && parent.isValid) (parent.#openChildren ??= /* @__PURE__ */ new Set()).add(this);
 		this.tags = !options.parent && options.tags?.length ? options.tags : void 0;
 		const entityParent = this.getParentSpan(false);
 		this.entityType = options.entityType ?? entityParent?.entityType;
@@ -2826,6 +2830,16 @@ var BaseSpan = class {
 			return value;
 		}
 	}
+	/** End the span and any descendant spans that are still open, applying `options` to each */
+	endTree(options) {
+		if (this.#openChildren) for (const child of [...this.#openChildren]) child.endTree(options);
+		this.end(options);
+	}
+	/** Release this span from its parent's open-child set once it has ended */
+	detachFromParent() {
+		const parent = this.parent;
+		if (parent instanceof BaseSpan) parent.#openChildren?.delete(this);
+	}
 	createChildSpan(options) {
 		return this.observabilityInstance.startSpan({
 			...options,
@@ -3029,7 +3043,9 @@ var DefaultSpan = class extends BaseSpan {
 	}
 	end(options) {
 		if (this.isEvent) return;
+		if (this.endTime) return;
 		this.endTime = /* @__PURE__ */ new Date();
+		this.detachFromParent();
 		if (options?.metadata) this.metadata = {
 			...this.metadata,
 			...deepClean(this.prepareSpanMetadata(options.metadata), this.deepCleanOptions)
@@ -3485,6 +3501,7 @@ var BaseObservabilityInstance = class extends _mastra_core_base.MastraBase {
 				this.logger.warn(`End event is not available on event spans`);
 				return;
 			}
+			if (span.endTime) return;
 			const rollupTarget = this.captureModelUsageRollup(span, options);
 			const excludedModelUsage = this.captureExcludedModelUsage(span, options);
 			originalEnd(options);