@mastra/server 1.58.0-alpha.1 → 1.58.0-alpha.12

package/dist/server/handlers/workflows.js

@@ -1,14 +1,14 @@
 import { n as __exportAll } from "../../rolldown-runtime-DuvcJU0W.js";
 import { t as HTTPException } from "../../http-exception-CyfxZB6W.js";
-import { l as runIdSchema, o as optionalRunIdSchema } from "../../common-CPNOpqxA.js";
-import { _ as workflowRunPathParams, a as observeWorkflowQuerySchema, b as workflowRunsResponseSchema, d as streamWorkflowBodySchema, f as timeTravelBodySchema, g as workflowInfoSchema, h as workflowIdPathParams, i as listWorkflowsResponseSchema, l as startAsyncWorkflowBodySchema, m as workflowExecutionResultSchema, n as createWorkflowRunResponseSchema, o as restartBodySchema, p as workflowControlResponseSchema, r as listWorkflowRunsQuerySchema, s as resumeBodySchema, t as createWorkflowRunBodySchema, v as workflowRunResultQuerySchema, y as workflowRunResultSchema } from "../../workflows-DPW6FrY0.js";
-import { W as streamResponseSchema } from "../../agents-CYvun9iI.js";
-import { n as createRoute } from "../../route-builder-CIk5k35M.js";
-import { a as getWorkflowInfo, t as WorkflowRegistry } from "../../utils-BUK03Frl.js";
+import { l as runIdSchema, o as optionalRunIdSchema } from "../../common-5ZZuR6wj.js";
+import { _ as workflowRunCountsResponseSchema, a as observeWorkflowQuerySchema, b as workflowRunResultSchema, d as streamWorkflowBodySchema, f as timeTravelBodySchema, g as workflowInfoSchema, h as workflowIdPathParams, i as listWorkflowsResponseSchema, l as startAsyncWorkflowBodySchema, m as workflowExecutionResultSchema, n as createWorkflowRunResponseSchema, o as restartBodySchema, p as workflowControlResponseSchema, r as listWorkflowRunsQuerySchema, s as resumeBodySchema, t as createWorkflowRunBodySchema, v as workflowRunPathParams, x as workflowRunsResponseSchema, y as workflowRunResultQuerySchema } from "../../workflows-B2ooCiwN.js";
+import { W as streamResponseSchema } from "../../agents-uT8ZDE4k.js";
+import { n as createRoute } from "../../route-builder-CQkb7tQY.js";
+import { a as getWorkflowInfo, t as WorkflowRegistry } from "../../utils-QXKhjIm6.js";
 import { handleError } from "./error.js";
 import { d as MastraFGAPermissions, l as validateRunOwnership, n as getEffectiveResourceId } from "../../utils-CmHd1Sgh.js";
 import { z } from "zod/v4";
