@donkeylabs/server 2.0.3 → 2.0.5

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@donkeylabs/server",
-  "version": "2.0.3",
+  "version": "2.0.5",
   "type": "module",
   "description": "Type-safe plugin system for building RPC-style APIs with Bun",
   "main": "./src/index.ts",

package/src/core/sse.ts

@@ -58,6 +58,10 @@ class SSEImpl implements SSE {
         // Send retry interval to client
         const retryMsg = `retry: ${this.retryInterval}\n\n`;
         controller.enqueue(this.encoder.encode(retryMsg));
+
+        // Send immediate heartbeat to establish connection and prevent early timeout
+        const heartbeat = `: heartbeat ${Date.now()}\n\n`;
+        controller.enqueue(this.encoder.encode(heartbeat));
       },
       cancel: () => {
         this.removeClient(id);

package/src/handlers.ts

@@ -194,6 +194,11 @@ export const SSEHandler: SSEHandler = {
         ctx.core.sse.subscribe(client.id, channel);
       }
 
+      // Clean up client when connection is aborted
+      req.signal.addEventListener("abort", () => {
+        ctx.core.sse.removeClient(client.id);
+      });
+
       return response;
     } catch (e: any) {
       console.error(e);