@mastra/fastify 1.3.19-alpha.2 → 1.3.19-alpha.4

package/CHANGELOG.md

@@ -1,5 +1,23 @@
 # @mastra/fastify
 
+## 1.3.19-alpha.4
+
+### Patch Changes
+
+- Fixed Fastify stream cleanup and route abort signals when clients disconnect before streamed responses finish. ([#16308](https://github.com/mastra-ai/mastra/pull/16308))
+
+- Updated dependencies [[`9f17410`](https://github.com/mastra-ai/mastra/commit/9f1741080def23d42ee50b39887a385ae316a3c6), [`c6eb39e`](https://github.com/mastra-ai/mastra/commit/c6eb39ea6dca381c6563cb240237fbe608e02f93), [`900d086`](https://github.com/mastra-ai/mastra/commit/900d086bb737b9cf2fcf68f11b0389b801a2738c), [`4c0e286`](https://github.com/mastra-ai/mastra/commit/4c0e28637c9cfb4f416549b55e97ebfa13319dfc), [`c6eb39e`](https://github.com/mastra-ai/mastra/commit/c6eb39ea6dca381c6563cb240237fbe608e02f93), [`25184ff`](https://github.com/mastra-ai/mastra/commit/25184ffaf1293ec95119426eb1a1f8d38831b96c), [`25184ff`](https://github.com/mastra-ai/mastra/commit/25184ffaf1293ec95119426eb1a1f8d38831b96c), [`aebde9c`](https://github.com/mastra-ai/mastra/commit/aebde9cfacf56592c6b6350cae721740fe090b8a)]:
+  - @mastra/core@1.33.0-alpha.4
+  - @mastra/server@1.33.0-alpha.4
+
+## 1.3.19-alpha.3
+
+### Patch Changes
+
+- Updated dependencies [[`087e413`](https://github.com/mastra-ai/mastra/commit/087e4133e5d6efa36619e9556c16750e4179c047), [`087e413`](https://github.com/mastra-ai/mastra/commit/087e4133e5d6efa36619e9556c16750e4179c047), [`087e413`](https://github.com/mastra-ai/mastra/commit/087e4133e5d6efa36619e9556c16750e4179c047), [`087e413`](https://github.com/mastra-ai/mastra/commit/087e4133e5d6efa36619e9556c16750e4179c047), [`0461546`](https://github.com/mastra-ai/mastra/commit/0461546755951706ca81bc24d1d31013d9d70a6d)]:
+  - @mastra/core@1.33.0-alpha.3
+  - @mastra/server@1.33.0-alpha.3
+
 ## 1.3.19-alpha.2
 
 ### Patch Changes

package/dist/index.cjs

@@ -306,9 +306,12 @@ function toWebRequest2(request) {
     headers
   });
 }
+function isRequestAborted(rawRequest) {
+  return rawRequest.aborted || rawRequest.readableAborted || !rawRequest.complete;
+}
 var MastraServer = class extends serverAdapter.MastraServer {
   createContextMiddleware() {
-    return async (request, _reply) => {
+    return async (request, reply) => {
       let bodyRequestContext;
       let paramsRequestContext;
       if (request.method === "POST" || request.method === "PUT") {
@@ -348,14 +351,19 @@ var MastraServer = class extends serverAdapter.MastraServer {
       request.customRouteAuthConfig = this.customRouteAuthConfig;
       const controller = new AbortController();
       request.raw.on("close", () => {
-        if (!request.raw.complete) {
+        if (isRequestAborted(request.raw)) {
+          controller.abort();
+        }
+      });
+      reply.raw.on("close", () => {
+        if (!reply.raw.writableEnded) {
           controller.abort();
         }
       });
       request.abortSignal = controller.signal;
     };
   }
-  async stream(route, reply, result) {
+  async stream(route, reply, result, request) {
     const rawHeaders = reply.getHeaders();
     const existingHeaders = {};
     for (const [key, value] of Object.entries(rawHeaders)) {
@@ -381,9 +389,20 @@ var MastraServer = class extends serverAdapter.MastraServer {
     });
     const readableStream = result instanceof ReadableStream ? result : result.fullStream;
     const reader = readableStream.getReader();
-    reply.raw.on("close", () => {
-      void reader.cancel("request aborted");
-    });
+    let readerCanceled = false;
+    const cancelReader = (reason) => {
+      if (readerCanceled) return;
+      readerCanceled = true;
+      void reader.cancel(reason);
+    };
+    const cancelReaderOnResponseClose = () => cancelReader("request aborted");
+    const cancelReaderOnRequestClose = () => {
+      if (request && isRequestAborted(request.raw)) {
+        cancelReader("request aborted");
+      }
+    };
+    reply.raw.on("close", cancelReaderOnResponseClose);
+    request?.raw.on("close", cancelReaderOnRequestClose);
     try {
       while (true) {
         const { done, value } = await reader.read();
@@ -405,7 +424,11 @@ var MastraServer = class extends serverAdapter.MastraServer {
         error: error instanceof Error ? { message: error.message, stack: error.stack } : error
       });
     } finally {
-      reply.raw.end();
+      reply.raw.off("close", cancelReaderOnResponseClose);
+      request?.raw.off("close", cancelReaderOnRequestClose);
+      if (!reply.raw.writableEnded && !reply.raw.destroyed) {
+        reply.raw.end();
+      }
     }
   }
   async getParams(route, request) {
@@ -500,20 +523,34 @@ var MastraServer = class extends serverAdapter.MastraServer {
     if (route.responseType === "json") {
       await reply.send(result);
     } else if (route.responseType === "stream") {
-      await this.stream(route, reply, result);
+      await this.stream(route, reply, result, request);
     } else if (route.responseType === "datastream-response") {
       const fetchResponse = result;
       fetchResponse.headers.forEach((value, key) => reply.header(key, value));
       reply.status(fetchResponse.status);
       if (fetchResponse.body) {
         const reader = fetchResponse.body.getReader();
+        let readerCanceled = false;
+        const cancelReader = (reason) => {
+          if (readerCanceled) return;
+          readerCanceled = true;
+          void reader.cancel(reason);
+        };
+        const cancelReaderOnResponseClose = () => cancelReader("request aborted");
+        const cancelReaderOnRequestClose = () => {
+          if (request && isRequestAborted(request.raw)) {
+            cancelReader("request aborted");
+          }
+        };
         const onResError = (err) => {
           this.mastra.getLogger()?.error("Error writing datastream response", {
             error: err instanceof Error ? { message: err.message, stack: err.stack } : err
           });
-          void reader.cancel("response write error");
+          cancelReader("response write error");
         };
         reply.raw.once("error", onResError);
+        reply.raw.on("close", cancelReaderOnResponseClose);
+        request?.raw.on("close", cancelReaderOnRequestClose);
         try {
           while (true) {
             const { done, value } = await reader.read();
@@ -526,7 +563,11 @@ var MastraServer = class extends serverAdapter.MastraServer {
           });
         } finally {
           reply.raw.off("error", onResError);
-          reply.raw.end();
+          reply.raw.off("close", cancelReaderOnResponseClose);
+          request?.raw.off("close", cancelReaderOnRequestClose);
+          if (!reply.raw.writableEnded && !reply.raw.destroyed) {
+            reply.raw.end();
+          }
         }
       } else {
         reply.raw.end();