-import { createCachingTransformStream, createReplayStream } from "@mastra/core/stream";
+import { createReplayStream } from "@mastra/core/stream";
 //#region src/server/handlers/workflows.ts
 var workflows_exports = /* @__PURE__ */ __exportAll({
 	CANCEL_WORKFLOW_RUN_ROUTE: () => CANCEL_WORKFLOW_RUN_ROUTE,
@@ -19,6 +19,7 @@ var workflows_exports = /* @__PURE__ */ __exportAll({
 	GET_WORKFLOW_RUN_BY_ID_ROUTE: () => GET_WORKFLOW_RUN_BY_ID_ROUTE,
 	LIST_WORKFLOWS_ROUTE: () => LIST_WORKFLOWS_ROUTE,
 	LIST_WORKFLOW_RUNS_ROUTE: () => LIST_WORKFLOW_RUNS_ROUTE,
+	LIST_WORKFLOW_RUN_COUNTS_ROUTE: () => LIST_WORKFLOW_RUN_COUNTS_ROUTE,
 	OBSERVE_STREAM_LEGACY_WORKFLOW_ROUTE: () => OBSERVE_STREAM_LEGACY_WORKFLOW_ROUTE,
 	OBSERVE_STREAM_WORKFLOW_ROUTE: () => OBSERVE_STREAM_WORKFLOW_ROUTE,
 	RECEIVE_WORKFLOW_EVENT_ROUTE: () => RECEIVE_WORKFLOW_EVENT_ROUTE,
@@ -36,7 +37,8 @@ var workflows_exports = /* @__PURE__ */ __exportAll({
 	STREAM_WORKFLOW_ROUTE: () => STREAM_WORKFLOW_ROUTE,
 	TIME_TRAVEL_ASYNC_WORKFLOW_ROUTE: () => TIME_TRAVEL_ASYNC_WORKFLOW_ROUTE,
 	TIME_TRAVEL_STREAM_WORKFLOW_ROUTE: () => TIME_TRAVEL_STREAM_WORKFLOW_ROUTE,
-	TIME_TRAVEL_WORKFLOW_ROUTE: () => TIME_TRAVEL_WORKFLOW_ROUTE
+	TIME_TRAVEL_WORKFLOW_ROUTE: () => TIME_TRAVEL_WORKFLOW_ROUTE,
+	__setWorkflowRunCountsNow: () => __setWorkflowRunCountsNow
 });
 /**
 * Statuses a run cannot come back from. Streaming one of these runIds again would
@@ -50,6 +52,43 @@ const TERMINAL_RUN_STATUSES = [
 	"canceled",
 	"tripwire"
 ];
+/**
+* Runs whose chunks are already being written to the cache in this process,
+* keyed by runId. A run has exactly one writer: without this, two concurrent
+* requests for the same runId each cache every chunk and the replayed history
+* comes back duplicated.
+*/
+const activeRunStreamCachers = /* @__PURE__ */ new Set();
+/**
+* Cache a run's chunks for later replay and return the stream to hand the client.
+*
+* The cached history belongs to the run, not to whoever happens to be watching,
+* so the caching side is driven by its own reader rather than by the client's
+* consumption. If the client disconnects mid-run its branch is cancelled while
+* this reader keeps draining, so `/observe` still replays a complete history —
+* the failure that makes reconnection necessary is exactly the one the cache has
+* to survive. Chunks buffer in the tee while a client lags, bounded by the run's
+* own length.
+*/
+function cacheRunStream({ cache, runId, source }) {
+	if (activeRunStreamCachers.has(runId)) return source;
+	const [toCache, toClient] = source.tee();
+	activeRunStreamCachers.add(runId);
+	(async () => {
+		const reader = toCache.getReader();
+		try {
+			while (true) {
+				const { done, value } = await reader.read();
+				if (done) break;
+				await cache.listPush(runId, value).catch(() => {});
+			}
+		} catch {} finally {
+			reader.releaseLock();
+			activeRunStreamCachers.delete(runId);
+		}
+	})();
+	return toClient;
+}
 async function listWorkflowsFromSystem({ mastra, workflowId }) {
 	const logger = mastra.getLogger();
 	if (!workflowId) throw new HTTPException(400, { message: "Workflow ID is required" });
@@ -112,6 +151,81 @@ const LIST_WORKFLOWS_ROUTE = createRoute({
 		}
 	})
 });
+const RUN_COUNTS_CACHE_TTL_MS = 5e3;
+const runCountsCache = /* @__PURE__ */ new WeakMap();
+let runCountsNow = () => Date.now();
+/** Test seam — lets TTL expiry be tested without wall-clock waits. */
+function __setWorkflowRunCountsNow(now) {
+	runCountsNow = now ?? (() => Date.now());
+}
+const LIST_WORKFLOW_RUN_COUNTS_ROUTE = createRoute({
+	method: "GET",
+	path: "/workflows/run-counts",
+	responseType: "json",
+	responseSchema: workflowRunCountsResponseSchema,
+	summary: "List workflow run counts",
+	description: "Returns per-workflow counts of currently running and suspended (awaiting resume) runs, keyed by the workflow registry key used in the Mastra config",
+	tags: ["Workflows"],
+	requiresAuth: true,
+	handler: (async ({ mastra, requestContext }) => {
+		try {
+			const fgaProvider = mastra.getServer?.()?.fga;
+			const user = requestContext?.get("user");
+			if (fgaProvider && !user) return {};
+			const workflows = mastra.listWorkflows({ serialized: false });
+			const counts = {};
+			const workflowIdToRegistryKey = /* @__PURE__ */ new Map();
+			for (const [registryKey, workflow] of Object.entries(workflows)) {
+				counts[registryKey] = {
+					running: 0,
+					suspended: 0
+				};
+				workflowIdToRegistryKey.set(workflow.id, registryKey);
+			}
+			const effectiveResourceId = getEffectiveResourceId(requestContext, void 0);
+			const cacheable = !fgaProvider && !effectiveResourceId;
+			if (cacheable) {
+				const cached = runCountsCache.get(mastra);
+				if (cached && runCountsNow() - cached.at < RUN_COUNTS_CACHE_TTL_MS) return cached.value;
+			}
+			const storage = mastra.getStorage();
+			const workflowsStore = storage ? await storage.getStore("workflows") : void 0;
+			if (workflowsStore) {
+				const [running, suspended] = await Promise.all([workflowsStore.listWorkflowRuns({
+					status: "running",
+					resourceId: effectiveResourceId
+				}), workflowsStore.listWorkflowRuns({
+					status: "suspended",
+					resourceId: effectiveResourceId
+				})]);
+				for (const run of running.runs) {
+					const registryKey = workflowIdToRegistryKey.get(run.workflowName);
+					const entry = registryKey ? counts[registryKey] : void 0;
+					if (entry) entry.running++;
+				}
+				for (const run of suspended.runs) {
+					const registryKey = workflowIdToRegistryKey.get(run.workflowName);
+					const entry = registryKey ? counts[registryKey] : void 0;
+					if (entry) entry.suspended++;
+				}
+			}
+			if (fgaProvider) {
+				const workflowList = Object.keys(counts).map((id) => ({ id }));
+				const accessible = await fgaProvider.filterAccessible(user, workflowList, "workflow", MastraFGAPermissions.WORKFLOWS_READ);
+				const accessibleSet = new Set(accessible.map((w) => w.id));
+				for (const id of Object.keys(counts)) if (!accessibleSet.has(id)) delete counts[id];
+				return counts;
+			}
+			if (cacheable) runCountsCache.set(mastra, {
+				at: runCountsNow(),
+				value: counts
+			});
+			return counts;
+		} catch (error) {
+			return handleError(error, "Error getting workflow run counts");
+		}
+	})
+});
 const GET_WORKFLOW_BY_ID_ROUTE = createRoute({
 	method: "GET",
 	path: "/workflows/:workflowId",
@@ -304,13 +418,11 @@ const STREAM_WORKFLOW_ROUTE = createRoute({
 				...params,
 				requestContext
 			});
-			if (serverCache) {
-				const { transform } = createCachingTransformStream({
-					cache: serverCache,
-					cacheKey: runId
-				});
-				return result.fullStream.pipeThrough(transform);
-			}
+			if (serverCache) return cacheRunStream({
+				cache: serverCache,
+				runId,
+				source: result.fullStream
+			});
 			return result.fullStream;
 		} catch (error) {
 			return handleError(error, "Error streaming workflow");
@@ -351,13 +463,11 @@ const RESUME_STREAM_WORKFLOW_ROUTE = createRoute({
 				...params,
 				requestContext
 			});
-			if (serverCache) {
-				const { transform } = createCachingTransformStream({
-					cache: serverCache,
-					cacheKey: runId
-				});
-				return resumeResult.fullStream.pipeThrough(transform);
-			}
+			if (serverCache) return cacheRunStream({
+				cache: serverCache,
+				runId,
+				source: resumeResult.fullStream
+			});
 			return resumeResult.fullStream;
 		} catch (error) {
 			return handleError(error, "Error resuming workflow");
@@ -596,6 +706,12 @@ const RESUME_WORKFLOW_ROUTE = createRoute({
 			})).resume({
 				...params,
 				requestContext
+			}).catch((error) => {
+				mastra.getLogger().error("Failed to resume workflow run", {
+					error,
+					workflowId,
+					runId
+				});
 			});
 			return { message: "Workflow run resumed" };
 		} catch (error) {
@@ -807,6 +923,12 @@ const TIME_TRAVEL_WORKFLOW_ROUTE = createRoute({
 			})).timeTravel({
 				...params,
 				requestContext
+			}).catch((error) => {
+				mastra.getLogger().error("Failed to time travel workflow run", {
+					error,
+					workflowId,
+					runId
+				});
 			});
 			return { message: "Workflow run time travel started" };
 		} catch (error) {
@@ -846,13 +968,11 @@ const TIME_TRAVEL_STREAM_WORKFLOW_ROUTE = createRoute({
 				...params,
 				requestContext
 			});
-			if (serverCache) {
-				const { transform } = createCachingTransformStream({
-					cache: serverCache,
-					cacheKey: runId
-				});
-				return result.fullStream.pipeThrough(transform);
-			}
+			if (serverCache) return cacheRunStream({
+				cache: serverCache,
+				runId,
+				source: result.fullStream
+			});
 			return result.fullStream;
 		} catch (error) {
 			return handleError(error, "Error time traveling workflow stream");
@@ -1087,6 +1207,6 @@ const RECEIVE_WORKFLOW_EVENT_ROUTE = createRoute({
 	})
 });
 //#endregion
