inngest 4.2.5 → 4.3.0

package/components/connect/strategies/core/requestProcessor.js

@@ -2,6 +2,7 @@ import { ConnectMessage, GatewayMessageType, WorkerRequestAckData, WorkerRequest
 import { ensureUnsharedArrayBuffer } from "../../buffer.js";
 import { ConnectionState } from "../../types.js";
 import { parseGatewayExecutorRequest, parseWorkerReplyAck } from "../../messages.js";
+import { WAKE_REASON } from "./types.js";
 
 //#region src/components/connect/strategies/core/requestProcessor.ts
 function toError(value) {
@@ -67,6 +68,7 @@ var RequestProcessor = class {
 		})).finish()));
 		this.accessor.inProgressRequests.wg.add(1);
 		this.accessor.inProgressRequests.requestLeases[gatewayExecutorRequest.requestId] = gatewayExecutorRequest.leaseId;
+		const leaseAcquiredAt = Date.now();
 		this.accessor.inProgressRequests.requestMeta[gatewayExecutorRequest.requestId] = {
 			requestId: gatewayExecutorRequest.requestId,
 			runId: gatewayExecutorRequest.runId,
@@ -74,7 +76,9 @@ var RequestProcessor = class {
 			appId: gatewayExecutorRequest.appId,
 			envId: gatewayExecutorRequest.envId,
 			functionSlug: gatewayExecutorRequest.functionSlug,
-			accountId: gatewayExecutorRequest.accountId
+			accountId: gatewayExecutorRequest.accountId,
+			leaseAcquiredAt,
+			leaseLastExtendedAt: leaseAcquiredAt
 		};
 		const inFlightCount = Object.keys(this.accessor.inProgressRequests.requestLeases).length;
 		this.logger.debug({
@@ -98,7 +102,11 @@ var RequestProcessor = class {
 			};
 			this.logger.debug({
 				connectionId: latestConn.id,
-				leaseId: currentLeaseId
+				leaseId: currentLeaseId,
+				requestId: gatewayExecutorRequest.requestId,
+				functionSlug: gatewayExecutorRequest.functionSlug,
+				runId: gatewayExecutorRequest.runId,
+				stepId: gatewayExecutorRequest.stepId
 			}, "Extending lease");
 			if (latestConn.ws.readyState !== WebSocket.OPEN) {
 				this.logger.warn({
@@ -123,6 +131,8 @@ var RequestProcessor = class {
 						leaseId: currentLeaseId
 					})).finish()
 				})).finish()));
+				const meta = this.accessor.inProgressRequests.requestMeta[gatewayExecutorRequest.requestId];
+				if (meta) meta.leaseLastExtendedAt = Date.now();
 			} catch (err) {
 				this.logger.warn({
 					connectionId: latestConn.id,
@@ -169,7 +179,7 @@ var RequestProcessor = class {
 				requestId: gatewayExecutorRequest.requestId,
 				remainingInFlight
 			}, "Request finished");
-			if (this.accessor.shutdownRequested && !this.hasInFlightRequests()) this.wakeSignal.wake();
+			if (this.accessor.shutdownRequested && !this.hasInFlightRequests()) this.wakeSignal.wake(WAKE_REASON.RequestFinishedOnShutdown);
 		}
 	}
 	/** Handle a reply ACK from the gateway. */
