@mastra/duckdb 1.10.0-alpha.2 → 1.10.0

package/dist/{observability-CCPCADER.cjs → observability-POEhuHwk.cjs}

@@ -3188,7 +3188,7 @@ function compileDuckDBTraceQuery(plan) {
 	}
 	const ctes = compileDuckDBTraceScope(collectRelatedCollections(plan.where));
 	ctes.push(`candidates AS (
-    SELECT ${TRACE_SELECT}
+    SELECT ${TRACE_SELECT}${plan.paginationMode === "delta" ? ", r.cursorId AS deltaWatermark" : ""}
     FROM root_scope r
     WHERE ${conditions.slice(3).join("\n      AND ") || "TRUE"}
   )`);
@@ -3208,6 +3208,26 @@ LIMIT ?`,
 			values
 		};
 	}
+	if (plan.paginationMode === "delta") {
+		const watermark = _mastra_core_storage.getTraceQueryDeltaWatermark(plan, "duckdb");
+		if (watermark !== void 0 && (!/^\d+$/.test(watermark) || BigInt(watermark) > 9223372036854775807n)) throw new _mastra_core_storage.TraceQueryCursorError("TRACE_QUERY_CURSOR_MALFORMED");
+		values.push(watermark ?? "0", plan.limit + 1);
+		return {
+			sql: `${candidates},
+  delta_head AS (SELECT coalesce(max(cursorId), 0) AS streamHead FROM root_events),
+  delta_rows AS (
+    SELECT * FROM candidates
+    WHERE deltaWatermark > CAST(? AS BIGINT) ${watermark === void 0 ? "AND FALSE" : ""}
+    ORDER BY deltaWatermark ASC, traceId ASC
+    LIMIT ?
+  )
+SELECT delta_rows.*, delta_head.streamHead
+FROM delta_head
+LEFT JOIN delta_rows ON TRUE
+ORDER BY delta_rows.deltaWatermark ASC NULLS LAST, delta_rows.traceId ASC`,
+			values
+		};
+	}
 	const orderField = plan.orderBy.field;
 	const direction = plan.orderBy.direction === "asc" ? "ASC" : "DESC";
 	if (plan.paginationMode === "page") {
@@ -3224,7 +3244,7 @@ LIMIT ?`,
     SELECT COUNT(*) AS total
     FROM candidates
   )
-SELECT page_rows.*, page_total.total
+SELECT page_rows.*, page_total.total${deltaPollingFeatureEnabled() ? ", (SELECT coalesce(max(cursorId), 0) FROM root_events) AS streamHead" : ""}
 FROM page_total
 LEFT JOIN page_rows ON TRUE
 ORDER BY page_rows.__row_position ASC NULLS LAST`,
@@ -3393,6 +3413,7 @@ function asIsoTimestamp(value) {
 	return value instanceof Date ? value.toISOString() : new Date(value).toISOString();
 }
 async function queryTraces(db, plan) {
+	if (plan.paginationMode === "delta") assertDeltaPollingEnabled();
 	if (plan.paginationMode === "page") {
 		const query = compileDuckDBTraceQuery(plan);
 		const rows = await db.query(query.sql, query.values);
@@ -3417,6 +3438,7 @@ async function queryTraces(db, plan) {
 		}));
 		return _mastra_core_storage.traceQueryResponseSchema.parse({
 			traces,
+			...deltaPollingFeatureEnabled() && typeof _mastra_core_storage.encodeTraceQueryDeltaCursor === "function" ? { deltaCursor: _mastra_core_storage.encodeTraceQueryDeltaCursor(plan, "duckdb", String(rows[0]?.streamHead ?? 0)) } : {},
 			pagination: {
 				total,
 				page: plan.page,
@@ -3427,7 +3449,8 @@ async function queryTraces(db, plan) {
 	}
 	const query = compileDuckDBTraceQuery(plan);
 	const rows = await db.query(query.sql, query.values);
-	const visibleRows = rows.slice(0, plan.limit);
+	const matchingRows = plan.paginationMode === "delta" ? rows.filter((row) => row.traceId != null) : rows;
+	const visibleRows = matchingRows.slice(0, plan.limit);
 	if (plan.result === "groups") {
 		const groups = visibleRows.map((row) => ({ threadId: String(row.threadId) }));
 		const last = groups.at(-1);
@@ -3457,6 +3480,19 @@ async function queryTraces(db, plan) {
 		environment: row.environment == null ? null : String(row.environment),
 		status: row.status
 	}));
+	if (plan.paginationMode === "delta") {
+		const previous = _mastra_core_storage.getTraceQueryDeltaWatermark(plan, "duckdb") ?? "0";
+		const head = String(rows[0]?.streamHead ?? 0);
+		const watermark = visibleRows.length ? String(visibleRows.at(-1).deltaWatermark) : BigInt(head) > BigInt(previous) ? head : previous;
+		return _mastra_core_storage.traceQueryResponseSchema.parse({
+			traces,
+			delta: {
+				limit: plan.limit,
+				hasMore: matchingRows.length > plan.limit
+			},
+			deltaCursor: _mastra_core_storage.encodeTraceQueryDeltaCursor(plan, "duckdb", watermark)
+		});
+	}
 	const last = traces.at(-1);
 	return _mastra_core_storage.traceQueryResponseSchema.parse({
 		traces,
@@ -4931,4 +4967,4 @@ var ObservabilityStorageDuckDB = class ObservabilityStorageDuckDB extends _mastr
 //#endregion
 exports.ObservabilityStorageDuckDB = ObservabilityStorageDuckDB;
 
-//# sourceMappingURL=observability-CCPCADER.cjs.map
+//# sourceMappingURL=observability-POEhuHwk.cjs.map