fix: streams not closing properly

Vali-98 · Dec 5, 2024 · bdcb08f · bdcb08f
1 parent 657947d
commit bdcb08f
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/constants/SSEFetch.ts b/constants/SSEFetch.ts
@@ -13,9 +13,12 @@ export class SSEFetch {
     private onEvent = (data: string) => {}
     private onError = () => {}
     private onClose = () => {}
+    private closeStream = () => {}
 
     public abort() {
         this.abortController.abort()
+        this.closeStream()
+        this.closeStream = () => {}
     }
 
     public async start(values: SSEValues) {
@@ -29,8 +32,8 @@ export class SSEFetch {
             ...body,
         }).then(async (res) => {
             if (res.status !== 200 || !res.body) return this.onError()
+            this.closeStream = res.body.cancel
             for await (const chunk of res.body) {
-                if (this.abortController.signal.aborted) break
                 const data = this.decoder.decode(chunk)
                 const output = parseSSE(data)
                 output.forEach((item) => this.onEvent(item))