@@ -184,13 +194,24 @@ var RequestProcessor = class {
 	/** Handle a lease extension ACK from the gateway. */
 	handleExtendLeaseAck(connectMessage, connectionId) {
 		const extendLeaseAck = WorkerRequestExtendLeaseAckData.decode(connectMessage.payload);
+		const hasLease = Object.hasOwn(this.accessor.inProgressRequests.requestLeases, extendLeaseAck.requestId);
+		const meta = this.accessor.inProgressRequests.requestMeta[extendLeaseAck.requestId];
+		if (!hasLease || !meta) {
+			this.logger.debug({
+				connectionId,
+				requestId: extendLeaseAck.requestId,
+				newLeaseId: extendLeaseAck.newLeaseId,
+				hadLease: hasLease,
+				hadMeta: !!meta
+			}, "Ignoring extend lease ack for non-in-flight request");
+			return;
+		}
 		this.logger.debug({
 			connectionId,
 			newLeaseId: extendLeaseAck.newLeaseId
 		}, "Received extend lease ack");
 		if (extendLeaseAck.newLeaseId) this.accessor.inProgressRequests.requestLeases[extendLeaseAck.requestId] = extendLeaseAck.newLeaseId;
 		else {
-			const meta = this.accessor.inProgressRequests.requestMeta[extendLeaseAck.requestId];
 			this.logger.error({
 				connectionId,
 				requestId: extendLeaseAck.requestId,
@@ -199,6 +220,8 @@ var RequestProcessor = class {
 				stepId: meta?.stepId
 			}, "Lease lost: the server did not renew the lease for this request. Another worker may have claimed it. The in-progress execution will continue but its result may be discarded.");
 			delete this.accessor.inProgressRequests.requestLeases[extendLeaseAck.requestId];
+			delete this.accessor.inProgressRequests.requestMeta[extendLeaseAck.requestId];
+			if (this.accessor.shutdownRequested && !this.hasInFlightRequests()) this.wakeSignal.wake(WAKE_REASON.LeaseLostOnShutdown);
 		}
 	}
 	hasInFlightRequests() {

package/components/connect/strategies/core/requestProcessor.js.map

@@ -1 +1 @@
-{"version":3,"file":"requestProcessor.js","names":["accessor: ConnectionAccessor","wakeSignal: WakeSignal","callbacks: ConnectionCoreCallbacks","logger: Logger","extendLeaseInterval: ReturnType<typeof setInterval> | undefined"],"sources":["../../../../../src/components/connect/strategies/core/requestProcessor.ts"],"sourcesContent":["/**\n * Processes incoming executor requests, manages lease extensions, and handles\n * reply acknowledgements.\n *\n * Extracted from ConnectionCore so the reconcile loop orchestrator only\n * dispatches messages to this module rather than containing the full\n * execution flow inline.\n */\n\nimport type { Logger } from \"../../../../middleware/logger.ts\";\nimport {\n  ConnectMessage,\n  type ConnectMessage as ConnectMessageType,\n  type GatewayExecutorRequestData,\n  GatewayMessageType,\n  WorkerRequestAckData,\n  WorkerRequestExtendLeaseAckData,\n  WorkerRequestExtendLeaseData,\n} from \"../../../../proto/src/components/connect/protobuf/connect.ts\";\nimport { ensureUnsharedArrayBuffer } from \"../../buffer.ts\";\nimport {\n  parseGatewayExecutorRequest,\n  parseWorkerReplyAck,\n} from \"../../messages.ts\";\nimport { ConnectionState } from \"../../types.ts\";\nimport type { Connection, ConnectionCoreCallbacks } from \"./connection.ts\";\nimport type { ConnectionAccessor, WakeSignal } from \"./types.ts\";\n\nfunction toError(value: unknown): Error {\n  if (value instanceof Error) {\n    return value;\n  }\n  return new Error(String(value));\n}\n\nexport class RequestProcessor {\n  constructor(\n    private readonly accessor: ConnectionAccessor,\n    private readonly wakeSignal: WakeSignal,\n    private readonly callbacks: ConnectionCoreCallbacks,\n    private readonly logger: Logger,\n  ) {}\n\n  /** Handle an incoming executor request. */\n  async handleExecutorRequest(\n    connectMessage: ConnectMessageType,\n    conn: Connection,\n  ): Promise<void> {\n    const currentState = this.callbacks.getState();\n    if (currentState !== ConnectionState.ACTIVE) {\n      this.logger.warn(\n        { connectionId: conn.id },\n        \"Received request while not active, skipping\",\n      );\n      return;\n    }\n\n    const gatewayExecutorRequest = parseGatewayExecutorRequest(\n      connectMessage.payload,\n    );\n\n    this.logger.debug(\n      {\n        requestId: gatewayExecutorRequest.requestId,\n        appId: gatewayExecutorRequest.appId,\n        appName: gatewayExecutorRequest.appName,\n        functionSlug: gatewayExecutorRequest.functionSlug,\n        stepId: gatewayExecutorRequest.stepId,\n        connectionId: conn.id,\n      },\n      \"Received gateway executor request\",\n    );\n\n    if (\n      typeof gatewayExecutorRequest.appName !== \"string\" ||\n      gatewayExecutorRequest.appName.length === 0\n    ) {\n      this.logger.warn(\n        {\n          requestId: gatewayExecutorRequest.requestId,\n          appId: gatewayExecutorRequest.appId,\n          functionSlug: gatewayExecutorRequest.functionSlug,\n          stepId: gatewayExecutorRequest.stepId,\n          connectionId: conn.id,\n        },\n        \"No app name in request, skipping\",\n      );\n      return;\n    }\n\n    if (!this.accessor.appIds.includes(gatewayExecutorRequest.appName)) {\n      this.logger.warn(\n        {\n          requestId: gatewayExecutorRequest.requestId,\n          appId: gatewayExecutorRequest.appId,\n          appName: gatewayExecutorRequest.appName,\n          functionSlug: gatewayExecutorRequest.functionSlug,\n          stepId: gatewayExecutorRequest.stepId,\n          connectionId: conn.id,\n        },\n        \"No request handler found for app, skipping\",\n      );\n      return;\n    }\n\n    // Send ACK\n    conn.ws.send(\n      ensureUnsharedArrayBuffer(\n        ConnectMessage.encode(\n          ConnectMessage.create({\n            kind: GatewayMessageType.WORKER_REQUEST_ACK,\n            payload: WorkerRequestAckData.encode(\n              WorkerRequestAckData.create({\n                accountId: gatewayExecutorRequest.accountId,\n                envId: gatewayExecutorRequest.envId,\n                appId: gatewayExecutorRequest.appId,\n                functionSlug: gatewayExecutorRequest.functionSlug,\n                requestId: gatewayExecutorRequest.requestId,\n                stepId: gatewayExecutorRequest.stepId,\n                userTraceCtx: gatewayExecutorRequest.userTraceCtx,\n                systemTraceCtx: gatewayExecutorRequest.systemTraceCtx,\n                runId: gatewayExecutorRequest.runId,\n              }),\n            ).finish(),\n          }),\n        ).finish(),\n      ),\n    );\n\n    this.accessor.inProgressRequests.wg.add(1);\n    this.accessor.inProgressRequests.requestLeases[\n      gatewayExecutorRequest.requestId\n    ] = gatewayExecutorRequest.leaseId;\n    this.accessor.inProgressRequests.requestMeta[\n      gatewayExecutorRequest.requestId\n    ] = {\n      requestId: gatewayExecutorRequest.requestId,\n      runId: gatewayExecutorRequest.runId,\n      stepId: gatewayExecutorRequest.stepId,\n      appId: gatewayExecutorRequest.appId,\n      envId: gatewayExecutorRequest.envId,\n      functionSlug: gatewayExecutorRequest.functionSlug,\n      accountId: gatewayExecutorRequest.accountId,\n    };\n\n    const inFlightCount = Object.keys(\n      this.accessor.inProgressRequests.requestLeases,\n    ).length;\n    this.logger.debug(\n      {\n        requestId: gatewayExecutorRequest.requestId,\n        functionSlug: gatewayExecutorRequest.functionSlug,\n        inFlightCount,\n      },\n      \"Request acknowledged\",\n    );\n\n    const startedAt = Date.now();\n\n    // Start lease extension interval\n    const originalWs = conn.ws;\n    const originalConnectionId = conn.id;\n    let extendLeaseInterval: ReturnType<typeof setInterval> | undefined;\n    extendLeaseInterval = setInterval(() => {\n      const currentLeaseId =\n        this.accessor.inProgressRequests.requestLeases[\n          gatewayExecutorRequest.requestId\n        ];\n      if (!currentLeaseId) {\n        clearInterval(extendLeaseInterval);\n        return;\n      }\n\n      // Use the current live connection's WebSocket for lease extensions.\n      // During a drain, the original WebSocket may be closed by the gateway\n      // while the request is still in flight.\n      const latestConn = {\n        ws: this.accessor.activeConnection?.ws ?? originalWs,\n        id: this.accessor.activeConnection?.id ?? originalConnectionId,\n      };\n\n      this.logger.debug(\n        { connectionId: latestConn.id, leaseId: currentLeaseId },\n        \"Extending lease\",\n      );\n\n      if (latestConn.ws.readyState !== WebSocket.OPEN) {\n        this.logger.warn(\n          {\n            connectionId: latestConn.id,\n            requestId: gatewayExecutorRequest.requestId,\n          },\n          \"Cannot extend lease, no open WebSocket available\",\n        );\n        return;\n      }\n\n      try {\n        latestConn.ws.send(\n          ensureUnsharedArrayBuffer(\n            ConnectMessage.encode(\n              ConnectMessage.create({\n                kind: GatewayMessageType.WORKER_REQUEST_EXTEND_LEASE,\n                payload: WorkerRequestExtendLeaseData.encode(\n                  WorkerRequestExtendLeaseData.create({\n                    accountId: gatewayExecutorRequest.accountId,\n                    envId: gatewayExecutorRequest.envId,\n                    appId: gatewayExecutorRequest.appId,\n                    functionSlug: gatewayExecutorRequest.functionSlug,\n                    requestId: gatewayExecutorRequest.requestId,\n                    stepId: gatewayExecutorRequest.stepId,\n                    runId: gatewayExecutorRequest.runId,\n                    userTraceCtx: gatewayExecutorRequest.userTraceCtx,\n                    systemTraceCtx: gatewayExecutorRequest.systemTraceCtx,\n                    leaseId: currentLeaseId,\n                  }),\n                ).finish(),\n              }),\n            ).finish(),\n          ),\n        );\n      } catch (err) {\n        this.logger.warn(\n          {\n            connectionId: latestConn.id,\n            requestId: gatewayExecutorRequest.requestId,\n            err: toError(err),\n          },\n          \"Failed to send lease extension\",\n        );\n      }\n    }, conn.extendLeaseIntervalMs);\n\n    try {\n      const responseBytes = await this.callbacks.handleExecutionRequest(\n        gatewayExecutorRequest,\n      );\n\n      const durationMs = Date.now() - startedAt;\n      this.logger.debug(\n        {\n          requestId: gatewayExecutorRequest.requestId,\n          functionSlug: gatewayExecutorRequest.functionSlug,\n          durationMs,\n        },\n        \"Request execution completed\",\n      );\n\n      if (!this.accessor.activeConnection) {\n        this.logger.warn(\n          { requestId: gatewayExecutorRequest.requestId },\n          \"No current WebSocket, buffering response\",\n        );\n        if (this.callbacks.onBufferResponse) {\n          this.callbacks.onBufferResponse(\n            gatewayExecutorRequest.requestId,\n            responseBytes,\n          );\n        }\n        return;\n      }\n\n      this.logger.debug(\n        {\n          connectionId: this.accessor.activeConnection.id,\n          requestId: gatewayExecutorRequest.requestId,\n        },\n        \"Sending worker reply\",\n      );\n\n      this.accessor.activeConnection.ws.send(\n        ensureUnsharedArrayBuffer(\n          ConnectMessage.encode(\n            ConnectMessage.create({\n              kind: GatewayMessageType.WORKER_REPLY,\n              payload: responseBytes,\n            }),\n          ).finish(),\n        ),\n      );\n    } catch (err) {\n      const durationMs = Date.now() - startedAt;\n      this.logger.warn(\n        {\n          requestId: gatewayExecutorRequest.requestId,\n          durationMs,\n          err: toError(err),\n        },\n        \"Execution error\",\n      );\n    } finally {\n      this.accessor.inProgressRequests.wg.done();\n      delete this.accessor.inProgressRequests.requestLeases[\n        gatewayExecutorRequest.requestId\n      ];\n      delete this.accessor.inProgressRequests.requestMeta[\n        gatewayExecutorRequest.requestId\n      ];\n      clearInterval(extendLeaseInterval);\n\n      const remainingInFlight = Object.keys(\n        this.accessor.inProgressRequests.requestLeases,\n      ).length;\n      this.logger.debug(\n        {\n          requestId: gatewayExecutorRequest.requestId,\n          remainingInFlight,\n        },\n        \"Request finished\",\n      );\n\n      // Wake the loop if shutdown is pending and this was the last request\n      if (this.accessor.shutdownRequested && !this.hasInFlightRequests()) {\n        this.wakeSignal.wake();\n      }\n    }\n  }\n\n  /** Handle a reply ACK from the gateway. */\n  handleReplyAck(\n    connectMessage: ConnectMessageType,\n    connectionId: string,\n  ): void {\n    const replyAck = parseWorkerReplyAck(connectMessage.payload);\n\n    this.logger.debug(\n      { connectionId, requestId: replyAck.requestId },\n      \"Acknowledging reply ack\",\n    );\n\n    this.callbacks.onReplyAck?.(replyAck.requestId);\n  }\n\n  /** Handle a lease extension ACK from the gateway. */\n  handleExtendLeaseAck(\n    connectMessage: ConnectMessageType,\n    connectionId: string,\n  ): void {\n    const extendLeaseAck = WorkerRequestExtendLeaseAckData.decode(\n      connectMessage.payload,\n    );\n\n    this.logger.debug(\n      { connectionId, newLeaseId: extendLeaseAck.newLeaseId },\n      \"Received extend lease ack\",\n    );\n\n    if (extendLeaseAck.newLeaseId) {\n      this.accessor.inProgressRequests.requestLeases[extendLeaseAck.requestId] =\n        extendLeaseAck.newLeaseId;\n    } else {\n      const meta =\n        this.accessor.inProgressRequests.requestMeta[extendLeaseAck.requestId];\n\n      this.logger.error(\n        {\n          connectionId,\n          requestId: extendLeaseAck.requestId,\n          functionSlug: meta?.functionSlug,\n          runId: meta?.runId,\n          stepId: meta?.stepId,\n        },\n        \"Lease lost: the server did not renew the lease for this request. \" +\n          \"Another worker may have claimed it. The in-progress execution \" +\n          \"will continue but its result may be discarded.\",\n      );\n      delete this.accessor.inProgressRequests.requestLeases[\n        extendLeaseAck.requestId\n      ];\n    }\n  }\n\n  private hasInFlightRequests(): boolean {\n    return (\n      Object.keys(this.accessor.inProgressRequests.requestLeases).length > 0\n    );\n  }\n}\n"],"mappings":";;;;;;AA4BA,SAAS,QAAQ,OAAuB;AACtC,KAAI,iBAAiB,MACnB,QAAO;AAET,QAAO,IAAI,MAAM,OAAO,MAAM,CAAC;;AAGjC,IAAa,mBAAb,MAA8B;CAC5B,YACE,AAAiBA,UACjB,AAAiBC,YACjB,AAAiBC,WACjB,AAAiBC,QACjB;EAJiB;EACA;EACA;EACA;;;CAInB,MAAM,sBACJ,gBACA,MACe;AAEf,MADqB,KAAK,UAAU,UAAU,KACzB,gBAAgB,QAAQ;AAC3C,QAAK,OAAO,KACV,EAAE,cAAc,KAAK,IAAI,EACzB,8CACD;AACD;;EAGF,MAAM,yBAAyB,4BAC7B,eAAe,QAChB;AAED,OAAK,OAAO,MACV;GACE,WAAW,uBAAuB;GAClC,OAAO,uBAAuB;GAC9B,SAAS,uBAAuB;GAChC,cAAc,uBAAuB;GACrC,QAAQ,uBAAuB;GAC/B,cAAc,KAAK;GACpB,EACD,oCACD;AAED,MACE,OAAO,uBAAuB,YAAY,YAC1C,uBAAuB,QAAQ,WAAW,GAC1C;AACA,QAAK,OAAO,KACV;IACE,WAAW,uBAAuB;IAClC,OAAO,uBAAuB;IAC9B,cAAc,uBAAuB;IACrC,QAAQ,uBAAuB;IAC/B,cAAc,KAAK;IACpB,EACD,mCACD;AACD;;AAGF,MAAI,CAAC,KAAK,SAAS,OAAO,SAAS,uBAAuB,QAAQ,EAAE;AAClE,QAAK,OAAO,KACV;IACE,WAAW,uBAAuB;IAClC,OAAO,uBAAuB;IAC9B,SAAS,uBAAuB;IAChC,cAAc,uBAAuB;IACrC,QAAQ,uBAAuB;IAC/B,cAAc,KAAK;IACpB,EACD,6CACD;AACD;;AAIF,OAAK,GAAG,KACN,0BACE,eAAe,OACb,eAAe,OAAO;GACpB,MAAM,mBAAmB;GACzB,SAAS,qBAAqB,OAC5B,qBAAqB,OAAO;IAC1B,WAAW,uBAAuB;IAClC,OAAO,uBAAuB;IAC9B,OAAO,uBAAuB;IAC9B,cAAc,uBAAuB;IACrC,WAAW,uBAAuB;IAClC,QAAQ,uBAAuB;IAC/B,cAAc,uBAAuB;IACrC,gBAAgB,uBAAuB;IACvC,OAAO,uBAAuB;IAC/B,CAAC,CACH,CAAC,QAAQ;GACX,CAAC,CACH,CAAC,QAAQ,CACX,CACF;AAED,OAAK,SAAS,mBAAmB,GAAG,IAAI,EAAE;AAC1C,OAAK,SAAS,mBAAmB,cAC/B,uBAAuB,aACrB,uBAAuB;AAC3B,OAAK,SAAS,mBAAmB,YAC/B,uBAAuB,aACrB;GACF,WAAW,uBAAuB;GAClC,OAAO,uBAAuB;GAC9B,QAAQ,uBAAuB;GAC/B,OAAO,uBAAuB;GAC9B,OAAO,uBAAuB;GAC9B,cAAc,uBAAuB;GACrC,WAAW,uBAAuB;GACnC;EAED,MAAM,gBAAgB,OAAO,KAC3B,KAAK,SAAS,mBAAmB,cAClC,CAAC;AACF,OAAK,OAAO,MACV;GACE,WAAW,uBAAuB;GAClC,cAAc,uBAAuB;GACrC;GACD,EACD,uBACD;EAED,MAAM,YAAY,KAAK,KAAK;EAG5B,MAAM,aAAa,KAAK;EACxB,MAAM,uBAAuB,KAAK;EAClC,IAAIC;AACJ,wBAAsB,kBAAkB;GACtC,MAAM,iBACJ,KAAK,SAAS,mBAAmB,cAC/B,uBAAuB;AAE3B,OAAI,CAAC,gBAAgB;AACnB,kBAAc,oBAAoB;AAClC;;GAMF,MAAM,aAAa;IACjB,IAAI,KAAK,SAAS,kBAAkB,MAAM;IAC1C,IAAI,KAAK,SAAS,kBAAkB,MAAM;IAC3C;AAED,QAAK,OAAO,MACV;IAAE,cAAc,WAAW;IAAI,SAAS;IAAgB,EACxD,kBACD;AAED,OAAI,WAAW,GAAG,eAAe,UAAU,MAAM;AAC/C,SAAK,OAAO,KACV;KACE,cAAc,WAAW;KACzB,WAAW,uBAAuB;KACnC,EACD,mDACD;AACD;;AAGF,OAAI;AACF,eAAW,GAAG,KACZ,0BACE,eAAe,OACb,eAAe,OAAO;KACpB,MAAM,mBAAmB;KACzB,SAAS,6BAA6B,OACpC,6BAA6B,OAAO;MAClC,WAAW,uBAAuB;MAClC,OAAO,uBAAuB;MAC9B,OAAO,uBAAuB;MAC9B,cAAc,uBAAuB;MACrC,WAAW,uBAAuB;MAClC,QAAQ,uBAAuB;MAC/B,OAAO,uBAAuB;MAC9B,cAAc,uBAAuB;MACrC,gBAAgB,uBAAuB;MACvC,SAAS;MACV,CAAC,CACH,CAAC,QAAQ;KACX,CAAC,CACH,CAAC,QAAQ,CACX,CACF;YACM,KAAK;AACZ,SAAK,OAAO,KACV;KACE,cAAc,WAAW;KACzB,WAAW,uBAAuB;KAClC,KAAK,QAAQ,IAAI;KAClB,EACD,iCACD;;KAEF,KAAK,sBAAsB;AAE9B,MAAI;GACF,MAAM,gBAAgB,MAAM,KAAK,UAAU,uBACzC,uBACD;GAED,MAAM,aAAa,KAAK,KAAK,GAAG;AAChC,QAAK,OAAO,MACV;IACE,WAAW,uBAAuB;IAClC,cAAc,uBAAuB;IACrC;IACD,EACD,8BACD;AAED,OAAI,CAAC,KAAK,SAAS,kBAAkB;AACnC,SAAK,OAAO,KACV,EAAE,WAAW,uBAAuB,WAAW,EAC/C,2CACD;AACD,QAAI,KAAK,UAAU,iBACjB,MAAK,UAAU,iBACb,uBAAuB,WACvB,cACD;AAEH;;AAGF,QAAK,OAAO,MACV;IACE,cAAc,KAAK,SAAS,iBAAiB;IAC7C,WAAW,uBAAuB;IACnC,EACD,uBACD;AAED,QAAK,SAAS,iBAAiB,GAAG,KAChC,0BACE,eAAe,OACb,eAAe,OAAO;IACpB,MAAM,mBAAmB;IACzB,SAAS;IACV,CAAC,CACH,CAAC,QAAQ,CACX,CACF;WACM,KAAK;GACZ,MAAM,aAAa,KAAK,KAAK,GAAG;AAChC,QAAK,OAAO,KACV;IACE,WAAW,uBAAuB;IAClC;IACA,KAAK,QAAQ,IAAI;IAClB,EACD,kBACD;YACO;AACR,QAAK,SAAS,mBAAmB,GAAG,MAAM;AAC1C,UAAO,KAAK,SAAS,mBAAmB,cACtC,uBAAuB;AAEzB,UAAO,KAAK,SAAS,mBAAmB,YACtC,uBAAuB;AAEzB,iBAAc,oBAAoB;GAElC,MAAM,oBAAoB,OAAO,KAC/B,KAAK,SAAS,mBAAmB,cAClC,CAAC;AACF,QAAK,OAAO,MACV;IACE,WAAW,uBAAuB;IAClC;IACD,EACD,mBACD;AAGD,OAAI,KAAK,SAAS,qBAAqB,CAAC,KAAK,qBAAqB,CAChE,MAAK,WAAW,MAAM;;;;CAM5B,eACE,gBACA,cACM;EACN,MAAM,WAAW,oBAAoB,eAAe,QAAQ;AAE5D,OAAK,OAAO,MACV;GAAE;GAAc,WAAW,SAAS;GAAW,EAC/C,0BACD;AAED,OAAK,UAAU,aAAa,SAAS,UAAU;;;CAIjD,qBACE,gBACA,cACM;EACN,MAAM,iBAAiB,gCAAgC,OACrD,eAAe,QAChB;AAED,OAAK,OAAO,MACV;GAAE;GAAc,YAAY,eAAe;GAAY,EACvD,4BACD;AAED,MAAI,eAAe,WACjB,MAAK,SAAS,mBAAmB,cAAc,eAAe,aAC5D,eAAe;OACZ;GACL,MAAM,OACJ,KAAK,SAAS,mBAAmB,YAAY,eAAe;AAE9D,QAAK,OAAO,MACV;IACE;IACA,WAAW,eAAe;IAC1B,cAAc,MAAM;IACpB,OAAO,MAAM;IACb,QAAQ,MAAM;IACf,EACD,gLAGD;AACD,UAAO,KAAK,SAAS,mBAAmB,cACtC,eAAe;;;CAKrB,AAAQ,sBAA+B;AACrC,SACE,OAAO,KAAK,KAAK,SAAS,mBAAmB,cAAc,CAAC,SAAS"}
+{"version":3,"file":"requestProcessor.js","names":["accessor: ConnectionAccessor","wakeSignal: WakeSignal","callbacks: ConnectionCoreCallbacks","logger: Logger","extendLeaseInterval: ReturnType<typeof setInterval> | undefined"],"sources":["../../../../../src/components/connect/strategies/core/requestProcessor.ts"],"sourcesContent":["/**\n * Processes incoming executor requests, manages lease extensions, and handles\n * reply acknowledgements.\n *\n * Extracted from ConnectionCore so the reconcile loop orchestrator only\n * dispatches messages to this module rather than containing the full\n * execution flow inline.\n */\n\nimport type { Logger } from \"../../../../middleware/logger.ts\";\nimport {\n  ConnectMessage,\n  type ConnectMessage as ConnectMessageType,\n  GatewayMessageType,\n  WorkerRequestAckData,\n  WorkerRequestExtendLeaseAckData,\n  WorkerRequestExtendLeaseData,\n} from \"../../../../proto/src/components/connect/protobuf/connect.ts\";\nimport { ensureUnsharedArrayBuffer } from \"../../buffer.ts\";\nimport {\n  parseGatewayExecutorRequest,\n  parseWorkerReplyAck,\n} from \"../../messages.ts\";\nimport { ConnectionState } from \"../../types.ts\";\nimport type { Connection, ConnectionCoreCallbacks } from \"./connection.ts\";\nimport {\n  type ConnectionAccessor,\n  WAKE_REASON,\n  type WakeSignal,\n} from \"./types.ts\";\n\nfunction toError(value: unknown): Error {\n  if (value instanceof Error) {\n    return value;\n  }\n  return new Error(String(value));\n}\n\nexport class RequestProcessor {\n  constructor(\n    private readonly accessor: ConnectionAccessor,\n    private readonly wakeSignal: WakeSignal,\n    private readonly callbacks: ConnectionCoreCallbacks,\n    private readonly logger: Logger,\n  ) {}\n\n  /** Handle an incoming executor request. */\n  async handleExecutorRequest(\n    connectMessage: ConnectMessageType,\n    conn: Connection,\n  ): Promise<void> {\n    const currentState = this.callbacks.getState();\n    if (currentState !== ConnectionState.ACTIVE) {\n      this.logger.warn(\n        { connectionId: conn.id },\n        \"Received request while not active, skipping\",\n      );\n      return;\n    }\n\n    const gatewayExecutorRequest = parseGatewayExecutorRequest(\n      connectMessage.payload,\n    );\n\n    this.logger.debug(\n      {\n        requestId: gatewayExecutorRequest.requestId,\n        appId: gatewayExecutorRequest.appId,\n        appName: gatewayExecutorRequest.appName,\n        functionSlug: gatewayExecutorRequest.functionSlug,\n        stepId: gatewayExecutorRequest.stepId,\n        connectionId: conn.id,\n      },\n      \"Received gateway executor request\",\n    );\n\n    if (\n      typeof gatewayExecutorRequest.appName !== \"string\" ||\n      gatewayExecutorRequest.appName.length === 0\n    ) {\n      this.logger.warn(\n        {\n          requestId: gatewayExecutorRequest.requestId,\n          appId: gatewayExecutorRequest.appId,\n          functionSlug: gatewayExecutorRequest.functionSlug,\n          stepId: gatewayExecutorRequest.stepId,\n          connectionId: conn.id,\n        },\n        \"No app name in request, skipping\",\n      );\n      return;\n    }\n\n    if (!this.accessor.appIds.includes(gatewayExecutorRequest.appName)) {\n      this.logger.warn(\n        {\n          requestId: gatewayExecutorRequest.requestId,\n          appId: gatewayExecutorRequest.appId,\n          appName: gatewayExecutorRequest.appName,\n          functionSlug: gatewayExecutorRequest.functionSlug,\n          stepId: gatewayExecutorRequest.stepId,\n          connectionId: conn.id,\n        },\n        \"No request handler found for app, skipping\",\n      );\n      return;\n    }\n\n    // Send ACK\n    conn.ws.send(\n      ensureUnsharedArrayBuffer(\n        ConnectMessage.encode(\n          ConnectMessage.create({\n            kind: GatewayMessageType.WORKER_REQUEST_ACK,\n            payload: WorkerRequestAckData.encode(\n              WorkerRequestAckData.create({\n                accountId: gatewayExecutorRequest.accountId,\n                envId: gatewayExecutorRequest.envId,\n                appId: gatewayExecutorRequest.appId,\n                functionSlug: gatewayExecutorRequest.functionSlug,\n                requestId: gatewayExecutorRequest.requestId,\n                stepId: gatewayExecutorRequest.stepId,\n                userTraceCtx: gatewayExecutorRequest.userTraceCtx,\n                systemTraceCtx: gatewayExecutorRequest.systemTraceCtx,\n                runId: gatewayExecutorRequest.runId,\n              }),\n            ).finish(),\n          }),\n        ).finish(),\n      ),\n    );\n\n    this.accessor.inProgressRequests.wg.add(1);\n    this.accessor.inProgressRequests.requestLeases[\n      gatewayExecutorRequest.requestId\n    ] = gatewayExecutorRequest.leaseId;\n    const leaseAcquiredAt = Date.now();\n    this.accessor.inProgressRequests.requestMeta[\n      gatewayExecutorRequest.requestId\n    ] = {\n      requestId: gatewayExecutorRequest.requestId,\n      runId: gatewayExecutorRequest.runId,\n      stepId: gatewayExecutorRequest.stepId,\n      appId: gatewayExecutorRequest.appId,\n      envId: gatewayExecutorRequest.envId,\n      functionSlug: gatewayExecutorRequest.functionSlug,\n      accountId: gatewayExecutorRequest.accountId,\n      leaseAcquiredAt,\n      leaseLastExtendedAt: leaseAcquiredAt,\n    };\n\n    const inFlightCount = Object.keys(\n      this.accessor.inProgressRequests.requestLeases,\n    ).length;\n    this.logger.debug(\n      {\n        requestId: gatewayExecutorRequest.requestId,\n        functionSlug: gatewayExecutorRequest.functionSlug,\n        inFlightCount,\n      },\n      \"Request acknowledged\",\n    );\n\n    const startedAt = Date.now();\n\n    // Start lease extension interval\n    const originalWs = conn.ws;\n    const originalConnectionId = conn.id;\n    let extendLeaseInterval: ReturnType<typeof setInterval> | undefined;\n    extendLeaseInterval = setInterval(() => {\n      const currentLeaseId =\n        this.accessor.inProgressRequests.requestLeases[\n          gatewayExecutorRequest.requestId\n        ];\n      if (!currentLeaseId) {\n        clearInterval(extendLeaseInterval);\n        return;\n      }\n\n      // Use the current live connection's WebSocket for lease extensions.\n      // During a drain, the original WebSocket may be closed by the gateway\n      // while the request is still in flight.\n      const latestConn = {\n        ws: this.accessor.activeConnection?.ws ?? originalWs,\n        id: this.accessor.activeConnection?.id ?? originalConnectionId,\n      };\n\n      this.logger.debug(\n        {\n          connectionId: latestConn.id,\n          leaseId: currentLeaseId,\n          requestId: gatewayExecutorRequest.requestId,\n          functionSlug: gatewayExecutorRequest.functionSlug,\n          runId: gatewayExecutorRequest.runId,\n          stepId: gatewayExecutorRequest.stepId,\n        },\n        \"Extending lease\",\n      );\n\n      if (latestConn.ws.readyState !== WebSocket.OPEN) {\n        this.logger.warn(\n          {\n            connectionId: latestConn.id,\n            requestId: gatewayExecutorRequest.requestId,\n          },\n          \"Cannot extend lease, no open WebSocket available\",\n        );\n        return;\n      }\n\n      try {\n        latestConn.ws.send(\n          ensureUnsharedArrayBuffer(\n            ConnectMessage.encode(\n              ConnectMessage.create({\n                kind: GatewayMessageType.WORKER_REQUEST_EXTEND_LEASE,\n                payload: WorkerRequestExtendLeaseData.encode(\n                  WorkerRequestExtendLeaseData.create({\n                    accountId: gatewayExecutorRequest.accountId,\n                    envId: gatewayExecutorRequest.envId,\n                    appId: gatewayExecutorRequest.appId,\n                    functionSlug: gatewayExecutorRequest.functionSlug,\n                    requestId: gatewayExecutorRequest.requestId,\n                    stepId: gatewayExecutorRequest.stepId,\n                    runId: gatewayExecutorRequest.runId,\n                    userTraceCtx: gatewayExecutorRequest.userTraceCtx,\n                    systemTraceCtx: gatewayExecutorRequest.systemTraceCtx,\n                    leaseId: currentLeaseId,\n                  }),\n                ).finish(),\n              }),\n            ).finish(),\n          ),\n        );\n        const meta =\n          this.accessor.inProgressRequests.requestMeta[\n            gatewayExecutorRequest.requestId\n          ];\n        if (meta) meta.leaseLastExtendedAt = Date.now();\n      } catch (err) {\n        this.logger.warn(\n          {\n            connectionId: latestConn.id,\n            requestId: gatewayExecutorRequest.requestId,\n            err: toError(err),\n          },\n          \"Failed to send lease extension\",\n        );\n      }\n    }, conn.extendLeaseIntervalMs);\n\n    try {\n      const responseBytes = await this.callbacks.handleExecutionRequest(\n        gatewayExecutorRequest,\n      );\n\n      const durationMs = Date.now() - startedAt;\n      this.logger.debug(\n        {\n          requestId: gatewayExecutorRequest.requestId,\n          functionSlug: gatewayExecutorRequest.functionSlug,\n          durationMs,\n        },\n        \"Request execution completed\",\n      );\n\n      if (!this.accessor.activeConnection) {\n        this.logger.warn(\n          { requestId: gatewayExecutorRequest.requestId },\n          \"No current WebSocket, buffering response\",\n        );\n        if (this.callbacks.onBufferResponse) {\n          this.callbacks.onBufferResponse(\n            gatewayExecutorRequest.requestId,\n            responseBytes,\n          );\n        }\n        return;\n      }\n\n      this.logger.debug(\n        {\n          connectionId: this.accessor.activeConnection.id,\n          requestId: gatewayExecutorRequest.requestId,\n        },\n        \"Sending worker reply\",\n      );\n\n      this.accessor.activeConnection.ws.send(\n        ensureUnsharedArrayBuffer(\n          ConnectMessage.encode(\n            ConnectMessage.create({\n              kind: GatewayMessageType.WORKER_REPLY,\n              payload: responseBytes,\n            }),\n          ).finish(),\n        ),\n      );\n    } catch (err) {\n      const durationMs = Date.now() - startedAt;\n      this.logger.warn(\n        {\n          requestId: gatewayExecutorRequest.requestId,\n          durationMs,\n          err: toError(err),\n        },\n        \"Execution error\",\n      );\n    } finally {\n      this.accessor.inProgressRequests.wg.done();\n      delete this.accessor.inProgressRequests.requestLeases[\n        gatewayExecutorRequest.requestId\n      ];\n      delete this.accessor.inProgressRequests.requestMeta[\n        gatewayExecutorRequest.requestId\n      ];\n      clearInterval(extendLeaseInterval);\n\n      const remainingInFlight = Object.keys(\n        this.accessor.inProgressRequests.requestLeases,\n      ).length;\n      this.logger.debug(\n        {\n          requestId: gatewayExecutorRequest.requestId,\n          remainingInFlight,\n        },\n        \"Request finished\",\n      );\n\n      // Wake the loop if shutdown is pending and this was the last request\n      if (this.accessor.shutdownRequested && !this.hasInFlightRequests()) {\n        this.wakeSignal.wake(WAKE_REASON.RequestFinishedOnShutdown);\n      }\n    }\n  }\n\n  /** Handle a reply ACK from the gateway. */\n  handleReplyAck(\n    connectMessage: ConnectMessageType,\n    connectionId: string,\n  ): void {\n    const replyAck = parseWorkerReplyAck(connectMessage.payload);\n\n    this.logger.debug(\n      { connectionId, requestId: replyAck.requestId },\n      \"Acknowledging reply ack\",\n    );\n\n    this.callbacks.onReplyAck?.(replyAck.requestId);\n  }\n\n  /** Handle a lease extension ACK from the gateway. */\n  handleExtendLeaseAck(\n    connectMessage: ConnectMessageType,\n    connectionId: string,\n  ): void {\n    const extendLeaseAck = WorkerRequestExtendLeaseAckData.decode(\n      connectMessage.payload,\n    );\n\n    // Late lease-extend ACKs can arrive after a request has already cleaned up\n    // both maps in the completion or lease-lost path. This handler stays\n    // synchronous, so once we observe the request missing from either map we\n    // can safely ignore the ACK instead of recreating a stale lease entry.\n    const hasLease = Object.hasOwn(\n      this.accessor.inProgressRequests.requestLeases,\n      extendLeaseAck.requestId,\n    );\n    const meta =\n      this.accessor.inProgressRequests.requestMeta[extendLeaseAck.requestId];\n    if (!hasLease || !meta) {\n      this.logger.debug(\n        {\n          connectionId,\n          requestId: extendLeaseAck.requestId,\n          newLeaseId: extendLeaseAck.newLeaseId,\n          hadLease: hasLease,\n          hadMeta: !!meta,\n        },\n        \"Ignoring extend lease ack for non-in-flight request\",\n      );\n      return;\n    }\n\n    this.logger.debug(\n      { connectionId, newLeaseId: extendLeaseAck.newLeaseId },\n      \"Received extend lease ack\",\n    );\n\n    if (extendLeaseAck.newLeaseId) {\n      this.accessor.inProgressRequests.requestLeases[extendLeaseAck.requestId] =\n        extendLeaseAck.newLeaseId;\n    } else {\n      this.logger.error(\n        {\n          connectionId,\n          requestId: extendLeaseAck.requestId,\n          functionSlug: meta?.functionSlug,\n          runId: meta?.runId,\n          stepId: meta?.stepId,\n        },\n        \"Lease lost: the server did not renew the lease for this request. \" +\n          \"Another worker may have claimed it. The in-progress execution \" +\n          \"will continue but its result may be discarded.\",\n      );\n      delete this.accessor.inProgressRequests.requestLeases[\n        extendLeaseAck.requestId\n      ];\n      // Also drop meta so the shutdown dump helper and debug-state snapshot\n      // don't report a request we've explicitly released the lease for.\n      delete this.accessor.inProgressRequests.requestMeta[\n        extendLeaseAck.requestId\n      ];\n\n      // If this was the last in-flight request and a shutdown has been\n      // requested, wake the reconcile loop so close() can observe the\n      // empty lease map and exit. Without this, the loop stays parked on\n      // wakeSignal.promise because the finally-block decrement in\n      // handleExecutorRequest never runs (user code is still hanging).\n      if (this.accessor.shutdownRequested && !this.hasInFlightRequests()) {\n        this.wakeSignal.wake(WAKE_REASON.LeaseLostOnShutdown);\n      }\n    }\n  }\n\n  private hasInFlightRequests(): boolean {\n    return (\n      Object.keys(this.accessor.inProgressRequests.requestLeases).length > 0\n    );\n  }\n}\n"],"mappings":";;;;;;;AA+BA,SAAS,QAAQ,OAAuB;AACtC,KAAI,iBAAiB,MACnB,QAAO;AAET,QAAO,IAAI,MAAM,OAAO,MAAM,CAAC;;AAGjC,IAAa,mBAAb,MAA8B;CAC5B,YACE,AAAiBA,UACjB,AAAiBC,YACjB,AAAiBC,WACjB,AAAiBC,QACjB;EAJiB;EACA;EACA;EACA;;;CAInB,MAAM,sBACJ,gBACA,MACe;AAEf,MADqB,KAAK,UAAU,UAAU,KACzB,gBAAgB,QAAQ;AAC3C,QAAK,OAAO,KACV,EAAE,cAAc,KAAK,IAAI,EACzB,8CACD;AACD;;EAGF,MAAM,yBAAyB,4BAC7B,eAAe,QAChB;AAED,OAAK,OAAO,MACV;GACE,WAAW,uBAAuB;GAClC,OAAO,uBAAuB;GAC9B,SAAS,uBAAuB;GAChC,cAAc,uBAAuB;GACrC,QAAQ,uBAAuB;GAC/B,cAAc,KAAK;GACpB,EACD,oCACD;AAED,MACE,OAAO,uBAAuB,YAAY,YAC1C,uBAAuB,QAAQ,WAAW,GAC1C;AACA,QAAK,OAAO,KACV;IACE,WAAW,uBAAuB;IAClC,OAAO,uBAAuB;IAC9B,cAAc,uBAAuB;IACrC,QAAQ,uBAAuB;IAC/B,cAAc,KAAK;IACpB,EACD,mCACD;AACD;;AAGF,MAAI,CAAC,KAAK,SAAS,OAAO,SAAS,uBAAuB,QAAQ,EAAE;AAClE,QAAK,OAAO,KACV;IACE,WAAW,uBAAuB;IAClC,OAAO,uBAAuB;IAC9B,SAAS,uBAAuB;IAChC,cAAc,uBAAuB;IACrC,QAAQ,uBAAuB;IAC/B,cAAc,KAAK;IACpB,EACD,6CACD;AACD;;AAIF,OAAK,GAAG,KACN,0BACE,eAAe,OACb,eAAe,OAAO;GACpB,MAAM,mBAAmB;GACzB,SAAS,qBAAqB,OAC5B,qBAAqB,OAAO;IAC1B,WAAW,uBAAuB;IAClC,OAAO,uBAAuB;IAC9B,OAAO,uBAAuB;IAC9B,cAAc,uBAAuB;IACrC,WAAW,uBAAuB;IAClC,QAAQ,uBAAuB;IAC/B,cAAc,uBAAuB;IACrC,gBAAgB,uBAAuB;IACvC,OAAO,uBAAuB;IAC/B,CAAC,CACH,CAAC,QAAQ;GACX,CAAC,CACH,CAAC,QAAQ,CACX,CACF;AAED,OAAK,SAAS,mBAAmB,GAAG,IAAI,EAAE;AAC1C,OAAK,SAAS,mBAAmB,cAC/B,uBAAuB,aACrB,uBAAuB;EAC3B,MAAM,kBAAkB,KAAK,KAAK;AAClC,OAAK,SAAS,mBAAmB,YAC/B,uBAAuB,aACrB;GACF,WAAW,uBAAuB;GAClC,OAAO,uBAAuB;GAC9B,QAAQ,uBAAuB;GAC/B,OAAO,uBAAuB;GAC9B,OAAO,uBAAuB;GAC9B,cAAc,uBAAuB;GACrC,WAAW,uBAAuB;GAClC;GACA,qBAAqB;GACtB;EAED,MAAM,gBAAgB,OAAO,KAC3B,KAAK,SAAS,mBAAmB,cAClC,CAAC;AACF,OAAK,OAAO,MACV;GACE,WAAW,uBAAuB;GAClC,cAAc,uBAAuB;GACrC;GACD,EACD,uBACD;EAED,MAAM,YAAY,KAAK,KAAK;EAG5B,MAAM,aAAa,KAAK;EACxB,MAAM,uBAAuB,KAAK;EAClC,IAAIC;AACJ,wBAAsB,kBAAkB;GACtC,MAAM,iBACJ,KAAK,SAAS,mBAAmB,cAC/B,uBAAuB;AAE3B,OAAI,CAAC,gBAAgB;AACnB,kBAAc,oBAAoB;AAClC;;GAMF,MAAM,aAAa;IACjB,IAAI,KAAK,SAAS,kBAAkB,MAAM;IAC1C,IAAI,KAAK,SAAS,kBAAkB,MAAM;IAC3C;AAED,QAAK,OAAO,MACV;IACE,cAAc,WAAW;IACzB,SAAS;IACT,WAAW,uBAAuB;IAClC,cAAc,uBAAuB;IACrC,OAAO,uBAAuB;IAC9B,QAAQ,uBAAuB;IAChC,EACD,kBACD;AAED,OAAI,WAAW,GAAG,eAAe,UAAU,MAAM;AAC/C,SAAK,OAAO,KACV;KACE,cAAc,WAAW;KACzB,WAAW,uBAAuB;KACnC,EACD,mDACD;AACD;;AAGF,OAAI;AACF,eAAW,GAAG,KACZ,0BACE,eAAe,OACb,eAAe,OAAO;KACpB,MAAM,mBAAmB;KACzB,SAAS,6BAA6B,OACpC,6BAA6B,OAAO;MAClC,WAAW,uBAAuB;MAClC,OAAO,uBAAuB;MAC9B,OAAO,uBAAuB;MAC9B,cAAc,uBAAuB;MACrC,WAAW,uBAAuB;MAClC,QAAQ,uBAAuB;MAC/B,OAAO,uBAAuB;MAC9B,cAAc,uBAAuB;MACrC,gBAAgB,uBAAuB;MACvC,SAAS;MACV,CAAC,CACH,CAAC,QAAQ;KACX,CAAC,CACH,CAAC,QAAQ,CACX,CACF;IACD,MAAM,OACJ,KAAK,SAAS,mBAAmB,YAC/B,uBAAuB;AAE3B,QAAI,KAAM,MAAK,sBAAsB,KAAK,KAAK;YACxC,KAAK;AACZ,SAAK,OAAO,KACV;KACE,cAAc,WAAW;KACzB,WAAW,uBAAuB;KAClC,KAAK,QAAQ,IAAI;KAClB,EACD,iCACD;;KAEF,KAAK,sBAAsB;AAE9B,MAAI;GACF,MAAM,gBAAgB,MAAM,KAAK,UAAU,uBACzC,uBACD;GAED,MAAM,aAAa,KAAK,KAAK,GAAG;AAChC,QAAK,OAAO,MACV;IACE,WAAW,uBAAuB;IAClC,cAAc,uBAAuB;IACrC;IACD,EACD,8BACD;AAED,OAAI,CAAC,KAAK,SAAS,kBAAkB;AACnC,SAAK,OAAO,KACV,EAAE,WAAW,uBAAuB,WAAW,EAC/C,2CACD;AACD,QAAI,KAAK,UAAU,iBACjB,MAAK,UAAU,iBACb,uBAAuB,WACvB,cACD;AAEH;;AAGF,QAAK,OAAO,MACV;IACE,cAAc,KAAK,SAAS,iBAAiB;IAC7C,WAAW,uBAAuB;IACnC,EACD,uBACD;AAED,QAAK,SAAS,iBAAiB,GAAG,KAChC,0BACE,eAAe,OACb,eAAe,OAAO;IACpB,MAAM,mBAAmB;IACzB,SAAS;IACV,CAAC,CACH,CAAC,QAAQ,CACX,CACF;WACM,KAAK;GACZ,MAAM,aAAa,KAAK,KAAK,GAAG;AAChC,QAAK,OAAO,KACV;IACE,WAAW,uBAAuB;IAClC;IACA,KAAK,QAAQ,IAAI;IAClB,EACD,kBACD;YACO;AACR,QAAK,SAAS,mBAAmB,GAAG,MAAM;AAC1C,UAAO,KAAK,SAAS,mBAAmB,cACtC,uBAAuB;AAEzB,UAAO,KAAK,SAAS,mBAAmB,YACtC,uBAAuB;AAEzB,iBAAc,oBAAoB;GAElC,MAAM,oBAAoB,OAAO,KAC/B,KAAK,SAAS,mBAAmB,cAClC,CAAC;AACF,QAAK,OAAO,MACV;IACE,WAAW,uBAAuB;IAClC;IACD,EACD,mBACD;AAGD,OAAI,KAAK,SAAS,qBAAqB,CAAC,KAAK,qBAAqB,CAChE,MAAK,WAAW,KAAK,YAAY,0BAA0B;;;;CAMjE,eACE,gBACA,cACM;EACN,MAAM,WAAW,oBAAoB,eAAe,QAAQ;AAE5D,OAAK,OAAO,MACV;GAAE;GAAc,WAAW,SAAS;GAAW,EAC/C,0BACD;AAED,OAAK,UAAU,aAAa,SAAS,UAAU;;;CAIjD,qBACE,gBACA,cACM;EACN,MAAM,iBAAiB,gCAAgC,OACrD,eAAe,QAChB;EAMD,MAAM,WAAW,OAAO,OACtB,KAAK,SAAS,mBAAmB,eACjC,eAAe,UAChB;EACD,MAAM,OACJ,KAAK,SAAS,mBAAmB,YAAY,eAAe;AAC9D,MAAI,CAAC,YAAY,CAAC,MAAM;AACtB,QAAK,OAAO,MACV;IACE;IACA,WAAW,eAAe;IAC1B,YAAY,eAAe;IAC3B,UAAU;IACV,SAAS,CAAC,CAAC;IACZ,EACD,sDACD;AACD;;AAGF,OAAK,OAAO,MACV;GAAE;GAAc,YAAY,eAAe;GAAY,EACvD,4BACD;AAED,MAAI,eAAe,WACjB,MAAK,SAAS,mBAAmB,cAAc,eAAe,aAC5D,eAAe;OACZ;AACL,QAAK,OAAO,MACV;IACE;IACA,WAAW,eAAe;IAC1B,cAAc,MAAM;IACpB,OAAO,MAAM;IACb,QAAQ,MAAM;IACf,EACD,gLAGD;AACD,UAAO,KAAK,SAAS,mBAAmB,cACtC,eAAe;AAIjB,UAAO,KAAK,SAAS,mBAAmB,YACtC,eAAe;AAQjB,OAAI,KAAK,SAAS,qBAAqB,CAAC,KAAK,qBAAqB,CAChE,MAAK,WAAW,KAAK,YAAY,oBAAoB;;;CAK3D,AAAQ,sBAA+B;AACrC,SACE,OAAO,KAAK,KAAK,SAAS,mBAAmB,cAAc,CAAC,SAAS"}

package/components/connect/strategies/core/types.cjs

@@ -0,0 +1,21 @@
+
+//#region src/components/connect/strategies/core/types.ts
+/**
+* Reasons the reconcile loop may be woken. Centralised so call sites are
+* grep-able and the "Reconcile loop woken" log carries a stable vocabulary.
+*/
+const WAKE_REASON = {
+	ShutdownRequested: "shutdown-requested",
+	WsError: "ws-error",
+	WsClose: "ws-close",
+	GatewayClosing: "gateway-closing",
+	HeartbeatMissed: "heartbeat-missed",
+	RequestFinishedOnShutdown: "request-finished-on-shutdown",
+	LeaseLostOnShutdown: "lease-lost-on-shutdown",
+	ShutdownStillPending: "shutdown-still-pending",
+	Unknown: "unknown"
+};
+
+//#endregion
+exports.WAKE_REASON = WAKE_REASON;
+//# sourceMappingURL=types.cjs.map

package/components/connect/strategies/core/types.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.cjs","names":[],"sources":["../../../../../src/components/connect/strategies/core/types.ts"],"sourcesContent":["import type { WaitGroup } from \"@jpwilliams/waitgroup\";\nimport type { Mode } from \"../../../../helpers/env.ts\";\nimport type { Logger } from \"../../../../middleware/logger.ts\";\nimport type {\n  GatewayExecutorRequestData,\n  SDKResponse,\n} from \"../../../../proto/src/components/connect/protobuf/connect.ts\";\nimport type { Inngest } from \"../../../Inngest.ts\";\nimport type {\n  ConnectDebugState,\n  ConnectHandlerOptions,\n  ConnectionState,\n  InFlightRequest,\n} from \"../../types.ts\";\nimport type { Connection } from \"./connection.ts\";\n\n/**\n * A request handler that processes executor requests and returns SDK responses.\n */\nexport type RequestHandler = (\n  msg: GatewayExecutorRequestData,\n) => Promise<SDKResponse>;\n\n/**\n * Data needed to establish a connection with the gateway.\n */\nexport interface ConnectionEstablishData {\n  marshaledCapabilities: string;\n  manualReadinessAck: boolean;\n  apps: {\n    appName: string;\n    appVersion?: string;\n    functions: Uint8Array;\n  }[];\n}\n\n/**\n * Base configuration shared across all connection config types.\n * Contains the core fields needed for authentication and connection establishment.\n */\nexport interface BaseConnectionConfig {\n  /**\n   * The hashed signing key for authentication.\n   */\n  hashedSigningKey: string | undefined;\n\n  /**\n   * The hashed fallback signing key for authentication.\n   */\n  hashedFallbackKey: string | undefined;\n\n  /**\n   * The Inngest environment name.\n   */\n  envName: string | undefined;\n\n  /**\n   * Data for establishing the connection.\n   */\n  connectionData: ConnectionEstablishData;\n\n  /**\n   * The base URL for the Inngest API, as defined when constructing the Inngest\n   * client (field or env var).\n   */\n  apiBaseUrl: string | undefined;\n\n  /**\n   * The mode of the Inngest client.\n   */\n  mode: Mode;\n}\n\n/**\n * Configuration required by connection strategies.\n * Extends BaseConnectionConfig with strategy-specific fields.\n */\nexport interface StrategyConfig extends BaseConnectionConfig {\n  /**\n   * The logger to use for internal logging.\n   */\n  internalLogger: Logger;\n\n  /**\n   * Request handlers mapped by app ID.\n   */\n  requestHandlers: Record<string, RequestHandler>;\n\n  /**\n   * Connection options from the user.\n   */\n  options: ConnectHandlerOptions;\n}\n\n/**\n * Events emitted by connection strategies.\n */\nexport interface StrategyEvents {\n  /**\n   * Called when the connection state changes.\n   */\n  onStateChange: (state: ConnectionState) => void;\n\n  /**\n   * Called when a connection error occurs.\n   */\n  onError: (error: Error) => void;\n}\n\n/**\n * Interface for connection strategies.\n *\n * A connection strategy manages the WebSocket connection to the Inngest gateway,\n * including heartbeats, lease extension, and request/response handling.\n */\nexport interface ConnectionStrategy {\n  /**\n   * The current state of the connection.\n   */\n  readonly state: ConnectionState;\n\n  /**\n   * The current connection ID, if connected.\n   */\n  readonly connectionId: string | undefined;\n\n  /**\n   * Establish a connection to the gateway.\n   *\n   * @param attempt - The current connection attempt number for exponential backoff.\n   */\n  connect(attempt?: number): Promise<void>;\n\n  /**\n   * Close the connection gracefully.\n   */\n  close(): Promise<void>;\n\n  /**\n   * A promise that resolves when the connection is fully closed.\n   */\n  readonly closed: Promise<void>;\n\n  /**\n   * Return a snapshot of debug/health information for this connection.\n   */\n  getDebugState(): ConnectDebugState;\n}\n\n/**\n * Narrow view of ConnectionCore state, passed to sub-modules so they can\n * read (but not mutate) the connection state they need.\n */\nexport interface ConnectionAccessor {\n  readonly activeConnection: Connection | undefined;\n  readonly drainingConnection: Connection | undefined;\n  readonly shutdownRequested: boolean;\n  readonly inProgressRequests: {\n    wg: WaitGroup;\n    requestLeases: Record<string, string>;\n    requestMeta: Record<string, InFlightRequest>;\n  };\n  readonly appIds: string[];\n}\n\n/**\n * Reasons the reconcile loop may be woken. Centralised so call sites are\n * grep-able and the \"Reconcile loop woken\" log carries a stable vocabulary.\n */\nexport const WAKE_REASON = {\n  ShutdownRequested: \"shutdown-requested\",\n  WsError: \"ws-error\",\n  WsClose: \"ws-close\",\n  GatewayClosing: \"gateway-closing\",\n  HeartbeatMissed: \"heartbeat-missed\",\n  RequestFinishedOnShutdown: \"request-finished-on-shutdown\",\n  LeaseLostOnShutdown: \"lease-lost-on-shutdown\",\n  ShutdownStillPending: \"shutdown-still-pending\",\n  Unknown: \"unknown\",\n} as const;\nexport type WakeReason = (typeof WAKE_REASON)[keyof typeof WAKE_REASON];\n\n/**\n * Wake signal interface for sub-modules to wake the reconcile loop.\n */\nexport interface WakeSignal {\n  wake(reason?: WakeReason): void;\n}\n"],"mappings":";;;;;;AAyKA,MAAa,cAAc;CACzB,mBAAmB;CACnB,SAAS;CACT,SAAS;CACT,gBAAgB;CAChB,iBAAiB;CACjB,2BAA2B;CAC3B,qBAAqB;CACrB,sBAAsB;CACtB,SAAS;CACV"}

package/components/connect/strategies/core/types.js

@@ -0,0 +1,20 @@
+//#region src/components/connect/strategies/core/types.ts
+/**
+* Reasons the reconcile loop may be woken. Centralised so call sites are
+* grep-able and the "Reconcile loop woken" log carries a stable vocabulary.
+*/
+const WAKE_REASON = {
+	ShutdownRequested: "shutdown-requested",
+	WsError: "ws-error",
+	WsClose: "ws-close",
+	GatewayClosing: "gateway-closing",
+	HeartbeatMissed: "heartbeat-missed",
+	RequestFinishedOnShutdown: "request-finished-on-shutdown",
+	LeaseLostOnShutdown: "lease-lost-on-shutdown",
+	ShutdownStillPending: "shutdown-still-pending",
+	Unknown: "unknown"
+};
+
+//#endregion
+export { WAKE_REASON };
+//# sourceMappingURL=types.js.map

package/components/connect/strategies/core/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","names":[],"sources":["../../../../../src/components/connect/strategies/core/types.ts"],"sourcesContent":["import type { WaitGroup } from \"@jpwilliams/waitgroup\";\nimport type { Mode } from \"../../../../helpers/env.ts\";\nimport type { Logger } from \"../../../../middleware/logger.ts\";\nimport type {\n  GatewayExecutorRequestData,\n  SDKResponse,\n} from \"../../../../proto/src/components/connect/protobuf/connect.ts\";\nimport type { Inngest } from \"../../../Inngest.ts\";\nimport type {\n  ConnectDebugState,\n  ConnectHandlerOptions,\n  ConnectionState,\n  InFlightRequest,\n} from \"../../types.ts\";\nimport type { Connection } from \"./connection.ts\";\n\n/**\n * A request handler that processes executor requests and returns SDK responses.\n */\nexport type RequestHandler = (\n  msg: GatewayExecutorRequestData,\n) => Promise<SDKResponse>;\n\n/**\n * Data needed to establish a connection with the gateway.\n */\nexport interface ConnectionEstablishData {\n  marshaledCapabilities: string;\n  manualReadinessAck: boolean;\n  apps: {\n    appName: string;\n    appVersion?: string;\n    functions: Uint8Array;\n  }[];\n}\n\n/**\n * Base configuration shared across all connection config types.\n * Contains the core fields needed for authentication and connection establishment.\n */\nexport interface BaseConnectionConfig {\n  /**\n   * The hashed signing key for authentication.\n   */\n  hashedSigningKey: string | undefined;\n\n  /**\n   * The hashed fallback signing key for authentication.\n   */\n  hashedFallbackKey: string | undefined;\n\n  /**\n   * The Inngest environment name.\n   */\n  envName: string | undefined;\n\n  /**\n   * Data for establishing the connection.\n   */\n  connectionData: ConnectionEstablishData;\n\n  /**\n   * The base URL for the Inngest API, as defined when constructing the Inngest\n   * client (field or env var).\n   */\n  apiBaseUrl: string | undefined;\n\n  /**\n   * The mode of the Inngest client.\n   */\n  mode: Mode;\n}\n\n/**\n * Configuration required by connection strategies.\n * Extends BaseConnectionConfig with strategy-specific fields.\n */\nexport interface StrategyConfig extends BaseConnectionConfig {\n  /**\n   * The logger to use for internal logging.\n   */\n  internalLogger: Logger;\n\n  /**\n   * Request handlers mapped by app ID.\n   */\n  requestHandlers: Record<string, RequestHandler>;\n\n  /**\n   * Connection options from the user.\n   */\n  options: ConnectHandlerOptions;\n}\n\n/**\n * Events emitted by connection strategies.\n */\nexport interface StrategyEvents {\n  /**\n   * Called when the connection state changes.\n   */\n  onStateChange: (state: ConnectionState) => void;\n\n  /**\n   * Called when a connection error occurs.\n   */\n  onError: (error: Error) => void;\n}\n\n/**\n * Interface for connection strategies.\n *\n * A connection strategy manages the WebSocket connection to the Inngest gateway,\n * including heartbeats, lease extension, and request/response handling.\n */\nexport interface ConnectionStrategy {\n  /**\n   * The current state of the connection.\n   */\n  readonly state: ConnectionState;\n\n  /**\n   * The current connection ID, if connected.\n   */\n  readonly connectionId: string | undefined;\n\n  /**\n   * Establish a connection to the gateway.\n   *\n   * @param attempt - The current connection attempt number for exponential backoff.\n   */\n  connect(attempt?: number): Promise<void>;\n\n  /**\n   * Close the connection gracefully.\n   */\n  close(): Promise<void>;\n\n  /**\n   * A promise that resolves when the connection is fully closed.\n   */\n  readonly closed: Promise<void>;\n\n  /**\n   * Return a snapshot of debug/health information for this connection.\n   */\n  getDebugState(): ConnectDebugState;\n}\n\n/**\n * Narrow view of ConnectionCore state, passed to sub-modules so they can\n * read (but not mutate) the connection state they need.\n */\nexport interface ConnectionAccessor {\n  readonly activeConnection: Connection | undefined;\n  readonly drainingConnection: Connection | undefined;\n  readonly shutdownRequested: boolean;\n  readonly inProgressRequests: {\n    wg: WaitGroup;\n    requestLeases: Record<string, string>;\n    requestMeta: Record<string, InFlightRequest>;\n  };\n  readonly appIds: string[];\n}\n\n/**\n * Reasons the reconcile loop may be woken. Centralised so call sites are\n * grep-able and the \"Reconcile loop woken\" log carries a stable vocabulary.\n */\nexport const WAKE_REASON = {\n  ShutdownRequested: \"shutdown-requested\",\n  WsError: \"ws-error\",\n  WsClose: \"ws-close\",\n  GatewayClosing: \"gateway-closing\",\n  HeartbeatMissed: \"heartbeat-missed\",\n  RequestFinishedOnShutdown: \"request-finished-on-shutdown\",\n  LeaseLostOnShutdown: \"lease-lost-on-shutdown\",\n  ShutdownStillPending: \"shutdown-still-pending\",\n  Unknown: \"unknown\",\n} as const;\nexport type WakeReason = (typeof WAKE_REASON)[keyof typeof WAKE_REASON];\n\n/**\n * Wake signal interface for sub-modules to wake the reconcile loop.\n */\nexport interface WakeSignal {\n  wake(reason?: WakeReason): void;\n}\n"],"mappings":";;;;;AAyKA,MAAa,cAAc;CACzB,mBAAmB;CACnB,SAAS;CACT,SAAS;CACT,gBAAgB;CAChB,iBAAiB;CACjB,2BAA2B;CAC3B,qBAAqB;CACrB,sBAAsB;CACtB,SAAS;CACV"}

