@apifuse/provider-sdk 2.2.0-beta.27 → 2.2.0-beta.28

package/src/stateful/errors.ts

@@ -1,5 +1,17 @@
 export type StatefulControlPlaneOperation = "resolve" | "acquire" | "renew" | "release";
 
+export class StatefulRoutingDeadlineError extends Error {
+	readonly requestId: string;
+	readonly deadlineAt: string;
+
+	constructor(requestId: string, deadlineAt: string) {
+		super(`Stateful operation request ${requestId} deadline has expired.`);
+		this.name = "StatefulRoutingDeadlineError";
+		this.requestId = requestId;
+		this.deadlineAt = deadlineAt;
+	}
+}
+
 export class StatefulControlPlaneError extends Error {
 	readonly code: string;
 	readonly operation: StatefulControlPlaneOperation;

package/src/stateful/stateful-provider-session-routing.ts

@@ -1,4 +1,5 @@
 import type { SessionKey } from "./session-key.js";
+import { StatefulRoutingDeadlineError } from "./errors.js";
 import {
 	NOOP_STATEFUL_PROVIDER_METRIC_EMITTER,
 	type StatefulProviderMetricEmitter,
@@ -82,17 +83,7 @@ type DeadlineSignal = {
 	readonly cleanup: () => void;
 };
 
-export class StatefulRoutingDeadlineError extends Error {
-	readonly requestId: string;
-	readonly deadlineAt: string;
-
-	constructor(requestId: string, deadlineAt: string) {
-		super(`Stateful operation request ${requestId} deadline has expired.`);
-		this.name = "StatefulRoutingDeadlineError";
-		this.requestId = requestId;
-		this.deadlineAt = deadlineAt;
-	}
-}
+export { StatefulRoutingDeadlineError } from "./errors.js";
 
 export class StatefulRoutingOwnershipError extends Error {
 	readonly requestId: string;

package/src/stream.ts

@@ -85,13 +85,20 @@ function sseFieldValue(value: string, field: "event" | "id"): string {
 
 export async function* readableBytes(body: ReadableStream<Uint8Array>): AsyncIterable<Uint8Array> {
 	const reader = body.getReader();
+	let completed = false;
 	try {
 		for (;;) {
 			const { value, done } = await reader.read();
-			if (done) return;
+			if (done) {
+				completed = true;
+				return;
+			}
 			if (value) yield value;
 		}
 	} finally {
+		if (!completed) {
+			await reader.cancel().catch(() => undefined);
+		}
 		reader.releaseLock();
 	}
 }