fix: do not send heartbeat to closed controller

alkinum · Aug 25, 2024 · 0842c17 · 0842c17
1 parent 84ce79d
commit 0842c17
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/src/pages/api/push.ts b/src/pages/api/push.ts
@@ -191,4 +191,4 @@ export const POST: APIRoute = async ({ request, locals }) => {
       headers: { 'Content-Type': 'application/json' },
     });
   }
-};
+};
diff --git a/src/pages/api/stream.ts b/src/pages/api/stream.ts
@@ -76,7 +76,12 @@ export const GET: APIRoute = async ({ request, locals }) => {
           return;
         }
         try {
-          controller.enqueue(`event: heartbeat\ndata: ${new Date().toISOString()}\n\n`);
+          // Check if the controller is closed
+          if (controller.desiredSize !== null) {
+            controller.enqueue(`event: heartbeat\ndata: ${new Date().toISOString()}\n\n`);
+          } else {
+            throw new Error('Controller is closed');
+          }
         } catch (error) {
           console.error('Error sending heartbeat:', error);
           clearInterval(heartbeatInterval);
-Original file line number
+Diff line change
@@ Expand Up @@
           headers: { 'Content-Type': 'application/json' },
         });
       }
-    };
+    };