package/components/connect/types.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"types.cjs","names":[],"sources":["../../../src/components/connect/types.ts"],"sourcesContent":["import type { RegisterOptions } from \"../../types.ts\";\nimport type { Inngest } from \"../Inngest.ts\";\nimport type { InngestFunction } from \"../InngestFunction.ts\";\n\nexport const DEFAULT_SHUTDOWN_SIGNALS = [\"SIGINT\", \"SIGTERM\"];\n\nexport interface ConnectApp {\n  client: Inngest.Any;\n  functions?: Array<InngestFunction.Like>;\n}\n\nexport interface ConnectHandlerOptions extends RegisterOptions {\n  apps: ConnectApp[];\n\n  /**\n   * InstanceId represents a stable identifier to be used for identifying connected SDKs.\n   * This can be a hostname or other identifier that remains stable across restarts.\n   *\n   * If nil, this defaults to the current machine's hostname.\n   */\n  instanceId?: string;\n\n  /**\n   * MaxWorkerConcurrency represents the maximum number of worker concurrency to use.\n   *\n   * If left undefined, there will be no limit on the number of concurrent requests on the worker.\n   */\n  maxWorkerConcurrency?: number;\n\n  /**\n   * By default, connections will be gracefully shut down when the current\n   * process receives a SIGINT or SIGTERM signal. Set this to an empty array to disable this behavior.\n   */\n  handleShutdownSignals?: string[];\n\n  /**\n   * Override the gateway WebSocket endpoint. When set, this URL is used\n   * instead of the endpoint returned by the Inngest API.\n   *\n   * Useful when there is a proxy between the worker and the gateway\n   * that requires a different hostname (e.g. `ws://localhost:8100`).\n   *\n   * Can also be set via the `INNGEST_CONNECT_GATEWAY_URL` environment variable.\n   * This option takes precedence over the env var.\n   */\n  gatewayUrl?: string;\n\n  /**\n   * Enable running the WebSocket connection, heartbeater, and lease extender\n   * in a separate worker thread. This prevents thread-blocking user code from\n   * interfering with connection health.\n   *\n   * Only works in environments that support worker_threads.\n   *\n   * Can also be disabled via the INNGEST_CONNECT_ISOLATE_EXECUTION=false environment variable.\n   *\n   * @default true\n   */\n  isolateExecution?: boolean;\n}\n\nexport interface WorkerConnection {\n  connectionId: string;\n  closed: Promise<void>;\n  close: () => Promise<void>;\n  state: ConnectionState;\n  getDebugState: () => ConnectDebugState;\n}\n\nexport interface InFlightRequest {\n  requestId: string;\n  runId: string;\n  stepId: string | undefined;\n  appId: string;\n  envId: string;\n  functionSlug: string;\n  accountId: string;\n}\n\nexport interface ConnectDebugState {\n  /** Current connection state */\n  state: ConnectionState;\n  /** Active connection ID, if any */\n  activeConnectionId: string | undefined;\n  /** Draining connection ID (during gateway drain), if any */\n  drainingConnectionId: string | undefined;\n  /** Timestamp of last heartbeat we sent */\n  lastHeartbeatSentAt: number | undefined;\n  /** Timestamp of last heartbeat response from gateway */\n  lastHeartbeatReceivedAt: number | undefined;\n  /** Timestamp of last WS message received (any type) */\n  lastMessageReceivedAt: number | undefined;\n  /** Whether shutdown has been requested (WORKER_PAUSE sent) */\n  shutdownRequested: boolean;\n  /** Number of in-flight requests */\n  inFlightRequestCount: number;\n  /** Detailed info about in-flight requests */\n  inFlightRequests: InFlightRequest[];\n}\n\nexport enum ConnectionState {\n  CONNECTING = \"CONNECTING\",\n  ACTIVE = \"ACTIVE\",\n  PAUSED = \"PAUSED\",\n  RECONNECTING = \"RECONNECTING\",\n  CLOSING = \"CLOSING\",\n  CLOSED = \"CLOSED\",\n}\n"],"mappings":";;AAIA,MAAa,2BAA2B,CAAC,UAAU,UAAU;AAgG7D,IAAY,8DAAL;AACL;AACA;AACA;AACA;AACA;AACA"}
+{"version":3,"file":"types.cjs","names":[],"sources":["../../../src/components/connect/types.ts"],"sourcesContent":["import type { RegisterOptions } from \"../../types.ts\";\nimport type { Inngest } from \"../Inngest.ts\";\nimport type { InngestFunction } from \"../InngestFunction.ts\";\n\nexport const DEFAULT_SHUTDOWN_SIGNALS = [\"SIGINT\", \"SIGTERM\"];\n\nexport interface ConnectApp {\n  client: Inngest.Any;\n  functions?: Array<InngestFunction.Like>;\n}\n\nexport interface ConnectHandlerOptions extends RegisterOptions {\n  apps: ConnectApp[];\n\n  /**\n   * InstanceId represents a stable identifier to be used for identifying connected SDKs.\n   * This can be a hostname or other identifier that remains stable across restarts.\n   *\n   * If nil, this defaults to the current machine's hostname.\n   */\n  instanceId?: string;\n\n  /**\n   * MaxWorkerConcurrency represents the maximum number of worker concurrency to use.\n   *\n   * If left undefined, there will be no limit on the number of concurrent requests on the worker.\n   */\n  maxWorkerConcurrency?: number;\n\n  /**\n   * By default, connections will be gracefully shut down when the current\n   * process receives a SIGINT or SIGTERM signal. Set this to an empty array to disable this behavior.\n   */\n  handleShutdownSignals?: string[];\n\n  /**\n   * Override the gateway WebSocket endpoint. When set, this URL is used\n   * instead of the endpoint returned by the Inngest API.\n   *\n   * Useful when there is a proxy between the worker and the gateway\n   * that requires a different hostname (e.g. `ws://localhost:8100`).\n   *\n   * Can also be set via the `INNGEST_CONNECT_GATEWAY_URL` environment variable.\n   * This option takes precedence over the env var.\n   */\n  gatewayUrl?: string;\n\n  /**\n   * Enable running the WebSocket connection, heartbeater, and lease extender\n   * in a separate worker thread. This prevents thread-blocking user code from\n   * interfering with connection health.\n   *\n   * Only works in environments that support worker_threads.\n   *\n   * Can also be disabled via the INNGEST_CONNECT_ISOLATE_EXECUTION=false environment variable.\n   *\n   * @default true\n   */\n  isolateExecution?: boolean;\n}\n\nexport interface WorkerConnection {\n  connectionId: string;\n  closed: Promise<void>;\n  close: () => Promise<void>;\n  state: ConnectionState;\n  getDebugState: () => ConnectDebugState;\n}\n\nexport interface InFlightRequest {\n  requestId: string;\n  runId: string;\n  stepId: string | undefined;\n  appId: string;\n  envId: string;\n  functionSlug: string;\n  accountId: string;\n  /**\n   * Timestamp (ms since epoch) when the worker acknowledged the request and\n   * took the lease. Used to compute age for shutdown diagnostics.\n   *\n   * Optional so external callers constructing an `InFlightRequest` (e.g.\n   * from {@link ConnectDebugState}) aren't forced to provide it.\n   */\n  leaseAcquiredAt?: number;\n  /**\n   * Timestamp (ms since epoch) of the most recent successful lease-extend send\n   * for this request. Starts equal to `leaseAcquiredAt` and is updated each\n   * time the worker sends a WORKER_REQUEST_EXTEND_LEASE message. Useful\n   * during shutdown diagnostics to tell whether a stuck request is still\n   * being actively leased or has gone silent.\n   *\n   * Optional for the same reason as {@link leaseAcquiredAt}.\n   */\n  leaseLastExtendedAt?: number;\n}\n\nexport interface ConnectDebugState {\n  /** Current connection state */\n  state: ConnectionState;\n  /** Active connection ID, if any */\n  activeConnectionId: string | undefined;\n  /** Draining connection ID (during gateway drain), if any */\n  drainingConnectionId: string | undefined;\n  /** Timestamp of last heartbeat we sent */\n  lastHeartbeatSentAt: number | undefined;\n  /** Timestamp of last heartbeat response from gateway */\n  lastHeartbeatReceivedAt: number | undefined;\n  /** Timestamp of last WS message received (any type) */\n  lastMessageReceivedAt: number | undefined;\n  /** Whether shutdown has been requested (WORKER_PAUSE sent) */\n  shutdownRequested: boolean;\n  /** Number of in-flight requests */\n  inFlightRequestCount: number;\n  /** Detailed info about in-flight requests */\n  inFlightRequests: InFlightRequest[];\n}\n\nexport enum ConnectionState {\n  CONNECTING = \"CONNECTING\",\n  ACTIVE = \"ACTIVE\",\n  PAUSED = \"PAUSED\",\n  RECONNECTING = \"RECONNECTING\",\n  CLOSING = \"CLOSING\",\n  CLOSED = \"CLOSED\",\n}\n"],"mappings":";;AAIA,MAAa,2BAA2B,CAAC,UAAU,UAAU;AAkH7D,IAAY,8DAAL;AACL;AACA;AACA;AACA;AACA;AACA"}