-export { CANCEL_WORKFLOW_RUN_ROUTE, CREATE_WORKFLOW_RUN_ROUTE, DELETE_WORKFLOW_RUN_BY_ID_ROUTE, EXECUTE_WORKFLOW_STEP_ROUTE, GET_WORKFLOW_BY_ID_ROUTE, GET_WORKFLOW_RUN_BY_ID_ROUTE, LIST_WORKFLOWS_ROUTE, LIST_WORKFLOW_RUNS_ROUTE, OBSERVE_STREAM_LEGACY_WORKFLOW_ROUTE, OBSERVE_STREAM_WORKFLOW_ROUTE, RECEIVE_WORKFLOW_EVENT_ROUTE, RESTART_ALL_ACTIVE_WORKFLOW_RUNS_ASYNC_ROUTE, RESTART_ALL_ACTIVE_WORKFLOW_RUNS_ROUTE, RESTART_ASYNC_WORKFLOW_ROUTE, RESTART_WORKFLOW_ROUTE, RESUME_ASYNC_WORKFLOW_ROUTE, RESUME_NO_WAIT_WORKFLOW_ROUTE, RESUME_STREAM_WORKFLOW_ROUTE, RESUME_WORKFLOW_ROUTE, START_ASYNC_WORKFLOW_ROUTE, START_WORKFLOW_RUN_ROUTE, STREAM_LEGACY_WORKFLOW_ROUTE, STREAM_WORKFLOW_ROUTE, TIME_TRAVEL_ASYNC_WORKFLOW_ROUTE, TIME_TRAVEL_STREAM_WORKFLOW_ROUTE, TIME_TRAVEL_WORKFLOW_ROUTE, workflows_exports as t };
+export { CANCEL_WORKFLOW_RUN_ROUTE, CREATE_WORKFLOW_RUN_ROUTE, DELETE_WORKFLOW_RUN_BY_ID_ROUTE, EXECUTE_WORKFLOW_STEP_ROUTE, GET_WORKFLOW_BY_ID_ROUTE, GET_WORKFLOW_RUN_BY_ID_ROUTE, LIST_WORKFLOWS_ROUTE, LIST_WORKFLOW_RUNS_ROUTE, LIST_WORKFLOW_RUN_COUNTS_ROUTE, OBSERVE_STREAM_LEGACY_WORKFLOW_ROUTE, OBSERVE_STREAM_WORKFLOW_ROUTE, RECEIVE_WORKFLOW_EVENT_ROUTE, RESTART_ALL_ACTIVE_WORKFLOW_RUNS_ASYNC_ROUTE, RESTART_ALL_ACTIVE_WORKFLOW_RUNS_ROUTE, RESTART_ASYNC_WORKFLOW_ROUTE, RESTART_WORKFLOW_ROUTE, RESUME_ASYNC_WORKFLOW_ROUTE, RESUME_NO_WAIT_WORKFLOW_ROUTE, RESUME_STREAM_WORKFLOW_ROUTE, RESUME_WORKFLOW_ROUTE, START_ASYNC_WORKFLOW_ROUTE, START_WORKFLOW_RUN_ROUTE, STREAM_LEGACY_WORKFLOW_ROUTE, STREAM_WORKFLOW_ROUTE, TIME_TRAVEL_ASYNC_WORKFLOW_ROUTE, TIME_TRAVEL_STREAM_WORKFLOW_ROUTE, TIME_TRAVEL_WORKFLOW_ROUTE, __setWorkflowRunCountsNow, workflows_exports as t };
 
 //# sourceMappingURL=workflows.js.map