svelte-adapter-uws 0.4.12 → 0.4.13

package/files/handler.js

@@ -12,7 +12,7 @@ import { manifest, prerendered, base } from 'MANIFEST';
 import { env } from 'ENV';
 import * as wsModule from 'WS_HANDLER';
 import { parseCookies, createCookies } from './cookies.js';
-import { mimeLookup, parse_as_bytes, parse_origin } from './utils.js';
+import { mimeLookup, parse_as_bytes, parse_origin, writeChunkWithBackpressure } from './utils.js';
 
 /* global ENV_PREFIX */
 /* global PRECOMPRESS */
@@ -1209,11 +1209,11 @@ async function writeResponse(res, response, state, acceptEncoding) {
 			return;
 		}
 
-		// Multi-chunk streaming response - write headers + first two chunks in one cork.
-		// cork() batches these writes into a single syscall, so backpressure from
-		// individual res.write() calls inside cork is not actionable  - the data is
-		// buffered and flushed together when cork returns. The backpressure loop
-		// below handles all subsequent chunks.
+		// Multi-chunk streaming response. Headers + first two chunks share one
+		// cork so they flush as a single syscall. Subsequent chunks are each
+		// written inside their own cork via writeChunkWithBackpressure, which
+		// also captures the drain signal from res.write() without tripping the
+		// uWS "writes must be made from within a corked callback" warning.
 		if (state.aborted) return;
 		streaming = true;
 		res.cork(() => {
@@ -1222,17 +1222,13 @@ async function writeResponse(res, response, state, acceptEncoding) {
 			res.write(second.value);
 		});
 
-		// Stream remaining chunks with backpressure (30s timeout per drain)
 		for (;;) {
 			const { done, value } = await reader.read();
 			if (done || state.aborted) break;
 
-			const ok = res.write(value);
-			if (!ok) {
-				const drained = await new Promise((resolve) => {
-					const timer = setTimeout(() => resolve(false), 30000);
-					res.onWritable(() => { clearTimeout(timer); resolve(true); return true; });
-				});
+			const result = writeChunkWithBackpressure(res, value);
+			if (result !== true) {
+				const drained = await result;
 				if (!drained) { streamTimedOut = true; break; }
 				if (state.aborted) break;
 			}
@@ -1243,7 +1239,7 @@ async function writeResponse(res, response, state, acceptEncoding) {
 				// Backpressure drained past the 30s deadline. Abruptly close the
 				// connection rather than sending a clean EOF on a partial body,
 				// which would look like a successful but truncated response.
-				res.close();
+				res.cork(() => res.close());
 			} else {
 				res.cork(() => res.end());
 			}

package/files/utils.js

@@ -111,6 +111,40 @@ export function parse_as_bytes(value) {
 	return Number(multiplier !== 1 ? normalized.slice(0, -1) : normalized) * multiplier;
 }
 
+/**
+ * Write a chunk to a uWS HttpResponse inside a cork and, if backpressure
+ * builds, return a Promise that resolves when the socket drains or the
+ * timeout elapses. Returns `true` synchronously when no drain is needed.
+ *
+ * All uWS response mutations (write + onWritable registration) happen
+ * inside the cork callback, which uWS invokes synchronously, so the
+ * boolean return value of `res.write()` is captured correctly.
+ *
+ * @param {{ cork: (fn: () => void) => void, write: (value: any) => boolean, onWritable: (fn: () => boolean) => void }} res
+ * @param {any} value
+ * @param {number} [timeoutMs]
+ * @returns {true | Promise<boolean>} true if the write succeeded without drain; otherwise a promise that resolves true on drain or false on timeout.
+ */
+export function writeChunkWithBackpressure(res, value, timeoutMs = 30000) {
+	let ok = false;
+	/** @type {Promise<boolean> | null} */
+	let drainPromise = null;
+	res.cork(() => {
+		ok = res.write(value);
+		if (!ok) {
+			drainPromise = new Promise((resolve) => {
+				const timer = setTimeout(() => resolve(false), timeoutMs);
+				res.onWritable(() => {
+					clearTimeout(timer);
+					resolve(true);
+					return true;
+				});
+			});
+		}
+	});
+	return ok ? true : /** @type {Promise<boolean>} */ (drainPromise);
+}
+
 /**
  * @param {string | undefined} value
  * @returns {string | undefined}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "svelte-adapter-uws",
-  "version": "0.4.12",
+  "version": "0.4.13",
   "description": "SvelteKit adapter for uWebSockets.js - high-performance C++ HTTP server with built-in WebSocket support",
   "author": "Kevin Radziszewski",
   "license": "MIT",