package/components/connect/types.d.cts

@@ -67,6 +67,24 @@ interface InFlightRequest {
   envId: string;
   functionSlug: string;
   accountId: string;
+  /**
+   * Timestamp (ms since epoch) when the worker acknowledged the request and
+   * took the lease. Used to compute age for shutdown diagnostics.
+   *
+   * Optional so external callers constructing an `InFlightRequest` (e.g.
+   * from {@link ConnectDebugState}) aren't forced to provide it.
+   */
+  leaseAcquiredAt?: number;
+  /**
+   * Timestamp (ms since epoch) of the most recent successful lease-extend send
+   * for this request. Starts equal to `leaseAcquiredAt` and is updated each
+   * time the worker sends a WORKER_REQUEST_EXTEND_LEASE message. Useful
+   * during shutdown diagnostics to tell whether a stuck request is still
+   * being actively leased or has gone silent.
+   *
+   * Optional for the same reason as {@link leaseAcquiredAt}.
+   */
+  leaseLastExtendedAt?: number;
 }
 interface ConnectDebugState {
   /** Current connection state */

package/components/connect/types.d.cts.map

@@ -1 +1 @@
-{"version":3,"file":"types.d.cts","names":[],"sources":["../../../src/components/connect/types.ts"],"sourcesContent":[],"mappings":";;;;;cAIa;UAEI,UAAA;EAFJ,MAAA,EAGH,OAAA,CAAQ,GAHL;EAEI,SAAA,CAAA,EAEH,KAFa,CAEP,eAAA,CAAgB,IAFT,CAAA;;AACjB,UAIO,qBAAA,SAA8B,eAJ7B,CAAA;MACE,EAIZ,UAJY,EAAA;;;AAGpB;;;;EAA8D,UAAA,CAAA,EAAA,MAAA;EAkD7C;;;;;sBAKM,CAAA,EAAA,MAAA;EAAiB;AAGxC;AAUA;;uBAES,CAAA,EAAA,MAAA,EAAA;;;AAmBT;;;;;;;;;;;;;;;;;;;;;;UAvCiB,gBAAA;;UAEP;eACK;SACN;uBACc;;UAGN,eAAA;;;;;;;;;UAUA,iBAAA;;SAER;;;;;;;;;;;;;;;;oBAgBW;;aAGR,eAAA"}
+{"version":3,"file":"types.d.cts","names":[],"sources":["../../../src/components/connect/types.ts"],"sourcesContent":[],"mappings":";;;;;cAIa;UAEI,UAAA;EAFJ,MAAA,EAGH,OAAA,CAAQ,GAHL;EAEI,SAAA,CAAA,EAEH,KAFa,CAEP,eAAA,CAAgB,IAFT,CAAA;;AACjB,UAIO,qBAAA,SAA8B,eAJ7B,CAAA;MACE,EAIZ,UAJY,EAAA;;;AAGpB;;;;EAA8D,UAAA,CAAA,EAAA,MAAA;EAkD7C;;;;;sBAKM,CAAA,EAAA,MAAA;EAAiB;AAGxC;AA4BA;;uBAES,CAAA,EAAA,MAAA,EAAA;;;AAmBT;;;;;;;;;;;;;;;;;;;;;;UAzDiB,gBAAA;;UAEP;eACK;SACN;uBACc;;UAGN,eAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;UA4BA,iBAAA;;SAER;;;;;;;;;;;;;;;;oBAgBW;;aAGR,eAAA"}

package/components/connect/types.d.ts

@@ -67,6 +67,24 @@ interface InFlightRequest {
   envId: string;
   functionSlug: string;
   accountId: string;
+  /**
+   * Timestamp (ms since epoch) when the worker acknowledged the request and
+   * took the lease. Used to compute age for shutdown diagnostics.
+   *
+   * Optional so external callers constructing an `InFlightRequest` (e.g.
+   * from {@link ConnectDebugState}) aren't forced to provide it.
+   */
+  leaseAcquiredAt?: number;
+  /**
+   * Timestamp (ms since epoch) of the most recent successful lease-extend send
+   * for this request. Starts equal to `leaseAcquiredAt` and is updated each
+   * time the worker sends a WORKER_REQUEST_EXTEND_LEASE message. Useful
+   * during shutdown diagnostics to tell whether a stuck request is still
+   * being actively leased or has gone silent.
+   *
+   * Optional for the same reason as {@link leaseAcquiredAt}.
+   */
+  leaseLastExtendedAt?: number;
 }
 interface ConnectDebugState {
   /** Current connection state */

package/components/connect/types.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"types.d.ts","names":[],"sources":["../../../src/components/connect/types.ts"],"sourcesContent":[],"mappings":";;;;;cAIa;UAEI,UAAA;EAFJ,MAAA,EAGH,OAAA,CAAQ,GAHL;EAEI,SAAA,CAAA,EAEH,KAFa,CAEP,eAAA,CAAgB,IAFT,CAAA;;AACjB,UAIO,qBAAA,SAA8B,eAJ7B,CAAA;MACE,EAIZ,UAJY,EAAA;;;AAGpB;;;;EAA8D,UAAA,CAAA,EAAA,MAAA;EAkD7C;;;;;sBAKM,CAAA,EAAA,MAAA;EAAiB;AAGxC;AAUA;;uBAES,CAAA,EAAA,MAAA,EAAA;;;AAmBT;;;;;;;;;;;;;;;;;;;;;;UAvCiB,gBAAA;;UAEP;eACK;SACN;uBACc;;UAGN,eAAA;;;;;;;;;UAUA,iBAAA;;SAER;;;;;;;;;;;;;;;;oBAgBW;;aAGR,eAAA"}
+{"version":3,"file":"types.d.ts","names":[],"sources":["../../../src/components/connect/types.ts"],"sourcesContent":[],"mappings":";;;;;cAIa;UAEI,UAAA;EAFJ,MAAA,EAGH,OAAA,CAAQ,GAHL;EAEI,SAAA,CAAA,EAEH,KAFa,CAEP,eAAA,CAAgB,IAFT,CAAA;;AACjB,UAIO,qBAAA,SAA8B,eAJ7B,CAAA;MACE,EAIZ,UAJY,EAAA;;;AAGpB;;;;EAA8D,UAAA,CAAA,EAAA,MAAA;EAkD7C;;;;;sBAKM,CAAA,EAAA,MAAA;EAAiB;AAGxC;AA4BA;;uBAES,CAAA,EAAA,MAAA,EAAA;;;AAmBT;;;;;;;;;;;;;;;;;;;;;;UAzDiB,gBAAA;;UAEP;eACK;SACN;uBACc;;UAGN,eAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;UA4BA,iBAAA;;SAER;;;;;;;;;;;;;;;;oBAgBW;;aAGR,eAAA"}

package/components/connect/types.js.map

@@ -1 +1 @@
-{"version":3,"file":"types.js","names":[],"sources":["../../../src/components/connect/types.ts"],"sourcesContent":["import type { RegisterOptions } from \"../../types.ts\";\nimport type { Inngest } from \"../Inngest.ts\";\nimport type { InngestFunction } from \"../InngestFunction.ts\";\n\nexport const DEFAULT_SHUTDOWN_SIGNALS = [\"SIGINT\", \"SIGTERM\"];\n\nexport interface ConnectApp {\n  client: Inngest.Any;\n  functions?: Array<InngestFunction.Like>;\n}\n\nexport interface ConnectHandlerOptions extends RegisterOptions {\n  apps: ConnectApp[];\n\n  /**\n   * InstanceId represents a stable identifier to be used for identifying connected SDKs.\n   * This can be a hostname or other identifier that remains stable across restarts.\n   *\n   * If nil, this defaults to the current machine's hostname.\n   */\n  instanceId?: string;\n\n  /**\n   * MaxWorkerConcurrency represents the maximum number of worker concurrency to use.\n   *\n   * If left undefined, there will be no limit on the number of concurrent requests on the worker.\n   */\n  maxWorkerConcurrency?: number;\n\n  /**\n   * By default, connections will be gracefully shut down when the current\n   * process receives a SIGINT or SIGTERM signal. Set this to an empty array to disable this behavior.\n   */\n  handleShutdownSignals?: string[];\n\n  /**\n   * Override the gateway WebSocket endpoint. When set, this URL is used\n   * instead of the endpoint returned by the Inngest API.\n   *\n   * Useful when there is a proxy between the worker and the gateway\n   * that requires a different hostname (e.g. `ws://localhost:8100`).\n   *\n   * Can also be set via the `INNGEST_CONNECT_GATEWAY_URL` environment variable.\n   * This option takes precedence over the env var.\n   */\n  gatewayUrl?: string;\n\n  /**\n   * Enable running the WebSocket connection, heartbeater, and lease extender\n   * in a separate worker thread. This prevents thread-blocking user code from\n   * interfering with connection health.\n   *\n   * Only works in environments that support worker_threads.\n   *\n   * Can also be disabled via the INNGEST_CONNECT_ISOLATE_EXECUTION=false environment variable.\n   *\n   * @default true\n   */\n  isolateExecution?: boolean;\n}\n\nexport interface WorkerConnection {\n  connectionId: string;\n  closed: Promise<void>;\n  close: () => Promise<void>;\n  state: ConnectionState;\n  getDebugState: () => ConnectDebugState;\n}\n\nexport interface InFlightRequest {\n  requestId: string;\n  runId: string;\n  stepId: string | undefined;\n  appId: string;\n  envId: string;\n  functionSlug: string;\n  accountId: string;\n}\n\nexport interface ConnectDebugState {\n  /** Current connection state */\n  state: ConnectionState;\n  /** Active connection ID, if any */\n  activeConnectionId: string | undefined;\n  /** Draining connection ID (during gateway drain), if any */\n  drainingConnectionId: string | undefined;\n  /** Timestamp of last heartbeat we sent */\n  lastHeartbeatSentAt: number | undefined;\n  /** Timestamp of last heartbeat response from gateway */\n  lastHeartbeatReceivedAt: number | undefined;\n  /** Timestamp of last WS message received (any type) */\n  lastMessageReceivedAt: number | undefined;\n  /** Whether shutdown has been requested (WORKER_PAUSE sent) */\n  shutdownRequested: boolean;\n  /** Number of in-flight requests */\n  inFlightRequestCount: number;\n  /** Detailed info about in-flight requests */\n  inFlightRequests: InFlightRequest[];\n}\n\nexport enum ConnectionState {\n  CONNECTING = \"CONNECTING\",\n  ACTIVE = \"ACTIVE\",\n  PAUSED = \"PAUSED\",\n  RECONNECTING = \"RECONNECTING\",\n  CLOSING = \"CLOSING\",\n  CLOSED = \"CLOSED\",\n}\n"],"mappings":";AAIA,MAAa,2BAA2B,CAAC,UAAU,UAAU;AAgG7D,IAAY,8DAAL;AACL;AACA;AACA;AACA;AACA;AACA"}
+{"version":3,"file":"types.js","names":[],"sources":["../../../src/components/connect/types.ts"],"sourcesContent":["import type { RegisterOptions } from \"../../types.ts\";\nimport type { Inngest } from \"../Inngest.ts\";\nimport type { InngestFunction } from \"../InngestFunction.ts\";\n\nexport const DEFAULT_SHUTDOWN_SIGNALS = [\"SIGINT\", \"SIGTERM\"];\n\nexport interface ConnectApp {\n  client: Inngest.Any;\n  functions?: Array<InngestFunction.Like>;\n}\n\nexport interface ConnectHandlerOptions extends RegisterOptions {\n  apps: ConnectApp[];\n\n  /**\n   * InstanceId represents a stable identifier to be used for identifying connected SDKs.\n   * This can be a hostname or other identifier that remains stable across restarts.\n   *\n   * If nil, this defaults to the current machine's hostname.\n   */\n  instanceId?: string;\n\n  /**\n   * MaxWorkerConcurrency represents the maximum number of worker concurrency to use.\n   *\n   * If left undefined, there will be no limit on the number of concurrent requests on the worker.\n   */\n  maxWorkerConcurrency?: number;\n\n  /**\n   * By default, connections will be gracefully shut down when the current\n   * process receives a SIGINT or SIGTERM signal. Set this to an empty array to disable this behavior.\n   */\n  handleShutdownSignals?: string[];\n\n  /**\n   * Override the gateway WebSocket endpoint. When set, this URL is used\n   * instead of the endpoint returned by the Inngest API.\n   *\n   * Useful when there is a proxy between the worker and the gateway\n   * that requires a different hostname (e.g. `ws://localhost:8100`).\n   *\n   * Can also be set via the `INNGEST_CONNECT_GATEWAY_URL` environment variable.\n   * This option takes precedence over the env var.\n   */\n  gatewayUrl?: string;\n\n  /**\n   * Enable running the WebSocket connection, heartbeater, and lease extender\n   * in a separate worker thread. This prevents thread-blocking user code from\n   * interfering with connection health.\n   *\n   * Only works in environments that support worker_threads.\n   *\n   * Can also be disabled via the INNGEST_CONNECT_ISOLATE_EXECUTION=false environment variable.\n   *\n   * @default true\n   */\n  isolateExecution?: boolean;\n}\n\nexport interface WorkerConnection {\n  connectionId: string;\n  closed: Promise<void>;\n  close: () => Promise<void>;\n  state: ConnectionState;\n  getDebugState: () => ConnectDebugState;\n}\n\nexport interface InFlightRequest {\n  requestId: string;\n  runId: string;\n  stepId: string | undefined;\n  appId: string;\n  envId: string;\n  functionSlug: string;\n  accountId: string;\n  /**\n   * Timestamp (ms since epoch) when the worker acknowledged the request and\n   * took the lease. Used to compute age for shutdown diagnostics.\n   *\n   * Optional so external callers constructing an `InFlightRequest` (e.g.\n   * from {@link ConnectDebugState}) aren't forced to provide it.\n   */\n  leaseAcquiredAt?: number;\n  /**\n   * Timestamp (ms since epoch) of the most recent successful lease-extend send\n   * for this request. Starts equal to `leaseAcquiredAt` and is updated each\n   * time the worker sends a WORKER_REQUEST_EXTEND_LEASE message. Useful\n   * during shutdown diagnostics to tell whether a stuck request is still\n   * being actively leased or has gone silent.\n   *\n   * Optional for the same reason as {@link leaseAcquiredAt}.\n   */\n  leaseLastExtendedAt?: number;\n}\n\nexport interface ConnectDebugState {\n  /** Current connection state */\n  state: ConnectionState;\n  /** Active connection ID, if any */\n  activeConnectionId: string | undefined;\n  /** Draining connection ID (during gateway drain), if any */\n  drainingConnectionId: string | undefined;\n  /** Timestamp of last heartbeat we sent */\n  lastHeartbeatSentAt: number | undefined;\n  /** Timestamp of last heartbeat response from gateway */\n  lastHeartbeatReceivedAt: number | undefined;\n  /** Timestamp of last WS message received (any type) */\n  lastMessageReceivedAt: number | undefined;\n  /** Whether shutdown has been requested (WORKER_PAUSE sent) */\n  shutdownRequested: boolean;\n  /** Number of in-flight requests */\n  inFlightRequestCount: number;\n  /** Detailed info about in-flight requests */\n  inFlightRequests: InFlightRequest[];\n}\n\nexport enum ConnectionState {\n  CONNECTING = \"CONNECTING\",\n  ACTIVE = \"ACTIVE\",\n  PAUSED = \"PAUSED\",\n  RECONNECTING = \"RECONNECTING\",\n  CLOSING = \"CLOSING\",\n  CLOSED = \"CLOSED\",\n}\n"],"mappings":";AAIA,MAAa,2BAA2B,CAAC,UAAU,UAAU;AAkH7D,IAAY,8DAAL;AACL;AACA;AACA;AACA;AACA;AACA"}

package/components/execution/engine.cjs

@@ -7,9 +7,9 @@ const require_types = require('../../types.cjs');
 const require_InngestExecution = require('./InngestExecution.cjs');
 const require_types$1 = require('../../helpers/types.cjs');
 const require_functions = require('../../helpers/functions.cjs');
+const require_temporal = require('../../helpers/temporal.cjs');
 const require_promises = require('../../helpers/promises.cjs');
 const require_als = require('./als.cjs');
-const require_temporal = require('../../helpers/temporal.cjs');
 const require_InngestStepTools = require('../InngestStepTools.cjs');
 const require_utils = require('../middleware/utils.cjs');
 const require_manager = require('../middleware/manager.cjs');

package/components/execution/engine.js

@@ -7,9 +7,9 @@ import { StepMode, StepOpCode, jsonErrorSchema } from "../../types.js";
 import { InngestExecution } from "./InngestExecution.js";
 import { isRecord } from "../../helpers/types.js";
 import { undefinedToNull } from "../../helpers/functions.js";
+import { isTemporalDuration } from "../../helpers/temporal.js";
 import { createDeferredPromise, createDeferredPromiseWithStack, createTimeoutPromise, goIntervalTiming, resolveAfterPending, resolveNextTick, retryWithBackoff, runAsPromise } from "../../helpers/promises.js";
 import { getAsyncCtx, getAsyncLocalStorage } from "./als.js";
-import { isTemporalDuration } from "../../helpers/temporal.js";
 import { STEP_INDEXING_SUFFIX, createStepTools, experimentStepRunSymbol, getStepOptions } from "../InngestStepTools.js";
 import { UnreachableError } from "../middleware/utils.js";
 import { MiddlewareManager } from "../middleware/manager.js";

package/components/execution/otel/middleware.d.cts

@@ -72,11 +72,13 @@ declare const extendedTracesMiddleware: ({
       ctx: {
         tracer: _opentelemetry_api0.Tracer;
         event: Simplify<EventPayload>;
+        events: AsTuple<Simplify<EventPayload>>;
+        runId: string;
         step: {
           sendEvent: (idOrOptions: StepOptionsOrId, payload: SendEventPayload) => Promise<SendEventOutput<ClientOptions>>;
           waitForSignal: <TData>(idOrOptions: StepOptionsOrId, opts: {
             signal: string;
-            timeout: number | string | Date;
+            timeout: number | string | Date | DurationLike | InstantLike | ZonedDateTimeLike;
             onConflict: "replace" | "fail";
           }) => Promise<{
             signal: string;
@@ -91,7 +93,7 @@ declare const extendedTracesMiddleware: ({
           }) => Promise<InngestApi.SendSignalResponse>;
           waitForEvent: <TOpts extends {
             event: string | EventType<string, any>;
-            timeout: number | string | Date;
+            timeout: number | string | Date | DurationLike | InstantLike | ZonedDateTimeLike;
           } & ExclusiveKeys<{
             match?: string;
             if?: string;
@@ -149,12 +151,10 @@ declare const extendedTracesMiddleware: ({
           invoke: <TFunction extends InvokeTargetFunctionDefinition>(idOrOptions: StepOptionsOrId, opts: {
             function: TFunction;
           } & Omit<TriggerEventFromFunction<TFunction>, "id"> & {
-            timeout?: number | string | Date;
+            timeout?: number | string | Date | DurationLike | InstantLike | ZonedDateTimeLike;
           }) => InvocationResult<Jsonify<GetFunctionOutputRaw<TFunction>>>;
           fetch: StepFetch;
         };
-        events: AsTuple<Simplify<EventPayload>>;
-        runId: string;
         group: ReturnType<typeof createGroupTools>;
         attempt: number;
         maxAttempts?: number | undefined;
@@ -234,7 +234,13 @@ declare const extendedTracesMiddleware: ({
             };
             readonly if?: string | undefined;
             readonly match?: string | undefined;
-            readonly timeout?: string | number | Readonly<Date> | undefined;
+            readonly timeout?: string | number | Readonly<Date> | {
+              readonly [Symbol.toStringTag]: "Temporal.Duration";
+            } | {
+              readonly [Symbol.toStringTag]: "Temporal.Instant";
+            } | {
+              readonly [Symbol.toStringTag]: "Temporal.ZonedDateTime";
+            } | undefined;
           }[] | undefined;
           readonly retries?: 0 | 1 | 2 | 3 | 4 | 5 | 6 | 7 | 8 | 9 | 10 | 11 | 12 | 13 | 14 | 15 | 16 | 17 | 18 | 19 | 20 | undefined;
           readonly onFailure?: any;

package/components/execution/otel/middleware.d.cts.map

@@ -1 +1 @@
-{"version":3,"file":"middleware.d.cts","names":[],"sources":["../../../../src/components/execution/otel/middleware.ts"],"sourcesContent":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;UA6BiB,+BAAA;;;;;;;;;;;;;EAAA,SAAA,CAAA,EAaH,SAbG;EAA+B;;;;;AAwChD;;kBAAyC,CAAA,EAlBpB,gBAkBoB;;;;;;;UA6GO,CAAA,EAvHnC,YAuHmC;;;;;;;;;cA7GnC;;;;IAIV;;;;;;;gCAyGQ,UAAA,CAAW;;gBAA0B,mBAAA,CAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;UAWI,CAAA,GAAA,SAAA;UAAe,SAAA,SAAA,CAAA,EAAA;YAAA,SAAA,GAAA,CAAA,EAAA,MAAA,GAAA,SAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;UApCb,SAAA,mBAAA,CAAA,EAAA,OAAA,GAAA,SAAA;UAAc,SAAA,yBAAA,CAAA,EAAA,OAAA,GAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OAoC3B,UAAA,CAAW,kBAAe,QAAA,UAAA,CAAA;;;;;;;;;;;;;;;;;;;;;;;;KApCxB,UAAA,CAAW"}
+{"version":3,"file":"middleware.d.cts","names":[],"sources":["../../../../src/components/execution/otel/middleware.ts"],"sourcesContent":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;UA6BiB,+BAAA;;;;;;;;;;;;;EAAA,SAAA,CAAA,EAaH,SAbG;EAA+B;;;;;AAwChD;;kBAAyC,CAAA,EAlBpB,gBAkBoB;;;;;;;UA6GO,CAAA,EAvHnC,YAuHmC;;;;;;;;;cA7GnC;;;;IAIV;;;;;;;gCAyGQ,UAAA,CAAW;;gBAA0B,mBAAA,CAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;gBAWI,EAAA,CAAA,EAAA,MAAA;gBAAe,EAAA,CAAA,EAAA,MAAA;gBAAA,CAAA,CAAA,EAAA,MAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;YApCb,SAAA,WAAA,CAAA,EAAA,MAAA,GAAA,MAAA,GAAA;cAAc,6BAAA,EAAA,mBAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OAoC3B,UAAA,CAAW,kBAAe,QAAA,UAAA,CAAA;;;;;;;;;;;;;;;;;;;;;;;;KApCxB,UAAA,CAAW"}

package/components/execution/otel/middleware.d.ts

@@ -72,11 +72,13 @@ declare const extendedTracesMiddleware: ({
       ctx: {
         tracer: _opentelemetry_api0.Tracer;
         event: Simplify<EventPayload>;
+        events: AsTuple<Simplify<EventPayload>>;
+        runId: string;
         step: {
           sendEvent: (idOrOptions: StepOptionsOrId, payload: SendEventPayload) => Promise<SendEventOutput<ClientOptions>>;
           waitForSignal: <TData>(idOrOptions: StepOptionsOrId, opts: {
             signal: string;
-            timeout: number | string | Date;
+            timeout: number | string | Date | DurationLike | InstantLike | ZonedDateTimeLike;
             onConflict: "replace" | "fail";
           }) => Promise<{
             signal: string;
@@ -91,7 +93,7 @@ declare const extendedTracesMiddleware: ({
           }) => Promise<InngestApi.SendSignalResponse>;
           waitForEvent: <TOpts extends {
             event: string | EventType<string, any>;
-            timeout: number | string | Date;
+            timeout: number | string | Date | DurationLike | InstantLike | ZonedDateTimeLike;
           } & ExclusiveKeys<{
             match?: string;
             if?: string;
@@ -149,12 +151,10 @@ declare const extendedTracesMiddleware: ({
           invoke: <TFunction extends InvokeTargetFunctionDefinition>(idOrOptions: StepOptionsOrId, opts: {
             function: TFunction;
           } & Omit<TriggerEventFromFunction<TFunction>, "id"> & {
-            timeout?: number | string | Date;
+            timeout?: number | string | Date | DurationLike | InstantLike | ZonedDateTimeLike;
           }) => InvocationResult<Jsonify<GetFunctionOutputRaw<TFunction>>>;
           fetch: StepFetch;
         };
-        events: AsTuple<Simplify<EventPayload>>;
-        runId: string;
         group: ReturnType<typeof createGroupTools>;
         attempt: number;
         maxAttempts?: number | undefined;
@@ -234,7 +234,13 @@ declare const extendedTracesMiddleware: ({
             };
             readonly if?: string | undefined;
             readonly match?: string | undefined;
-            readonly timeout?: string | number | Readonly<Date> | undefined;
+            readonly timeout?: string | number | Readonly<Date> | {
+              readonly [Symbol.toStringTag]: "Temporal.Duration";
+            } | {
+              readonly [Symbol.toStringTag]: "Temporal.Instant";
+            } | {
+              readonly [Symbol.toStringTag]: "Temporal.ZonedDateTime";
+            } | undefined;
           }[] | undefined;
           readonly retries?: 0 | 1 | 2 | 3 | 4 | 5 | 6 | 7 | 8 | 9 | 10 | 11 | 12 | 13 | 14 | 15 | 16 | 17 | 18 | 19 | 20 | undefined;
           readonly onFailure?: any;

package/components/execution/otel/middleware.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"middleware.d.ts","names":[],"sources":["../../../../src/components/execution/otel/middleware.ts"],"sourcesContent":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;UA6BiB,+BAAA;;;;;;;;;;;;;EAAA,SAAA,CAAA,EAaH,SAbG;EAA+B;;;;;AAwChD;;kBAAyC,CAAA,EAlBpB,gBAkBoB;;;;;;;UA6GO,CAAA,EAvHnC,YAuHmC;;;;;;;;;cA7GnC;;;;IAIV;;;;;;;gCAyGQ,UAAA,CAAW;;gBAA0B,mBAAA,CAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;UAWI,CAAA,GAAA,SAAA;UAAe,SAAA,SAAA,CAAA,EAAA;YAAA,SAAA,GAAA,CAAA,EAAA,MAAA,GAAA,SAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;UApCb,SAAA,mBAAA,CAAA,EAAA,OAAA,GAAA,SAAA;UAAc,SAAA,yBAAA,CAAA,EAAA,OAAA,GAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OAoC3B,UAAA,CAAW,kBAAe,QAAA,UAAA,CAAA;;;;;;;;;;;;;;;;;;;;;;;;KApCxB,UAAA,CAAW"}
+{"version":3,"file":"middleware.d.ts","names":[],"sources":["../../../../src/components/execution/otel/middleware.ts"],"sourcesContent":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;UA6BiB,+BAAA;;;;;;;;;;;;;EAAA,SAAA,CAAA,EAaH,SAbG;EAA+B;;;;;AAwChD;;kBAAyC,CAAA,EAlBpB,gBAkBoB;;;;;;;UA6GO,CAAA,EAvHnC,YAuHmC;;;;;;;;;cA7GnC;;;;IAIV;;;;;;;gCAyGQ,UAAA,CAAW;;gBAA0B,mBAAA,CAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;gBAWI,EAAA,CAAA,EAAA,MAAA;gBAAe,EAAA,CAAA,EAAA,MAAA;gBAAA,CAAA,CAAA,EAAA,MAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;YApCb,SAAA,WAAA,CAAA,EAAA,MAAA,GAAA,MAAA,GAAA;cAAc,6BAAA,EAAA,mBAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OAoC3B,UAAA,CAAW,kBAAe,QAAA,UAAA,CAAA;;;;;;;;;;;;;;;;;;;;;;;;KApCxB,UAAA,CAAW"}

package/components/middleware/manager.cjs

@@ -64,7 +64,7 @@ var MiddlewareManager = class {
 		}
 		let opName;
 		if (stepType === "sleep" && stepInfo.input !== void 0) {
-			if (!require_utils.isTimeStrInput(stepInfo.input[0])) throw new Error("Sleep time must be a string, number, or Date");
+			if (!require_utils.isSleepInput(stepInfo.input[0])) throw new Error("Sleep time must be a string, number, Date, or Temporal.Duration");
 			opName = require_strings.timeStr(stepInfo.input[0]);
 		}
 		const opOpts = require_utils.optsFromStepInput(stepType, stepInfo.input);