@mastra/duckdb 1.11.0 → 1.11.1-alpha.1

package/dist/{observability-CDtl20Mc.js → observability-BVE7R9Oz.js}

@@ -3773,6 +3773,14 @@ function reconstructForAnchorsWithBoundParam(reconstructSelect, anchorCte) {
       AND (traceId, spanId) IN (SELECT traceId, spanId FROM ${anchorCte})
     GROUP BY traceId, spanId`;
 }
+/**
+* Event spans are point-in-time, so they end at the instant they start. Rows
+* written before event spans carried an end time have a null endedAt.
+*/
+function rowEndedAt(row) {
+	if (row.isEvent) return toDate(row.startedAt);
+	return toDateOrNull(row.endedAt);
+}
 function rowToLightSpanRecord(row) {
 	return {
 		traceId: row.traceId,
@@ -3782,13 +3790,13 @@ function rowToLightSpanRecord(row) {
 		parentSpanId: row.parentSpanId ?? null,
 		isEvent: row.isEvent,
 		startedAt: toDate(row.startedAt),
-		endedAt: toDateOrNull(row.endedAt),
+		endedAt: rowEndedAt(row),
 		entityType: row.entityType ?? null,
 		entityId: row.entityId ?? null,
 		entityName: row.entityName ?? null,
 		error: parseJson(row.error),
 		createdAt: toDate(row.startedAt),
-		updatedAt: toDateOrNull(row.endedAt)
+		updatedAt: rowEndedAt(row)
 	};
 }
 function rowToLightSpanRecordWithPreview(row) {
@@ -3809,7 +3817,7 @@ function rowToSpanRecord(row) {
 		parentSpanId: row.parentSpanId ?? null,
 		isEvent: row.isEvent,
 		startedAt: toDate(row.startedAt),
-		endedAt: toDateOrNull(row.endedAt),
+		endedAt: rowEndedAt(row),
 		experimentId: row.experimentId ?? null,
 		entityType: row.entityType ?? null,
 		entityId: row.entityId ?? null,
@@ -4010,7 +4018,7 @@ function createStartSpanRow(s) {
 		name: s.name,
 		spanType: s.spanType,
 		isEvent: s.isEvent,
-		endedAt: null,
+		endedAt: s.isEvent ? s.startedAt : null,
 		experimentId: s.experimentId ?? null,
 		entityType: s.entityType ?? null,
 		entityId: s.entityId ?? null,
@@ -4081,7 +4089,7 @@ function createEndSpanRow(s) {
 /** Insert a 'start' event for a new span. */
 async function createSpan(db, args) {
 	const rows = [createStartSpanRow(args.span)];
-	if (args.span.endedAt) rows.push(createEndSpanRow(args.span));
+	if (args.span.endedAt && !args.span.isEvent) rows.push(createEndSpanRow(args.span));
 	await insertSpanEvents(db, rows);
 }
 /** Insert 'start' events for multiple spans in a single statement. */
@@ -4089,7 +4097,7 @@ async function batchCreateSpans(db, args) {
 	if (args.records.length === 0) return;
 	await insertSpanEvents(db, args.records.flatMap((record) => {
 		const events = [createStartSpanRow(record)];
-		if (record.endedAt) events.push(createEndSpanRow(record));
+		if (record.endedAt && !record.isEvent) events.push(createEndSpanRow(record));
 		return events;
 	}));
 }
@@ -4187,10 +4195,10 @@ async function listTraceRows(db, args, reconstructSelect, mapRow, toSpans) {
 	if (orderDir !== "ASC" && orderDir !== "DESC") throw new Error(`Invalid sort direction: ${orderBy.direction}`);
 	const canOrderInPrefilter = SAFE_PREFILTER_ORDER_FIELDS.has(orderBy.field);
 	if (!(Object.keys(postAgg).length > 0 || hasChildError !== void 0 || !canOrderInPrefilter)) {
-		const prefilterOrderBy = `ORDER BY timestamp ${orderDir}`;
+		const prefilterOrderBy = `ORDER BY anchorStartedAt ${orderDir}, traceId, spanId`;
 		const offset = page * perPage;
 		const countSql = `
-      SELECT COUNT(*) as total
+      SELECT COUNT(DISTINCT (traceId, spanId)) as total
       FROM span_events AS ${outerAlias}
       ${prefilterWhere}
     `;
@@ -4198,14 +4206,15 @@ async function listTraceRows(db, args, reconstructSelect, mapRow, toSpans) {
 		const total = Number(countResult[0]?.total ?? 0);
 		const pageSql = `
       WITH page_roots AS (
-        SELECT traceId, spanId, timestamp AS anchorStartedAt
+        SELECT traceId, spanId, min(timestamp) AS anchorStartedAt
         FROM span_events AS ${outerAlias}
         ${prefilterWhere}
+        GROUP BY traceId, spanId
         ${prefilterOrderBy}
         LIMIT ? OFFSET ?
       )
       ${reconstructForAnchors(reconstructSelect, "page_roots")}
-      ${buildOrderByClause(orderBy)}
+      ${buildOrderByClause(orderBy)}, traceId, spanId
     `;
 		const spans = (await db.query(pageSql, [
 			...prefilterParams,
@@ -4318,9 +4327,10 @@ async function listTraces(db, args) {
 		};
 		const dataSql = `
       WITH candidate_roots AS (
-        SELECT traceId, spanId, cursorId
+        SELECT traceId, spanId, max(cursorId) AS cursorId
         FROM span_events AS ${outerAlias}
         ${prefilterWhere}
+        GROUP BY traceId, spanId
       ),
       root_spans AS (
         SELECT reconstructed.*, candidate_roots.cursorId AS anchorCursorId
@@ -4494,9 +4504,10 @@ async function listBranches(db, args) {
 		};
 		const dataSql = `
       WITH candidate_anchors AS (
-        SELECT traceId, spanId, cursorId
+        SELECT traceId, spanId, max(cursorId) AS cursorId
         FROM span_events AS ${outerAlias}
         ${prefilterWhere}
+        GROUP BY traceId, spanId
       ),
       branch_anchors AS (
         SELECT reconstructed.*, candidate_anchors.cursorId AS anchorCursorId
@@ -4547,10 +4558,10 @@ async function listBranches(db, args) {
 	const currentDeltaCursor = deltaPollingFeatureEnabled() ? await getBranchDeltaCursor(db, filters) : void 0;
 	const canOrderInPrefilter = SAFE_PREFILTER_ORDER_FIELDS.has(orderBy.field);
 	if (!(Object.keys(postAgg).length > 0 || !canOrderInPrefilter)) {
-		const prefilterOrderBy = `ORDER BY timestamp ${orderDir}`;
+		const prefilterOrderBy = `ORDER BY anchorStartedAt ${orderDir}, traceId, spanId`;
 		const offset = page * perPage;
 		const countSql = `
-      SELECT COUNT(*) as total
+      SELECT COUNT(DISTINCT (traceId, spanId)) as total
       FROM span_events AS ${outerAlias}
       ${prefilterWhere}
     `;
@@ -4568,14 +4579,15 @@ async function listBranches(db, args) {
 		};
 		const pageSql = `
       WITH page_anchors AS (
-        SELECT traceId, spanId, timestamp AS anchorStartedAt
+        SELECT traceId, spanId, min(timestamp) AS anchorStartedAt
         FROM span_events AS ${outerAlias}
         ${prefilterWhere}
+        GROUP BY traceId, spanId
         ${prefilterOrderBy}
         LIMIT ? OFFSET ?
       )
       ${reconstructForAnchors(SPAN_RECONSTRUCT_SELECT, "page_anchors")}
-      ${buildOrderByClause(orderBy)}
+      ${buildOrderByClause(orderBy)}, traceId, spanId
     `;
 		const spans = (await db.query(pageSql, [
 			...prefilterParams,
@@ -4870,21 +4882,35 @@ var ObservabilityStorageDuckDB = class ObservabilityStorageDuckDB extends Observ
 		if (!deltaPollingFeatureEnabled()) return [
 			"metrics",
 			"logs",
+			"entity-type-discovery",
+			"entity-name-discovery",
+			"service-name-discovery",
+			"environment-discovery",
+			"tag-discovery",
+			"metric-discovery",
 			"trace-query",
 			"trace-query-root-duration",
 			"trace-query-discovery",
 			"thread-query",
-			"trace-query-tenant-scope"
+			"trace-query-tenant-scope",
+			"feedback"
 		];
 		return [
 			"metrics",
 			"logs",
+			"entity-type-discovery",
+			"entity-name-discovery",
+			"service-name-discovery",
+			"environment-discovery",
+			"tag-discovery",
+			"metric-discovery",
 			"delta-polling",
 			"trace-query",
 			"trace-query-root-duration",
 			"trace-query-discovery",
 			"thread-query",
-			"trace-query-tenant-scope"
+			"trace-query-tenant-scope",
+			"feedback"
 		];
 	}
 	async createSpan(args) {
@@ -5039,4 +5065,4 @@ var ObservabilityStorageDuckDB = class ObservabilityStorageDuckDB extends Observ
 //#endregion
 export { ObservabilityStorageDuckDB };
 
-//# sourceMappingURL=observability-CDtl20Mc.js.map
+//# sourceMappingURL=observability-BVE7R9Oz.js.map